From f3b730c805b6b0bcca62efd5f76bd083e450c0e4 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Wed, 12 Mar 2025 12:42:37 +0200 Subject: [PATCH] [no-relnote] Use FilterForGPU to create all devchar symlinks Signed-off-by: Evan Lezar --- .../system/create-dev-char-symlinks/all.go | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/cmd/nvidia-ctk/system/create-dev-char-symlinks/all.go b/cmd/nvidia-ctk/system/create-dev-char-symlinks/all.go index cafb8f9c..62aaf32d 100644 --- a/cmd/nvidia-ctk/system/create-dev-char-symlinks/all.go +++ b/cmd/nvidia-ctk/system/create-dev-char-symlinks/all.go @@ -145,21 +145,9 @@ func (m allPossible) getGPUDeviceNodes(gpu int) []deviceNode { // getNVCapDeviceNodes generates a list of cap device nodes for a given GPU. func (m allPossible) getNVCapDeviceNodes(gpu int) []deviceNode { var selectedCapMinors []nvcaps.MigMinor - for gi := 0; ; gi++ { - giCap := nvcaps.NewGPUInstanceCap(gpu, gi) - giMinor, exist := m.migCaps[giCap] - if !exist { - break - } - selectedCapMinors = append(selectedCapMinors, giMinor) - for ci := 0; ; ci++ { - ciCap := nvcaps.NewComputeInstanceCap(gpu, gi, ci) - ciMinor, exist := m.migCaps[ciCap] - if !exist { - break - } - selectedCapMinors = append(selectedCapMinors, ciMinor) - } + + for _, capMinors := range m.migCaps.FilterForGPU(nvcaps.Index(gpu)) { + selectedCapMinors = append(selectedCapMinors, capMinors) } var deviceNodes []deviceNode