don't require gpus to be on 1 node

CliMA · Oct 10, 2023 · cd3138f · cd3138f
1 parent 18dfc99
commit cd3138f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.buildkite/distributed/pipeline.yml b/.buildkite/distributed/pipeline.yml
@@ -50,9 +50,8 @@ steps:
     commands:
       - "srun julia -O0 --color=yes --project -e 'using Pkg; Pkg.test()'"
     agents:
-      slurm_nodes: 1
-      slurm_tasks_per_node: 4
-      slurm_gpus: 4
+      slurm_ntasks: 4
+      slurm_gpus_per_task: 1
 
   - label: "🦾 cpu distributed solvers tests"
     key: "distributed_solvers_cpu"