add priority and priorityClassName to ray-cluster template (#2171)

Signed-off-by: Rong Rong <rongr@apache.org>
ray-project · Jun 5, 2024 · dc8be94 · dc8be94
1 parent c147ad2
commit dc8be94
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -90,6 +90,12 @@ spec:
         volumes: {{- toYaml .Values.head.volumes | nindent 10 }}
         {{- end }}
         affinity: {{- toYaml .Values.head.affinity | nindent 10 }}
+        {{ if .Values.head.priorityClassName }}
+        priorityClassName: {{- toYaml .Values.head.priorityClassName | nindent 10 }}
+        {{- end }}
+        {{ if .Values.head.priority }}
+        priority: {{- toYaml .Values.head.priority | nindent 10 }}
+        {{- end }}
         tolerations: {{- toYaml .Values.head.tolerations | nindent 10 }}
         nodeSelector: {{- toYaml .Values.head.nodeSelector | nindent 10 }}
         {{- with .Values.head.podSecurityContext }}
@@ -179,6 +185,12 @@ spec:
         volumes: {{- toYaml $values.volumes | nindent 10 }}
         {{- end }}
         affinity: {{- toYaml $values.affinity | nindent 10 }}
+        {{ if $values.priorityClassName }}
+        priorityClassName: {{- toYaml $values.priorityClassName | nindent 10 }}
+        {{- end }}
+        {{ if $values.priority }}
+        priority: {{- toYaml $values.priority | nindent 10 }}
+        {{- end }}
         tolerations: {{- toYaml $values.tolerations | nindent 10 }}
         nodeSelector: {{- toYaml $values.nodeSelector | nindent 10 }}
         {{- with $values.podSecurityContext }}
@@ -268,6 +280,12 @@ spec:
         volumes: {{- toYaml .Values.worker.volumes | nindent 10 }}
         {{- end }}
         affinity: {{- toYaml .Values.worker.affinity | nindent 10 }}
+        {{ if .Values.worker.priorityClassName }}
+        priorityClassName: {{- toYaml .Values.worker.priorityClassName | nindent 10 }}
+        {{- end }}
+        {{ if .Values.worker.priority }}
+        priority: {{- toYaml .Values.worker.priority | nindent 10 }}
+        {{- end }}
         tolerations: {{- toYaml .Values.worker.tolerations | nindent 10 }}
         nodeSelector: {{- toYaml .Values.worker.nodeSelector | nindent 10 }}
         {{- with .Values.worker.podSecurityContext}}