diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml index 7f26c67a635..014a22968ee 100644 --- 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}}