Merge pull request #1441 from porter-dev/ym/gpu_enabled_jobs

feat: allow job umbrella charts to run on GPU

Author: yosefmih

applications/job/templates/cronjob.yaml

@@ -169,6 +169,11 @@ spec:
                 requests:
                   cpu: {{ .Values.resources.requests.cpu }}
                   memory: {{ .Values.resources.requests.memory }}
+              {{ if .Values.resources.requests }}
+              {{ if .Values.resources.requests.nvidiaGpu }}
+                  nvidia.com/gpu: {{ .Values.resources.requests.nvidiaGpu }}
+              {{ end }}
+              {{ end }}
                 limits:
                 {{- if .Values.resources.setCPULimits }}
                   {{- if .Values.resources.limits.cpu }}
@@ -182,6 +187,9 @@ spec:
                 {{- else }}
                   memory: {{ .Values.resources.requests.memory }}
                 {{- end }}
+                {{- if .Values.resources.limits.nvidiaGpu }}
+                  nvidia.com/gpu: {{ .Values.resources.limits.nvidiaGpu }}
+                {{- end }}
             - name: sidecar
               image: ghcr.io/porter-dev/job_sidecar_container:31e471f4d
               imagePullPolicy: Always
@@ -262,6 +270,13 @@ spec:
               value: {{ $nodeGroup.id | quote }}
               effect: "NoSchedule"
           {{- end }}
+          # nvidia.com/gpu toleration is automatically injected in EKS/GKE, but not in AKS 
+          # since we want to be cloud agnostic, we just add it ourselves wherever we need it
+          {{- if .Values.resources.requests.nvidiaGpu}}
+            - key: "nvidia.com/gpu"
+              operator: "Exists"
+              effect: "NoSchedule"
+          {{- end }}
       backoffLimit: 0
       {{- if (.Values.sidecar.timeout) }}
       activeDeadlineSeconds: {{ .Values.sidecar.timeout }}