We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 59452c0 + 76a3fc7 commit b8f986bCopy full SHA for b8f986b
scripts/performance/utils/executors.py
@@ -125,6 +125,8 @@ def slurm_executor(
125
126
numa_divisor = 2 if gpu.lower() in ["gb200", "gb300"] else 4
127
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))"
130
custom_bash_cmds.append(numa_cmd)
131
132
launcher = SlurmTemplate(
0 commit comments