From c952b2b2e9300ab2054f0f41ca638c5fc6713e17 Mon Sep 17 00:00:00 2001 From: Stefan Majer Date: Tue, 15 Oct 2024 08:42:11 +0200 Subject: [PATCH] Only consider display class for gpus --- cmd/register/register.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmd/register/register.go b/cmd/register/register.go index 0be6c09..fd66440 100644 --- 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)