gpu: Group GPU Library sets by variant (#6483)

The recent cuda variant changes uncovered a bug in ByLibrary which failed to group by common variant for GPU types.
2025-12-11 00:07:07 +00:00 · 2024-08-23 15:11:56 -07:00
parent 9638c24c58
commit 69be940bf6
2 changed files with 26 additions and 1 deletions
--- a/gpu/types.go
+++ b/gpu/types.go
@@ -94,7 +94,7 @@ func (l GpuInfoList) ByLibrary() []GpuInfoList {
 			}
 		}
 		if !found {
-			libs = append(libs, info.Library)
+			libs = append(libs, requested)
 			resp = append(resp, []GpuInfo{info})
 		}
 	}