Skip to content

Commit 9aa309a

Browse files
authored
Revert "Update phoenix.mako template (#822)" (#823)
1 parent 3995f4e commit 9aa309a

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

toolchain/templates/phoenix.mako

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
#SBATCH --qos=${quality_of_service}
1919
% endif
2020
% if gpu:
21-
#SBATCH -G${tasks_per_node}
21+
#SBATCH --gres=gpu:V100:${tasks_per_node}
22+
#SBATCH --mem-per-gpu=16G\
2223
% endif
2324
% if email:
2425
#SBATCH --mail-user=${email}
@@ -40,15 +41,10 @@ echo
4041
% if not mpi:
4142
(set -x; ${profiler} "${target.get_install_binpath(case)}")
4243
% else:
43-
(set -x; srun \
44-
% if engine == 'interactive':
45-
--nodes ${nodes} --ntasks-per-node ${tasks_per_node} \
46-
--cpus-per-task 1 \
47-
% if gpu:
48-
--gpus-per-task 1 --gpu-bind closest \
49-
% endif
50-
% endif
51-
${profiler} "${target.get_install_binpath(case)}")
44+
(set -x; ${profiler} \
45+
mpirun -np ${nodes*tasks_per_node} \
46+
--bind-to none \
47+
"${target.get_install_binpath(case)}")
5248
% endif
5349

5450
${helpers.run_epilogue(target)}

0 commit comments

Comments
 (0)