Merge pull request #1999 from lainedfles/termux_android_cpu_only

Fix CPU-only build under Android Termux enviornment.
This commit is contained in:
Daniel Hiltgen
2024-01-18 17:16:53 -08:00
committed by GitHub

View File

@@ -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)