diff --git a/src/radical/pilot/agent/launch_method/srun.py b/src/radical/pilot/agent/launch_method/srun.py index 94e42bff60..076b236040 100644 --- 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