[Rest Server] Add node selector for user jobs in k8s (#3694)

Add node selector for user jobs in k8s, only run user pods on pai-worker nodes.
microsoft · Oct 9, 2019 · cb6ca62 · cb6ca62
1 parent 1016a98
commit cb6ca62
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/rest-server/src/models/v2/job/k8s.js b/src/rest-server/src/models/v2/job/k8s.js
@@ -405,6 +405,23 @@ const generateTaskRole = (taskRole, labels, config) => {
               },
             },
           ],
+          affinity: {
+            nodeAffinity: {
+              requiredDuringSchedulingIgnoredDuringExecution: {
+                nodeSelectorTerms: [
+                  {
+                    matchExpressions: [
+                      {
+                        key: 'pai-worker',
+                        operator: 'In',
+                        values: ['true'],
+                      },
+                    ],
+                  },
+                ],
+              },
+            },
+          },
           imagePullSecrets: [
             {
               name: launcherConfig.runtimeImagePullSecrets,