Updates 'any' node type + 'n' gpus where n != 0 behavior

OSC · May 16, 2024 · 8d0ca26 · 8d0ca26
1 parent 52e1a01
commit 8d0ca26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/apps.awesim.org/apps/bc_desktop/submit/slurm.yml.erb b/apps.awesim.org/apps/bc_desktop/submit/slurm.yml.erb
@@ -30,7 +30,7 @@
   slurm_args = case node_type
               # 'any' case handled by scheduler, this is just a quick short circuit
               when "any"
-                base_slurm_args + any_node
+                base_slurm_args + any_node + ["--gpus-per-node", "#{gpu_count}"]
 
               when "any-40core"
                 base_slurm_args + p18_node

diff --git a/ondemand.osc.edu/apps/bc_desktop/submit/slurm.yml.erb b/ondemand.osc.edu/apps/bc_desktop/submit/slurm.yml.erb
@@ -30,7 +30,7 @@
   slurm_args = case node_type
               # 'any' case handled by scheduler, this is just a quick short circuit
               when "any"
-                base_slurm_args + any_node
+                base_slurm_args + any_node + ["--gpus-per-node", "#{gpu_count}"]
 
               when "any-40core"
                 base_slurm_args + p18_node