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 %}