File tree 1 file changed +3
-15
lines changed
cmd/nvidia-ctk/system/create-dev-char-symlinks
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 {
145
145
// getNVCapDeviceNodes generates a list of cap device nodes for a given GPU.
146
146
func (m allPossible ) getNVCapDeviceNodes (gpu int ) []deviceNode {
147
147
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 )
163
151
}
164
152
165
153
var deviceNodes []deviceNode
You can’t perform that action at this time.
0 commit comments