nvidia-container-toolkit: support both cdi and legacy stacks

Update the NVIDIA Container Toolkit configuration template to dynamically
select the mode based off the configurations used in the Kubernetes Device
Plugin.

Signed-off-by: Jingwei Wang <[email protected]>

Author: sky1122

packages/nvidia-container-toolkit/nvidia-container-toolkit-config-k8s

@@ -1,6 +1,6 @@
 [required-extensions]
 nvidia-container-runtime = "v1"
-std = { version = "v1", helpers = ["default"] }
+std = { version = "v1", helpers = ["default", "is_array", "is_string"] }
 
 +++
 ### generated from the template file ###
@@ -12,3 +12,22 @@ root = "/"
 path = "/usr/bin/nvidia-container-cli"
 environment = []
 ldconfig = "@/sbin/ldconfig"
+
+[nvidia-container-runtime]
+{{#if settings.kubelet-device-plugins.nvidia.device-list-strategy}}
+{{#if (is_array settings.kubelet-device-plugins.nvidia.device-list-strategy) }}         {{! Begin is array}}
+{{#if (eq settings.kubelet-device-plugins.nvidia.device-list-strategy.[0] "cdi-cri") }}
+mode="cdi"
+{{else}}
+mode="legacy"
+{{/if}}                                                                                 {{! End is array}}
+{{if (is_string settings.kubelet-device-plugins.nvidia.device-list-strategy) }}    {{! Begin is string}}
+{{#if (eq settings.kubelet-device-plugins.nvidia.device-list-strategy "cdi-cri") }}
+mode="cdi"
+{{else}}
+mode="legacy"
+{{/if}}
+{{/if}}                                                                                 {{! End is string}}
+{{else}}
+mode="legacy"
+{{/if}}

packages/nvidia-k8s-device-plugin/nvidia-k8s-device-plugin-conf

@@ -18,7 +18,7 @@ flags:
   nvidiaDriverRoot: "/"
   plugin:
     passDeviceSpecs: {{default true settings.kubelet-device-plugins.nvidia.pass-device-specs}}
-    deviceListStrategy: {{default "volume-mounts" settings.kubelet-device-plugins.nvidia.device-list-strategy}}
+    deviceListStrategy: {{#if settings.kubelet-device-plugins.nvidia.device-list-strategy}}{{settings.kubelet-device-plugins.nvidia.device-list-strategy}}{{else}}"volume-mounts"{{/if}}
     deviceIDStrategy: {{default "index" settings.kubelet-device-plugins.nvidia.device-id-strategy}}
     containerDriverRoot: "/"
 {{#if settings.kubelet-device-plugins.nvidia.device-sharing-strategy}}