From a5c114f5b9543d0a33929d8398eafdeb9da49115 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radim=20Janal=C3=ADk?= <radim.janalik@gmail.com>
Date: Fri, 18 Oct 2024 18:47:55 +0200
Subject: [PATCH] Debugging: use debug partition

---
 ci/templates/base_image.yml.j2   |  7 +++++++
 ci/templates/helper_image.yml.j2 |  7 +++++++
 ci/templates/test_image.yml.j2   | 20 ++++++++++++++++++++
 3 files changed, 34 insertions(+)

diff --git a/ci/templates/base_image.yml.j2 b/ci/templates/base_image.yml.j2
index fe31a78..bd7c2a0 100644
--- 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
index e87a9cb..e152551 100644
--- 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
index ea81f00..54e8416 100644
--- 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 %}