fix(backends gallery): correctly identify gpu vendor (#5739)

mudler · web-flow · commit 6f41a6f934ac · 2025-06-27T22:22:58.000+02:00
Signed-off-by: Ettore Di Giacinto &lt;mudler@localai.io&gt;
diff --git a/core/system/capabilities.go b/core/system/capabilities.go
@@ -30,13 +30,13 @@ func detectGPUVendor() (string, error) {
 		if gpu.DeviceInfo != nil {
 			if gpu.DeviceInfo.Vendor != nil {
 				gpuVendorName := strings.ToUpper(gpu.DeviceInfo.Vendor.Name)
-				if gpuVendorName == "NVIDIA" {
+				if strings.Contains(gpuVendorName, "NVIDIA") {
 					return "nvidia", nil
 				}
-				if gpuVendorName == "AMD" {
+				if strings.Contains(gpuVendorName, "AMD") {
 					return "amd", nil
 				}
-				if gpuVendorName == "INTEL" {
+				if strings.Contains(gpuVendorName, "INTEL") {
 					return "intel", nil
 				}
 				return "nvidia", nil

Original file line number	Diff line number	Diff line change
`@@ -30,13 +30,13 @@ func detectGPUVendor() (string, error) {`
`30`	`30`	`if gpu.DeviceInfo != nil {`
`31`	`31`	`if gpu.DeviceInfo.Vendor != nil {`
`32`	`32`	`gpuVendorName := strings.ToUpper(gpu.DeviceInfo.Vendor.Name)`
`33`		`- if gpuVendorName == "NVIDIA" {`
	`33`	`+ if strings.Contains(gpuVendorName, "NVIDIA") {`
`34`	`34`	`return "nvidia", nil`
`35`	`35`	`}`
`36`		`- if gpuVendorName == "AMD" {`
	`36`	`+ if strings.Contains(gpuVendorName, "AMD") {`
`37`	`37`	`return "amd", nil`
`38`	`38`	`}`
`39`		`- if gpuVendorName == "INTEL" {`
	`39`	`+ if strings.Contains(gpuVendorName, "INTEL") {`
`40`	`40`	`return "intel", nil`
`41`	`41`	`}`
`42`	`42`	`return "nvidia", nil`