From b095b39d7484273fda31fa91e30897b193381867 Mon Sep 17 00:00:00 2001 From: Stefan Majer Date: Tue, 15 Oct 2024 08:07:14 +0200 Subject: [PATCH 1/3] Support all NVidia devices --- cmd/register/register.go | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/cmd/register/register.go b/cmd/register/register.go index 7b6b8ef..3f16f2b 100644 --- a/cmd/register/register.go +++ b/cmd/register/register.go @@ -246,11 +246,8 @@ func (r *Register) detectGPUs() (pci.Devices, error) { continue } - // TODO if new models must be supported, this code must be refactored - if strings.Contains(strings.ToLower(device.DeviceName), "rtx") { - r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName) - result = append(result, device) - } + r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName) + result = append(result, device) } return result, nil From b0371a47db8718abba19b7542638e76e4d356b83 Mon Sep 17 00:00:00 2001 From: Stefan Majer Date: Tue, 15 Oct 2024 08:31:34 +0200 Subject: [PATCH 2/3] Only consider display class for gpus --- cmd/register/register.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/cmd/register/register.go b/cmd/register/register.go index 3f16f2b..0be6c09 100644 --- a/cmd/register/register.go +++ b/cmd/register/register.go @@ -245,8 +245,11 @@ func (r *Register) detectGPUs() (pci.Devices, error) { if !strings.Contains(strings.ToLower(device.VendorName), "nvidia") { continue } + r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName, "classname", device.ClassName) - r.log.Info("add gpu", "vendor", device.VendorName, "device", device.DeviceName) + if strings.ToLower(device.ClassName) != "display" { + continue + } result = append(result, device) } From c952b2b2e9300ab2054f0f41ca638c5fc6713e17 Mon Sep 17 00:00:00 2001 From: Stefan Majer Date: Tue, 15 Oct 2024 08:42:11 +0200 Subject: [PATCH 3/3] 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)