Skip to content

Commit a6116f2

Browse files
committed
Set cpus-per-task to 72 and update binding script
1 parent 457ae60 commit a6116f2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

misc/nvidia_uvm/bind.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ export MPICH_OFI_NIC_POLICY=USER
1515
export MPICH_OFI_NIC_MAPPING="0:0; 1:1; 2:2; 3:3"
1616

1717
# Bind to cores ( first core per socket )
18-
physcores=(0 72 144 216)
18+
physcores=(0-71 72-143 144-215 216-287)
1919

2020
#echo hostname: $(hostname), rank: $local_rank, cores: ${physcores[$local_rank]}, GPU: $CUDA_VISIBLE_DEVICES, NIC mapping: $MPICH_OFI_NIC_POLICY
2121

toolchain/templates/santis.mako

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ echo
7171
% else:
7272
(set -x; srun --unbuffered \
7373
--ntasks=${nodes*tasks_per_node} \
74-
--cpus-per-task 1 \
74+
--cpus-per-task 72 \
7575
--cpu-bind=none \
7676
% if gpu:
7777
--gpus-per-task 1 \

0 commit comments

Comments
 (0)