File tree Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Original file line number Diff line number Diff line change 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}
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)}
You can’t perform that action at this time.
0 commit comments