Skip to content

Commit e1cba98

Browse files
committed
Merge branch 'use_device_plugin_toleration' into 'master'
Apply device-plugin toleration, affinity, nodeSelector settings to device-plugin-validation too See merge request nvidia/kubernetes/gpu-operator!185
2 parents 7d3f8ba + 34cc777 commit e1cba98

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

pkg/controller/clusterpolicy/object_controls.go

+13
Original file line numberDiff line numberDiff line change
@@ -973,6 +973,19 @@ func TransformDevicePluginValidator(obj *v1.Pod, config *gpuv1.ClusterPolicySpec
973973
}
974974
}
975975

976+
// set node selector if specified for device-plugin
977+
if len(config.DevicePlugin.NodeSelector) > 0 {
978+
obj.Spec.NodeSelector = config.DevicePlugin.NodeSelector
979+
}
980+
// set node affinity if specified for device-plugin
981+
if config.DevicePlugin.Affinity != nil {
982+
obj.Spec.Affinity = config.DevicePlugin.Affinity
983+
}
984+
// set tolerations if specified for device-plugin
985+
if len(config.DevicePlugin.Tolerations) > 0 {
986+
obj.Spec.Tolerations = config.DevicePlugin.Tolerations
987+
}
988+
976989
return nil
977990
}
978991

0 commit comments

Comments
 (0)