UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-06-26 18:18:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'fix-device-symlinks' into 'main'

Browse Source 
Fix creation of device symlinks in /dev/char

See merge request nvidia/container-toolkit/container-toolkit!399
This commit is contained in:
Evan Lezar
2023-05-31 17:31:04 +00:00
parent fd48233c13 315f4adb8f
commit 528cbbb636
3 changed files with 43 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,12 +40,23 @@ func newAllPossible(logger *logrus.Logger, driverRoot string) (nodeLister, error
if err != nil {
return nil, fmt.Errorf("failed reading device majors: %v", err)
}
var requiredMajors []devices.Name
migCaps, err := nvcaps.NewMigCaps()
if err != nil {
return nil, fmt.Errorf("failed to read MIG caps: %v", err)
}
if migCaps == nil {
migCaps = make(nvcaps.MigCaps)
} else {
requiredMajors = append(requiredMajors, devices.NVIDIACaps)
}
requiredMajors = append(requiredMajors, devices.NVIDIAGPU, devices.NVIDIAUVM)
for _, name := range requiredMajors {
if !deviceMajors.Exists(name) {
return nil, fmt.Errorf("missing required device major %s", name)
}
}
l := allPossible{