Skip to content

Commit ae690f5

Browse files
committed
update phoenix template
1 parent 068da2c commit ae690f5

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

toolchain/templates/phoenix.mako

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,15 @@ echo
4141
% if not mpi:
4242
(set -x; ${profiler} "${target.get_install_binpath(case)}")
4343
% else:
44-
(set -x; ${profiler} \
45-
mpirun -np ${nodes*tasks_per_node} \
46-
--bind-to none \
47-
"${target.get_install_binpath(case)}")
44+
(set -x; srun \
45+
% if engine == 'interactive':
46+
--nodes ${nodes} --ntasks-per-node ${tasks_per_node} \
47+
--cpus-per-task 1 \
48+
% if gpu:
49+
--gpus-per-task 1 --gpu-bind closest \
50+
% endif
51+
% endif
52+
${profiler} "${target.get_install_binpath(case)}")
4853
% endif
4954

5055
${helpers.run_epilogue(target)}

0 commit comments

Comments
 (0)