Support nodeSelector and tolerations in ray clusters deployed by helm (…

…ray-project#181) Co-authored-by: Eugene Brodsky <[email protected]>
lowang-bh · Mar 10, 2022 · 038b4a6 · 038b4a6
1 parent 6b38c13
commit 038b4a6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -26,6 +26,8 @@ spec:
             {{- toYaml .Values.head.containerEnv | nindent 14}}
         volumes: {{- toYaml .Values.head.volumes | nindent 10 }}
         affinity: {{- toYaml .Values.head.affinity | nindent 10 }}
+        tolerations: {{- toYaml .Values.head.tolerations | nindent 10 }}
+        nodeSelector: {{- toYaml .Values.head.nodeSelector | nindent 10 }}
       metadata:
         annotations: {{- toYaml .Values.head.annotations | nindent 10 }}
         labels:
@@ -62,6 +64,8 @@ spec:
             ports: {{- toYaml .Values.worker.ports | nindent 14}}
         volumes: {{- toYaml .Values.worker.volumes | nindent 10 }}
         affinity: {{- toYaml .Values.worker.affinity | nindent 10 }}
+        tolerations: {{- toYaml .Values.worker.tolerations | nindent 10 }}
+        nodeSelector: {{- toYaml .Values.worker.nodeSelector | nindent 10 }}
       metadata:
         annotations: {{- toYaml .Values.worker.annotations | nindent 10 }}
         labels:

diff --git a/helm-chart/ray-cluster/values.yaml b/helm-chart/ray-cluster/values.yaml
@@ -34,6 +34,9 @@ head:
     requests:
        cpu: 200m
   annotations: {}
+  nodeSelector: {}
+  tolerations: []
+  affinity: {}
   volumes:
     - name: log-volume
       emptyDir: { }