diff --git a/pkg/controller/v1beta1/benchmark/controller.go b/pkg/controller/v1beta1/benchmark/controller.go index 318abd37..99d3e364 100644 --- a/pkg/controller/v1beta1/benchmark/controller.go +++ b/pkg/controller/v1beta1/benchmark/controller.go @@ -304,18 +304,8 @@ func (r *BenchmarkJobReconciler) createPodSpec(benchmarkJob *v1beta1.BenchmarkJo // If no overrides, return default spec if benchmarkJob.Spec.PodOverride == nil { return &v1.PodSpec{ - Containers: []v1.Container{defaultContainer}, - Volumes: volumes, - NodeSelector: map[string]string{ - "nvidia.com/gpu": "true", - }, - Tolerations: []v1.Toleration{ - { - Key: "nvidia.com/gpu", - Operator: v1.TolerationOpExists, - Effect: v1.TaintEffectNoSchedule, - }, - }, + Containers: []v1.Container{defaultContainer}, + Volumes: volumes, RestartPolicy: v1.RestartPolicyNever, }, nil } @@ -354,18 +344,8 @@ func (r *BenchmarkJobReconciler) createPodSpec(benchmarkJob *v1beta1.BenchmarkJo // Create and merge pod specs defaultPodSpec := &v1.PodSpec{ - Containers: []v1.Container{mergedContainer}, - Volumes: volumes, - NodeSelector: map[string]string{ - "nvidia.com/gpu": "true", - }, - Tolerations: []v1.Toleration{ - { - Key: "nvidia.com/gpu", - Operator: v1.TolerationOpExists, - Effect: v1.TaintEffectNoSchedule, - }, - }, + Containers: []v1.Container{mergedContainer}, + Volumes: volumes, RestartPolicy: v1.RestartPolicyNever, }