set gpu jobs using a configmap (#1525)

Signed-off-by: Paul S. Schweigert <[email protected]>
Qiskit · Oct 23, 2024 · fbd66dc · fbd66dc
1 parent b684b26
commit fbd66dc
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/charts/qiskit-serverless/charts/gateway/templates/deployment.yaml b/charts/qiskit-serverless/charts/gateway/templates/deployment.yaml
@@ -239,6 +239,9 @@ spec:
         - name: ray-cluster-template
           configMap:
             name: rayclustertemplate
+        - name: gpu-jobs
+          configMap:
+            name: gpujobs
       serviceAccountName: {{ include "gateway.serviceAccountName" . }}
       securityContext:
         {{- toYaml .Values.podSecurityContext | nindent 8 }}
@@ -277,6 +280,8 @@ spec:
               name: gateway-pv-storage
             - mountPath: "/tmp/templates/"
               name: ray-cluster-template
+            - mountPath: "/tmp/gpujobs/"
+              name: gpu-jobs
           resources:
             {{- toYaml .Values.scheduler.resources | nindent 12 }}
           env:

diff --git a/charts/qiskit-serverless/charts/gateway/templates/gpu-jobs.yaml b/charts/qiskit-serverless/charts/gateway/templates/gpu-jobs.yaml
@@ -0,0 +1,9 @@
+apiVersion: v1
+kind: ConfigMap
+metadata:
+  name: gpujobs
+data:
+  gpu-jobs.json: |
+    {
+        "gpu-functions": {}
+    }
diff --git a/charts/qiskit-serverless/values.yaml b/charts/qiskit-serverless/values.yaml
@@ -62,6 +62,7 @@ gateway:
       nodeImage: "icr.io/quantum-public/qiskit-serverless/ray-node:0.17.1"
       opensslImage: registry.access.redhat.com/ubi8/openssl:8.8-9
       kubectlImage: alpine/k8s:1.29.2@sha256:a51aa37f0a34ff827c7f2f9cb7f6fbb8f0e290fa625341be14c2fcc4b1880f60
+      gpuJobsConfig: "/tmp/gpujobs/gpu-jobs.json"
     limits:
       maxJobsPerUser: 2
       maxComputeResources: 4