From 767d0aa09fa5fd8d70f072622fa580dfd5fe800e Mon Sep 17 00:00:00 2001 From: Gabriele Bozzola Date: Mon, 15 Apr 2024 11:10:50 -0700 Subject: [PATCH] Fix pipeline --- .buildkite/pipeline.yml | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index 1896bcd489..05716ea127 100755 --- a/.buildkite/pipeline.yml +++ b/.buildkite/pipeline.yml @@ -243,8 +243,14 @@ steps: agents: slurm_ntasks: 4 - - label: "Unit: distributed remapping" - key: distributed_remapping + - label: "Unit: distributed remapping (1 process)" + key: distributed_remapping_1proc + command: "julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl" + env: + CLIMACOMMS_DEVICE: "CPU" + + - label: "Unit: distributed remapping (2 processes)" + key: distributed_remapping_2procs command: "srun julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl" env: CLIMACOMMS_CONTEXT: "MPI" @@ -252,21 +258,23 @@ steps: agents: slurm_ntasks: 2 - - label: "Unit: distributed remapping (1 process)" - key: distributed_remapping_1proc + - label: "Unit: distributed remapping with CUDA (1 process)" + key: distributed_remapping_gpu_1proc command: "julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl" env: - CLIMACOMMS_DEVICE: "CPU" + CLIMACOMMS_DEVICE: "CUDA" + agents: + slurm_gpus: 1 - - label: "Unit: distributed remapping with CUDA" - key: distributed_remapping_gpu + - label: "Unit: distributed remapping with CUDA (2 processes)" + key: distributed_remapping_gpu_2procs command: "srun julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl" env: CLIMACOMMS_CONTEXT: "MPI" CLIMACOMMS_DEVICE: "CUDA" agents: slurm_ntasks: 2 - slurm_gpus: 1 + slurm_gpus_per_task: 1 - label: "Unit: distributed gather" key: unit_distributed_gather4