Browse Source

Merge pull request #1999 from lainedfles/termux_android_cpu_only

Fix CPU-only build under Android Termux enviornment.
Daniel Hiltgen 1 year ago
parent
commit
62976087c6
1 changed files with 1 additions and 1 deletions
  1. 1 1
      gpu/gpu.go

+ 1 - 1
gpu/gpu.go

@@ -63,6 +63,7 @@ func initGPUHandles() {
 
 	// TODO - if the ollama build is CPU only, don't do these checks as they're irrelevant and confusing
 
+	gpuHandles = &handles{nil, nil}
 	var cudaMgmtName string
 	var cudaMgmtPatterns []string
 	var rocmMgmtName string
@@ -87,7 +88,6 @@ func initGPUHandles() {
 	}
 
 	slog.Info("Detecting GPU type")
-	gpuHandles = &handles{nil, nil}
 	cudaLibPaths := FindGPULibs(cudaMgmtName, cudaMgmtPatterns)
 	if len(cudaLibPaths) > 0 {
 		cuda := LoadCUDAMgmt(cudaLibPaths)