diff --git a/tools/container/toolkit/toolkit.go b/tools/container/toolkit/toolkit.go index 1f8d9b40..95ff0f8c 100644 --- a/tools/container/toolkit/toolkit.go +++ b/tools/container/toolkit/toolkit.go @@ -54,6 +54,8 @@ type options struct { ContainerRuntimeModesCdiDefaultKind string ContainerRuntimeModesCDIAnnotationPrefixes cli.StringSlice + ContainerRuntimeRuntimes cli.StringSlice + ContainerRuntimeHookSkipModeDetection bool ContainerCLIDebug string @@ -153,6 +155,11 @@ func main() { Destination: &opts.ContainerRuntimeModesCDIAnnotationPrefixes, EnvVars: []string{"NVIDIA_CONTAINER_RUNTIME_MODES_CDI_ANNOTATION_PREFIXES"}, }, + &cli.StringSliceFlag{ + Name: "nvidia-container-runtime.runtimes", + Destination: &opts.ContainerRuntimeRuntimes, + EnvVars: []string{"NVIDIA_CONTAINER_RUNTIME_RUNTIMES"}, + }, &cli.BoolFlag{ Name: "nvidia-container-runtime-hook.skip-mode-detection", Value: true, @@ -416,6 +423,7 @@ func installToolkitConfig(c *cli.Context, toolkitConfigPath string, nvidiaContai "nvidia-container-runtime.mode": opts.ContainerRuntimeMode, "nvidia-container-runtime.modes.cdi.annotation-prefixes": opts.ContainerRuntimeModesCDIAnnotationPrefixes, "nvidia-container-runtime.modes.cdi.default-kind": opts.ContainerRuntimeModesCdiDefaultKind, + "nvidia-container-runtime.runtimes": opts.ContainerRuntimeRuntimes, "nvidia-container-cli.debug": opts.ContainerCLIDebug, } for key, value := range optionalConfigValues {