Skip to content

Commit b8f986b

Browse files
committed
Merge branch 'main' into liding/nano-v3-pr
2 parents 59452c0 + 76a3fc7 commit b8f986b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

scripts/performance/utils/executors.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,8 @@ def slurm_executor(
125125

126126
numa_divisor = 2 if gpu.lower() in ["gb200", "gb300"] else 4
127127
numa_cmd = f"numactl --cpunodebind=$((SLURM_LOCALID/{numa_divisor})) --membind=$((SLURM_LOCALID/{numa_divisor}))"
128+
if gpu.lower() in ["b300"]:
129+
numa_cmd += " -C $((SLURM_LOCALID * 16)),$((SLURM_LOCALID * 16 + 1))"
128130
custom_bash_cmds.append(numa_cmd)
129131

130132
launcher = SlurmTemplate(

0 commit comments

Comments
 (0)