Merge pull request #1935 from dhiltgen/cpu_fallback

Fix up the CPU fallback selection
This commit is contained in:
Daniel Hiltgen
2024-01-11 15:52:32 -08:00
committed by GitHub
4 changed files with 29 additions and 16 deletions

View File

@@ -28,6 +28,13 @@ func getDynLibs(gpuInfo gpu.GpuInfo) []string {
if gpuInfo.Library == "default" {
return []string{"default"}
}
// TODO - temporary until we have multiple CPU variations for Darwin
// Short circuit on darwin with metal only
if len(availableDynLibs) == 1 {
if _, onlyMetal := availableDynLibs["metal"]; onlyMetal {
return []string{availableDynLibs["metal"]}
}
}
exactMatch := ""
dynLibs := []string{}