always specifu GPU number

radical-cybertools · Apr 2, 2021 · aee4fb8 · aee4fb8
1 parent acf15b8
commit aee4fb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/radical/pilot/agent/launch_method/srun.py b/src/radical/pilot/agent/launch_method/srun.py
@@ -110,6 +110,7 @@ def construct_command(self, t, launch_script_hop):
         mapping = '--exclusive --cpu-bind=none ' \
                 + '--nodes %d '        % n_nodes \
                 + '--ntasks %d '       % n_tasks \
+                + '--gpus %d '         % (n_gpus * n_tasks) \
                 + '--cpus-per-task %d' % n_task_threads
 
         # check that gpus were requested to be allocated