| 1507 | } |
| 1508 | |
| 1509 | func transformDevicePluginCtrForCDI(container *corev1.Container, config *gpuv1.ClusterPolicySpec) { |
| 1510 | setContainerEnv(container, CDIEnabledEnvName, "true") |
| 1511 | setContainerEnv(container, DeviceListStrategyEnvName, "cdi-annotations,cdi-cri") |
| 1512 | setContainerEnv(container, CDIAnnotationPrefixEnvName, "cdi.k8s.io/") |
| 1513 | |
| 1514 | if config.Toolkit.IsEnabled() { |
| 1515 | setContainerEnv(container, NvidiaCDIHookPathEnvName, filepath.Join(config.Toolkit.InstallDir, "toolkit/nvidia-cdi-hook")) |
| 1516 | } |
| 1517 | } |
| 1518 | |
| 1519 | // TransformDevicePlugin transforms k8s-device-plugin daemonset with required config as per ClusterPolicy |
| 1520 | func TransformDevicePlugin(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n ClusterPolicyController) error { |