diff --git a/src/Jobs_Templete/pod.yaml.template b/src/Jobs_Templete/pod.yaml.template index b5002446c..4bb58c510 100755 --- a/src/Jobs_Templete/pod.yaml.template +++ b/src/Jobs_Templete/pod.yaml.template @@ -40,6 +40,14 @@ metadata: spec: nodeSelector: worker: active + {% if job["nodeSelector"]|length > 0 %} + {% for key, value in job["nodeSelector"].items() %} + {{key}}: {{value}} + {% endfor %} + {% endif %} + {% if job["fragmentGpuJob"] %} + FragmentGPUJob: active + {% endif %} affinity: podAffinity: {% if jobRole == "ps" %} @@ -124,14 +132,6 @@ spec: - "1" topologyKey: "kubernetes.io/hostname" {% endif %} - {% if job["nodeSelector"]|length > 0 %} - {% for key, value in job["nodeSelector"].items() %} - {{key}}: {{value}} - {% endfor %} - {% endif %} - {% if job["fragmentGpuJob"] %} - FragmentGPUJob: active - {% endif %} {% if job["dnsPolicy"] %} dnsPolicy: {{ job["dnsPolicy" ]}} {% endif %}