diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 6a8dc5df1e9ee1da2c3c7ba8b682449b070d47da..7a32887e234458aa919503d93184bfe790a31321 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 0000000000000000000000000000000000000000..9ce65d95e200925fe7b5bd50d0f76c18b2f33097 --- /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 0000000000000000000000000000000000000000..98433e3b2c1f32a96105e08e00b299ac1312fafe --- /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 0000000000000000000000000000000000000000..ad6b8fb44336cd6730bb956c5d8489ff7f0acf0b --- /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'