@@ -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