TransformVGPUManager transforms NVIDIA vGPU Manager daemonset with required config as per ClusterPolicy
(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n ClusterPolicyController)
| 1085 | |
| 1086 | // TransformVGPUManager transforms NVIDIA vGPU Manager daemonset with required config as per ClusterPolicy |
| 1087 | func TransformVGPUManager(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n ClusterPolicyController) error { |
| 1088 | // update k8s-driver-manager initContainer |
| 1089 | err := transformDriverManagerInitContainer(obj, &config.VGPUManager.DriverManager, nil) |
| 1090 | if err != nil { |
| 1091 | return fmt.Errorf("failed to transform k8s-driver-manager initContainer for vGPU Manager: %v", err) |
| 1092 | } |
| 1093 | |
| 1094 | // update nvidia-vgpu-manager container |
| 1095 | err = transformVGPUManagerContainer(obj, config, n) |
| 1096 | if err != nil { |
| 1097 | return fmt.Errorf("failed to transform vGPU Manager container: %v", err) |
| 1098 | } |
| 1099 | |
| 1100 | // update OpenShift Driver Toolkit sidecar container |
| 1101 | err = transformOpenShiftDriverToolkitContainer(obj, config, n, "nvidia-vgpu-manager-ctr") |
| 1102 | if err != nil { |
| 1103 | return fmt.Errorf("failed to transform the Driver Toolkit container: %s", err) |
| 1104 | } |
| 1105 | |
| 1106 | // set hostNetwork for vgpu-manager if specified |
| 1107 | applyHostNetworkConfig(&obj.Spec.Template.Spec, config.VGPUManager.HostNetwork) |
| 1108 | |
| 1109 | return nil |
| 1110 | } |
| 1111 | |
| 1112 | // applyOCPProxySpec applies proxy settings to podSpec |
| 1113 | func applyOCPProxySpec(n ClusterPolicyController, podSpec *corev1.PodSpec) error { |