From b3136cfa206c20fbdb4cab16db0056aa8effee17 Mon Sep 17 00:00:00 2001
From: Nicolas Richart <networms@gmail.ch>
Date: Mon, 17 Jun 2024 23:09:14 +0200
Subject: [PATCH] Preparing runners

---
 .gitlab-ci.yml                            | 12 ++++++++++--
 stacks/pinot-noir/packages_izar.yaml      | 11 +++++++++++
 stacks/pinot-noir/packages_kuma_h100.yaml | 11 +++++++++++
 stacks/pinot-noir/packages_kuma_l40s.yaml | 11 +++++++++++
 4 files changed, 43 insertions(+), 2 deletions(-)
 create mode 100644 stacks/pinot-noir/packages_izar.yaml
 create mode 100644 stacks/pinot-noir/packages_kuma_h100.yaml
 create mode 100644 stacks/pinot-noir/packages_kuma_l40s.yaml

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 6a8dc5d..7a32887 100644
--- a/.gitlab-ci.yml
+++ b/.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
diff --git a/stacks/pinot-noir/packages_izar.yaml b/stacks/pinot-noir/packages_izar.yaml
new file mode 100644
index 0000000..9ce65d9
--- /dev/null
+++ b/stacks/pinot-noir/packages_izar.yaml
@@ -0,0 +1,11 @@
+packages:
+  all:
+    target: ["cannonlake"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=70
+        when: '+cuda'
+
+      - '+cuda'
diff --git a/stacks/pinot-noir/packages_kuma_h100.yaml b/stacks/pinot-noir/packages_kuma_h100.yaml
new file mode 100644
index 0000000..98433e3
--- /dev/null
+++ b/stacks/pinot-noir/packages_kuma_h100.yaml
@@ -0,0 +1,11 @@
+packages:
+  all:
+    target: ["zen4"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=70
+        when: '+cuda'
+
+      - '+cuda'
diff --git a/stacks/pinot-noir/packages_kuma_l40s.yaml b/stacks/pinot-noir/packages_kuma_l40s.yaml
new file mode 100644
index 0000000..ad6b8fb
--- /dev/null
+++ b/stacks/pinot-noir/packages_kuma_l40s.yaml
@@ -0,0 +1,11 @@
+packages:
+  all:
+    target: ["zen4"]
+    require:
+      - spec: ^openmpi +cuda
+        when: '%gcc ^mpi'
+
+      - spec: cuda_arch=89
+        when: '+cuda'
+
+      - '+cuda'
-- 
GitLab