Skip to content

Commit 7e5f88b

Browse files
committed
Update NCCL_BENCHMARKS_VERSION from 2.0.0 to 2.10.0 branch for nccl test
Signed-off-by: Yulei Wang <[email protected]>
1 parent b2deb23 commit 7e5f88b

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/integration-tests/tests/nccl/test_nccl/test_nccl/init_nccl_benchmarks.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ set -e
44
rm -rf /shared/${1}
55

66
module load ${1}
7-
NCCL_BENCHMARKS_VERSION='2.0.0'
7+
NCCL_BENCHMARKS_VERSION='2.10.0'
88
NCCL_VERSION='2.7.8-1'
99
ML_REPO_PKG='nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb'
1010
CUDA_VERSION='11.3'

tests/integration-tests/tests/nccl/test_nccl/test_nccl/nccl_tests_submit_openmpi.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@ mpirun \
1414
-x NCCL_ALGO=ring \
1515
-x NCCL_DEBUG=WARNING \
1616
--mca pml ^cm --mca btl tcp,self --mca btl_tcp_if_exclude lo,docker0 --bind-to none \
17-
/shared/openmpi/nccl-tests-2.0.0/build/all_reduce_perf -b 8 -e 1G -f 2 -g 1 -c 1 -n 100 > /shared/nccl_tests.out
17+
/shared/openmpi/nccl-tests-2.10.0/build/all_reduce_perf -b 8 -e 1G -f 2 -g 1 -c 1 -n 100 > /shared/nccl_tests.out

0 commit comments

Comments
 (0)