Debugging: use debug partition

eth-cscs · Oct 18, 2024 · d86670b · d86670b
1 parent f280d05
commit d86670b
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/ci/templates/base_image.yml.j2 b/ci/templates/base_image.yml.j2
@@ -6,6 +6,9 @@ build_job_build_image-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}:
     DOCKERFILE: "{{basedockerfile}}"
     PERSIST_IMAGE_NAME: "{{build_build_image_name_tag[arch]}}"
     DOCKER_BUILD_ARGS: '{{docker_build_args_base[arch]}}'
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}
 
 {% if arch == "cuda-aarch64" %}
 # We have multiarch image only for CUDA
@@ -24,6 +27,10 @@ test_job_build_image-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}:
   extends:
     - {{container_runner[arch]}}
   image: "{{registry_build_image_name_tag}}"
+{% if arch == "cuda-aarch64" %}
+  variables:
+    SLURM_PARTITION: debug
+{% endif %}
   script:
     - '[[ $(source /etc/os-release && echo $ID) == "{{os}}" ]]'
     - '[[ $(source /etc/os-release && echo $VERSION_ID) == "{{osver}}" ]]'

diff --git a/ci/templates/helper_image.yml.j2 b/ci/templates/helper_image.yml.j2
@@ -6,6 +6,9 @@ build_job_runtime_image-{{os}}{{osver}}-{{archstr[arch]}}:
     DOCKERFILE: "{{helperdockerfile}}"
     PERSIST_IMAGE_NAME: "{{build_runtime_image_name_tag[arch]}}"
     DOCKER_BUILD_ARGS: '{{docker_build_args_helper[arch]}}'
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}
 
 {% if arch == "cuda-aarch64" %}
 # We have multiarch image only for CUDA
@@ -24,6 +27,10 @@ test_job_runtime_image-{{os}}{{osver}}-{{archstr[arch]}}:
   extends:
     - {{container_runner[arch]}}
   image: "{{registry_runtime_image_name_tag}}"
+{% if arch == "cuda-aarch64" %}
+  variables:
+    SLURM_PARTITION: debug
+{% endif %}
   script:
     - '[[ $(source /etc/os-release && echo $ID) == "{{os}}" ]]'
     - '[[ $(source /etc/os-release && echo $VERSION_ID) == "{{osver}}" ]]'

diff --git a/ci/templates/test_image.yml.j2 b/ci/templates/test_image.yml.j2
@@ -8,12 +8,19 @@ build_job_test_image_libtree-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}
     DOCKERFILE: "test/Dockerfile.libtree"
     PERSIST_IMAGE_NAME: "{{registry_path}}/test-libtree:spack{{spackver}}-{{os}}{{osver}}-{{archstr[arch]}}"
     DOCKER_BUILD_ARGS: '["BUILDIMG={{registry_build_image_name_tag}}", "RUNTIMEIMG={{registry_runtime_image_name_tag}}", "TARGET={{spack_target}}"]'
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}
 
 test_job_test_image_libtree-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}:
   stage: test_test_image
   extends:
     - {{container_runner[arch]}}
   image: "{{registry_path}}/test-libtree:spack{{spackver}}-{{os}}{{osver}}-{{archstr[arch]}}"
+{% if arch == "cuda-aarch64" %}
+  variables:
+    SLURM_PARTITION: debug
+{% endif %}
   script:
     - which libtree
 
@@ -27,12 +34,19 @@ build_job_test_image_gromacs-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}
     DOCKERFILE: "test/Dockerfile.gromacs"
     PERSIST_IMAGE_NAME: "{{registry_path}}/test-gromacs:spack{{spackver}}-{{os}}{{osver}}-{{archstr[arch]}}"
     DOCKER_BUILD_ARGS: '["BUILDIMG={{registry_build_image_name_tag}}", "RUNTIMEIMG={{registry_runtime_image_name_tag}}", "TARGET={{spack_target}}"]'
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}
 
 test_job_test_image_gromacs-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}:
   stage: test_test_image
   extends:
     - {{container_runner[arch]}}
   image: "{{registry_path}}/test-gromacs:spack{{spackver}}-{{os}}{{osver}}-{{archstr[arch]}}"
+{% if arch == "cuda-aarch64" %}
+  variables:
+    SLURM_PARTITION: debug
+{% endif %}
   script:
     - which gmx_mpi
     - gmx_mpi -version
@@ -56,6 +70,9 @@ build_job_test_image_mpi-osu_bw-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch
     DOCKERFILE: "test/Dockerfile.mpi-osu_bw"
     PERSIST_IMAGE_NAME: "{{registry_path}}/test-mpi-osu_bw:spack{{spackver}}-{{os}}{{osver}}-{{archstr[arch]}}"
     DOCKER_BUILD_ARGS: '["BUILDIMG={{registry_build_image_name_tag}}", "RUNTIMEIMG={{registry_runtime_image_name_tag}}", "TARGET={{spack_target}}"]'
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}
 
 test_job_test_image_mpi-osu_bw-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]}}:
   stage: test_test_image
@@ -73,3 +90,6 @@ test_job_test_image_mpi-osu_bw-{{os}}{{osver}}-spack{{spackver}}-{{archstr[arch]
     SLURM_JOB_NUM_NODES: 2
     SLURM_NTASKS: 2
     SLURM_MPI_TYPE: pmi2
+{% if arch == "cuda-aarch64" %}
+    SLURM_PARTITION: debug
+{% endif %}