Fix pipeline

CliMA · Apr 15, 2024 · 767d0aa · 767d0aa
1 parent 8128274
commit 767d0aa
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -243,30 +243,38 @@ 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"
           CLIMACOMMS_DEVICE: "CPU"
         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