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