Add support for RUNTIME=none to disable runtime config in the nvidia-container-toolkit

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

controllers/object_controls.go

@@ -1314,14 +1314,19 @@ func TransformToolkit(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n
 		}
 	}
 
-	if len(config.Toolkit.Env) > 0 {
-		for _, env := range config.Toolkit.Env {
-			setContainerEnv(toolkitMainContainer, env.Name, env.Value)
+	// configure runtime
+	runtime := n.runtime.String()
+	// Update the main container environment from the user-specified values.
+	for _, env := range config.Toolkit.Env {
+		if env.Name == "RUNTIME" {
+			// If the user has specified the runtime, we overide the detected
+			// value.
+			// TODO: Add logging.
+			runtime = env.Value
 		}
+		setContainerEnv(toolkitMainContainer, env.Name, env.Value)
 	}
 
-	// configure runtime
-	runtime := n.runtime.String()
 	err = transformForRuntime(obj, config, runtime, toolkitMainContainer)
 	if err != nil {
 		return fmt.Errorf("error transforming toolkit daemonset : %w", err)
@@ -1332,6 +1337,11 @@ func TransformToolkit(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n
 
 func transformForRuntime(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, runtime string, container *corev1.Container) error {
 	setContainerEnv(container, "RUNTIME", runtime)
+	// If the user has explicitly requested 'none' as a runtime, we make no
+	// additional changes to the container.
+	if runtime == "none" {
+		return nil
+	}
 
 	if runtime == gpuv1.Containerd.String() {
 		// Set the runtime class name that is to be configured for containerd