Skip to content

Commit 8eef7e5

Browse files
author
Evan Lezar
committed
Merge branch 'add-runtimes' into 'main'
Add nvidia-container-runtime.runtimes config option See merge request nvidia/container-toolkit/container-toolkit!364
2 parents ee5be5e + be0cc9d commit 8eef7e5

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tools/container/toolkit/toolkit.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ type options struct {
5454
ContainerRuntimeModesCdiDefaultKind string
5555
ContainerRuntimeModesCDIAnnotationPrefixes cli.StringSlice
5656

57+
ContainerRuntimeRuntimes cli.StringSlice
58+
5759
ContainerRuntimeHookSkipModeDetection bool
5860

5961
ContainerCLIDebug string
@@ -153,6 +155,11 @@ func main() {
153155
Destination: &opts.ContainerRuntimeModesCDIAnnotationPrefixes,
154156
EnvVars: []string{"NVIDIA_CONTAINER_RUNTIME_MODES_CDI_ANNOTATION_PREFIXES"},
155157
},
158+
&cli.StringSliceFlag{
159+
Name: "nvidia-container-runtime.runtimes",
160+
Destination: &opts.ContainerRuntimeRuntimes,
161+
EnvVars: []string{"NVIDIA_CONTAINER_RUNTIME_RUNTIMES"},
162+
},
156163
&cli.BoolFlag{
157164
Name: "nvidia-container-runtime-hook.skip-mode-detection",
158165
Value: true,
@@ -416,6 +423,7 @@ func installToolkitConfig(c *cli.Context, toolkitConfigPath string, nvidiaContai
416423
"nvidia-container-runtime.mode": opts.ContainerRuntimeMode,
417424
"nvidia-container-runtime.modes.cdi.annotation-prefixes": opts.ContainerRuntimeModesCDIAnnotationPrefixes,
418425
"nvidia-container-runtime.modes.cdi.default-kind": opts.ContainerRuntimeModesCdiDefaultKind,
426+
"nvidia-container-runtime.runtimes": opts.ContainerRuntimeRuntimes,
419427
"nvidia-container-cli.debug": opts.ContainerCLIDebug,
420428
}
421429
for key, value := range optionalConfigValues {

0 commit comments

Comments
 (0)