diff --git a/.github/workflows/icon4py-test-model.yml b/.github/workflows/icon4py-test-model.yml index c4abf820cc..8bd9696242 100644 --- a/.github/workflows/icon4py-test-model.yml +++ b/.github/workflows/icon4py-test-model.yml @@ -14,7 +14,7 @@ jobs: strategy: matrix: python-version: ["3.10", "3.11"] - backend: ["embedded", "gtfn_cpu"] # TODO(): add dace-cpu? + backend: ["embedded", "dace_cpu", "gtfn_cpu"] component: ["advection", "diffusion", "dycore", "microphysics", "muphys", "driver", "common"] steps: - name: Checkout diff --git a/ci/dace.yml b/ci/dace.yml index 27db0f0679..052f2b4ef0 100644 --- a/ci/dace.yml +++ b/ci/dace.yml @@ -3,8 +3,6 @@ include: .test_model_stencils: stage: test - variables: - SLURM_TIMELIMIT: '00:10:00' script: - nox -s "test_model-3.10(stencils, $COMPONENT)" -- --backend=$BACKEND --grid=$GRID rules: @@ -12,6 +10,8 @@ include: variables: SLURM_TIMELIMIT: '00:30:00' - when: on_success + variables: + SLURM_TIMELIMIT: '00:15:00' parallel: matrix: - COMPONENT: [advection, diffusion, dycore, microphysics, common, driver] @@ -25,14 +25,10 @@ test_model_stencils_aarch64: .test_model_datatests: stage: test variables: - SLURM_TIMELIMIT: '00:30:00' + NUM_PROCESSES: 8 + SLURM_TIMELIMIT: '00:45:00' script: - nox -s "test_model-3.10(datatest, $COMPONENT)" -- --backend=$BACKEND --level=$LEVEL - rules: - - if: $COMPONENT == 'common' && $BACKEND == 'dace_gpu' && $LEVEL == 'integration' - variables: - SLURM_TIMELIMIT: '01:30:00' - - when: on_success parallel: matrix: - COMPONENT: [advection, diffusion, dycore, microphysics, common, driver] diff --git a/ci/default.yml b/ci/default.yml index 9d40e07ffc..5fe2b10db8 100644 --- a/ci/default.yml +++ b/ci/default.yml @@ -37,10 +37,16 @@ test_tools_datatests_aarch64: script: - nox -s "test_model-3.10(datatest, $COMPONENT)" -- --backend=$BACKEND --level=$LEVEL rules: + - if: $BACKEND == 'dace_gpu' && $COMPONENT != 'diffusion' && $COMPONENT != 'dycore' + when: never # run only in daily CI, to save compute resources - if: $COMPONENT == 'common' && $LEVEL == 'integration' variables: - NUM_PROCESSES: 2 + NUM_PROCESSES: 1 SLURM_TIMELIMIT: '00:30:00' + - if: $BACKEND == 'dace_gpu' + variables: + NUM_PROCESSES: 8 + SLURM_TIMELIMIT: '00:45:00' - if: $BACKEND == 'embedded' variables: SLURM_TIMELIMIT: '00:15:00' @@ -50,7 +56,7 @@ test_tools_datatests_aarch64: parallel: matrix: - COMPONENT: [advection, diffusion, dycore, microphysics, common, driver] - BACKEND: [embedded, gtfn_cpu, gtfn_gpu] + BACKEND: [embedded, dace_gpu, gtfn_cpu, gtfn_gpu] LEVEL: [integration] # test_model_datatests_x86_64: # extends: [.test_model_datatests, .test_template_x86_64]