Skip to content

Commit 76d17bc

Browse files
committed
reverted the number of processes to be from slurm variables for the sbatch script
1 parent cb5897c commit 76d17bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

samples/gpu/nccl_run_allreduce_H100.sbatch

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,4 +65,4 @@ fi
6565
-x NCCL_IGNORE_CPU_AFFINITY=1 \
6666
-x NCCL_IB_HCA="${var_NCCL_IB_HCA}" \
6767
-x NCCL_TOPO_FILE=~/H100-topology.xml \
68-
--np $np --hostfile $MACHINEFILE /opt/oci-hpc/nccl-test/build/all_reduce_perf -b1G -e10G -i$((1024*1024*1024*9)) -n 100
68+
--np $((SLURM_NNODES*SLURM_NTASKS_PER_NODE)) --hostfile $MACHINEFILE /opt/oci-hpc/nccl-test/build/all_reduce_perf -b1G -e10G -i$((1024*1024*1024*9)) -n 100

0 commit comments

Comments
 (0)