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

Allow CDI options to be set by envvars

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2023-03-09 12:25:05 +02:00
parent 86dd046c7c
commit eb2aec9da8
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/container/toolkit/toolkit.go
View File

@ -168,12 +168,14 @@ func main() {
Usage: "the directory where the CDI output files are to be written. If this is set to '', no CDI specification is generated.", Usage: "the directory where the CDI output files are to be written. If this is set to '', no CDI specification is generated.",
Value: "/var/run/cdi", Value: "/var/run/cdi",
Destination: &opts.cdiOutputDir, Destination: &opts.cdiOutputDir,
EnvVars: []string{"CDI_OUTPUT_DIR"},
}, },
&cli.StringFlag{ &cli.StringFlag{
Name: "cdi-kind", Name: "cdi-kind",
Usage: "the vendor string to use for the generated CDI specification", Usage: "the vendor string to use for the generated CDI specification",
Value: "management.nvidia.com/gpu", Value: "management.nvidia.com/gpu",
Destination: &opts.cdiKind, Destination: &opts.cdiKind,
EnvVars: []string{"CDI_KIND"},
}, },
} }