Skip to content

Commit f3b730c

Browse files
committed
[no-relnote] Use FilterForGPU to create all devchar symlinks
Signed-off-by: Evan Lezar <[email protected]>
1 parent 1cfaef4 commit f3b730c

File tree

1 file changed

+3
-15
lines changed

1 file changed

+3
-15
lines changed

cmd/nvidia-ctk/system/create-dev-char-symlinks/all.go

+3-15
Original file line numberDiff line numberDiff line change
@@ -145,21 +145,9 @@ func (m allPossible) getGPUDeviceNodes(gpu int) []deviceNode {
145145
// getNVCapDeviceNodes generates a list of cap device nodes for a given GPU.
146146
func (m allPossible) getNVCapDeviceNodes(gpu int) []deviceNode {
147147
var selectedCapMinors []nvcaps.MigMinor
148-
for gi := 0; ; gi++ {
149-
giCap := nvcaps.NewGPUInstanceCap(gpu, gi)
150-
giMinor, exist := m.migCaps[giCap]
151-
if !exist {
152-
break
153-
}
154-
selectedCapMinors = append(selectedCapMinors, giMinor)
155-
for ci := 0; ; ci++ {
156-
ciCap := nvcaps.NewComputeInstanceCap(gpu, gi, ci)
157-
ciMinor, exist := m.migCaps[ciCap]
158-
if !exist {
159-
break
160-
}
161-
selectedCapMinors = append(selectedCapMinors, ciMinor)
162-
}
148+
149+
for _, capMinors := range m.migCaps.FilterForGPU(nvcaps.Index(gpu)) {
150+
selectedCapMinors = append(selectedCapMinors, capMinors)
163151
}
164152

165153
var deviceNodes []deviceNode

0 commit comments

Comments
 (0)