UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-25 05:21:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support device name strategies for Tegra devices

Browse Source 
This change generates CDI specifications for Tegra devices
with the nvidia.com/gpu=0 name by default. The type-index
nameing strategy is also supported and will generate a device
with the name nvidia.com/gpu=gpu0.

The uuid naming strategy will raise an error if selected.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2023-07-18 14:12:00 +02:00
parent f07a0585fc
commit 9f46c34587
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/nvcdi/lib-csv.go
View File

@ -53,8 +53,13 @@ func (l *csvlib) GetAllDeviceSpecs() ([]specs.Device, error) {
return nil, fmt.Errorf("failed to create container edits for CSV files: %v", err) return nil, fmt.Errorf("failed to create container edits for CSV files: %v", err)
} }
name, err := l.deviceNamer.GetDeviceName(0, uuidUnsupported{})
if err != nil {
return nil, fmt.Errorf("failed to get device name: %v", err)
}
deviceSpec := specs.Device{ deviceSpec := specs.Device{
Name: "all", Name: name,
ContainerEdits: *e.ContainerEdits, ContainerEdits: *e.ContainerEdits,
} }
return []specs.Device{deviceSpec}, nil return []specs.Device{deviceSpec}, nil