File tree Expand file tree Collapse file tree 1 file changed +3
-15
lines changed
cmd/nvidia-ctk/system/create-dev-char-symlinks Expand file tree Collapse file tree 1 file changed +3
-15
lines changed Original file line number Diff line number Diff 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.
146146func (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
You can’t perform that action at this time.
0 commit comments