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 18
18
# SBATCH --qos=${quality_of_service}
19
19
% endif
20
20
% if gpu:
21
- # SBATCH -G${tasks_per_node}
21
+ # SBATCH --gres=gpu:V100:${tasks_per_node}
22
+ # SBATCH --mem-per-gpu=16G\
22
23
% endif
23
24
% if email:
24
25
# SBATCH --mail-user=${email}
40
41
% if not mpi:
41
42
(set -x; ${profiler} " ${target.get_install_binpath(case)} " )
42
43
% 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)} " )
52
48
% endif
53
49
54
50
${helpers.run_epilogue(target)}
You can’t perform that action at this time.
0 commit comments