Preparing runners

.gitlab-ci.yml

@@ -29,9 +29,17 @@ variables:
         slurm_options: ["-c 36"]
         stack: ["pinot-noir"]
       # - environment: [izar]
-      #   slurm_options: ['-c 40 --gres gpu:2']
+      #   slurm_options: ['-c 40 --gpus 2']
       #   apptainer_options: ['-nv']
-
+      #   stack: ["pinot-noir"]
+      # - environment: [kuma-l40s]
+      #   slurm_options: ['-c 64 --gres gpu:8']
+      #   apptainer_options: ['-nv']
+      #   stack: ["pinot-noir"]
+      # - environment: [kuma-h100]
+      #   slurm_options: ['-c 64 --gres gpu:4']
+      #   apptainer_options: ['-nv']
+      #   stack: ["pinot-noir"]
   tags:
     - ${environment}
     - stack

stacks/pinot-noir/packages_izar.yaml

+packages:
+  all:
+    target: ["cannonlake"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=70
+        when: '+cuda'
+
+      - '+cuda'

stacks/pinot-noir/packages_kuma_h100.yaml

+packages:
+  all:
+    target: ["zen4"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=70
+        when: '+cuda'
+
+      - '+cuda'

stacks/pinot-noir/packages_kuma_l40s.yaml

+packages:
+  all:
+    target: ["zen4"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=89
+        when: '+cuda'
+
+      - '+cuda'