Skip to content

Commit 6990415

Browse files
[integ-tests] Upgrade NCCL versions
NCCL version is retrieved from https://github.com/NVIDIA/nccl/tags NCCL benchmark version is retrieved from https://github.com/NVIDIA/nccl-tests/tags
1 parent d461d5a commit 6990415

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/integration-tests/tests/efa/test_efa/test_efa/nccl_benchmarks/init_nccl_benchmarks.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@ set -xe
55
rm -rf /shared/${1}
66

77
module load ${1}
8-
NCCL_BENCHMARKS_VERSION='2.13.8'
9-
NCCL_VERSION='2.19.4-1'
8+
NCCL_BENCHMARKS_VERSION='2.16.7'
9+
NCCL_VERSION='2.27.7-1'
1010
MPI_HOME=$(which mpirun | awk -F '/bin' '{print $1}')
1111
NVCC_GENCODE="-gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_90,code=sm_90 -gencode=arch=compute_90,code=compute_90" # Arch for NVIDIA A100 and H100, ref https://docs.nvidia.com/cuda/ada-compatibility-guide/index.html
1212

tests/integration-tests/tests/efa/test_efa/test_efa/nccl_benchmarks/nccl_tests_submit_openmpi.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
#SBATCH --ntasks-per-node=8
55

66
module load openmpi
7-
NCCL_VERSION='2.19.4-1'
8-
NCCL_BENCHMARKS_VERSION='2.13.8'
7+
NCCL_VERSION='2.27.7-1'
8+
NCCL_BENCHMARKS_VERSION='2.16.7'
99

1010
mpirun \
1111
-x FI_PROVIDER="efa" \

0 commit comments

Comments
 (0)