@@ -54,6 +54,8 @@ type options struct {
54
54
ContainerRuntimeModesCdiDefaultKind string
55
55
ContainerRuntimeModesCDIAnnotationPrefixes cli.StringSlice
56
56
57
+ ContainerRuntimeRuntimes cli.StringSlice
58
+
57
59
ContainerRuntimeHookSkipModeDetection bool
58
60
59
61
ContainerCLIDebug string
@@ -153,6 +155,11 @@ func main() {
153
155
Destination : & opts .ContainerRuntimeModesCDIAnnotationPrefixes ,
154
156
EnvVars : []string {"NVIDIA_CONTAINER_RUNTIME_MODES_CDI_ANNOTATION_PREFIXES" },
155
157
},
158
+ & cli.StringSliceFlag {
159
+ Name : "nvidia-container-runtime.runtimes" ,
160
+ Destination : & opts .ContainerRuntimeRuntimes ,
161
+ EnvVars : []string {"NVIDIA_CONTAINER_RUNTIME_RUNTIMES" },
162
+ },
156
163
& cli.BoolFlag {
157
164
Name : "nvidia-container-runtime-hook.skip-mode-detection" ,
158
165
Value : true ,
@@ -416,6 +423,7 @@ func installToolkitConfig(c *cli.Context, toolkitConfigPath string, nvidiaContai
416
423
"nvidia-container-runtime.mode" : opts .ContainerRuntimeMode ,
417
424
"nvidia-container-runtime.modes.cdi.annotation-prefixes" : opts .ContainerRuntimeModesCDIAnnotationPrefixes ,
418
425
"nvidia-container-runtime.modes.cdi.default-kind" : opts .ContainerRuntimeModesCdiDefaultKind ,
426
+ "nvidia-container-runtime.runtimes" : opts .ContainerRuntimeRuntimes ,
419
427
"nvidia-container-cli.debug" : opts .ContainerCLIDebug ,
420
428
}
421
429
for key , value := range optionalConfigValues {
0 commit comments