Only consider display class for gpus

metal-stack · Oct 15, 2024 · c952b2b · c952b2b
1 parent b0371a4
commit c952b2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmd/register/register.go b/cmd/register/register.go
@@ -247,7 +247,7 @@ func (r *Register) detectGPUs() (pci.Devices, error) {
 		}
 		r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName, "classname", device.ClassName)
 
-		if strings.ToLower(device.ClassName) != "display" {
+		if !strings.Contains(strings.ToLower(device.ClassName), "display") {
 			continue
 		}
 		result = append(result, device)