@@ -243,30 +243,38 @@ steps:
243
243
agents :
244
244
slurm_ntasks : 4
245
245
246
- - label : " Unit: distributed remapping"
247
- key : distributed_remapping
246
+ - label : " Unit: distributed remapping (1 process)"
247
+ key : distributed_remapping_1proc
248
+ command : " julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl"
249
+ env :
250
+ CLIMACOMMS_DEVICE : " CPU"
251
+
252
+ - label : " Unit: distributed remapping (2 processes)"
253
+ key : distributed_remapping_2procs
248
254
command : " srun julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl"
249
255
env :
250
256
CLIMACOMMS_CONTEXT : " MPI"
251
257
CLIMACOMMS_DEVICE : " CPU"
252
258
agents :
253
259
slurm_ntasks : 2
254
260
255
- - label : " Unit: distributed remapping (1 process)"
256
- key : distributed_remapping_1proc
261
+ - label : " Unit: distributed remapping with CUDA (1 process)"
262
+ key : distributed_remapping_gpu_1proc
257
263
command : " julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl"
258
264
env :
259
- CLIMACOMMS_DEVICE : " CPU"
265
+ CLIMACOMMS_DEVICE : " CUDA"
266
+ agents :
267
+ slurm_gpus : 1
260
268
261
- - label : " Unit: distributed remapping with CUDA"
262
- key : distributed_remapping_gpu
269
+ - label : " Unit: distributed remapping with CUDA (2 processes) "
270
+ key : distributed_remapping_gpu_2procs
263
271
command : " srun julia --color=yes --check-bounds=yes --project=test test/Remapping/distributed_remapping.jl"
264
272
env :
265
273
CLIMACOMMS_CONTEXT : " MPI"
266
274
CLIMACOMMS_DEVICE : " CUDA"
267
275
agents :
268
276
slurm_ntasks : 2
269
- slurm_gpus : 1
277
+ slurm_gpus_per_task : 1
270
278
271
279
- label : " Unit: distributed gather"
272
280
key : unit_distributed_gather4
0 commit comments