diff --git a/charts/mpijob/templates/mpijob.yaml b/charts/mpijob/templates/mpijob.yaml index 91832f56c..b4432b7ac 100644 --- a/charts/mpijob/templates/mpijob.yaml +++ b/charts/mpijob/templates/mpijob.yaml @@ -259,6 +259,9 @@ spec: {{- else}} nvidia.com/gpu: {{ $gpuCount | quote }} {{- end }} + {{- range $key, $value := .Values.devices }} + {{ $key }}: {{ $value }} + {{- end }} {{- end }} {{- if .Values.cpu }} cpu: {{ .Values.cpu | quote }}