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

Add csv mode to nvidia-ctk cdi generate command

Browse Source 
This chagne allows the csv mode option to specified in the
nvidia-ctk cdi generate command and adds a --csv.file option
that can be repeated to specify the CSV files to be processed.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2023-05-10 16:28:59 +02:00
parent 418e4014e6
commit e30fd0f4ad
3 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/nvcdi/lib.go
View File

@@ -19,6 +19,7 @@ package nvcdi
import (
"fmt"
"github.com/NVIDIA/nvidia-container-toolkit/internal/discover/csv"
"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec"
"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform"
"github.com/sirupsen/logrus"
@@ -84,11 +85,7 @@ func New(opts ...Option) (Interface, error) {
switch l.resolveMode() {
case ModeCSV:
if len(l.csvFiles) == 0 {
l.csvFiles = []string{
"/etc/nvidia-container-runtime/host-files-for-container.d/l4t.csv",
"/etc/nvidia-container-runtime/host-files-for-container.d/drivers.csv",
"/etc/nvidia-container-runtime/host-files-for-container.d/devices.csv",
}
l.csvFiles = csv.DefaultFileList()
}
lib = (*csvlib)(l)
case ModeManagement: