baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #6290 from dhiltgen/intel_npe

Browse source 
Harden intel boostrap for nil pointers
This commit is contained in:
Daniel Hiltgen 2024-08-09 12:14:42 -07:00 committed by GitHub
commit 71b0945fc6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 32 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
gpu/gpu.go
View file

@ -305,6 +305,8 @@ func GetGPUInfo() GpuInfoList {
// Intel // Intel
if envconfig.IntelGPU() { if envconfig.IntelGPU() {
oHandles = initOneAPIHandles() oHandles = initOneAPIHandles()
if oHandles != nil && oHandles.oneapi != nil {
// On windows we bundle the oneapi library one level above the runner dir // On windows we bundle the oneapi library one level above the runner dir
depPath = "" depPath = ""
if runtime.GOOS == "windows" && envconfig.RunnersDir() != "" { if runtime.GOOS == "windows" && envconfig.RunnersDir() != "" {
@ -340,6 +342,7 @@ func GetGPUInfo() GpuInfoList {
} }
} }
} }
}
rocmGPUs = AMDGetGPUInfo() rocmGPUs = AMDGetGPUInfo()
bootstrapped = true bootstrapped = true