File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tests/integration-tests/tests/efa/test_efa/test_efa/nccl_benchmarks Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -7,7 +7,6 @@ module load ${1}
77NCCL_BENCHMARKS_VERSION=' 2.10.0'
88NCCL_VERSION=' 2.7.8-1'
99ML_REPO_PKG=' nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb'
10- CUDA_VERSION=' 11.4'
1110OFI_NCCL_VERSION=' 1.1.1'
1211MPI_HOME=$( which mpirun | awk -F ' /bin' ' {print $1}' )
1312NVCC_GENCODE=" -gencode=arch=compute_80,code=sm_80" # Arch for NVIDIA A100
@@ -32,6 +31,6 @@ wget https://github.com/aws/aws-ofi-nccl/archive/v${OFI_NCCL_VERSION}.tar.gz
3231tar xvfz v${OFI_NCCL_VERSION} .tar.gz
3332cd aws-ofi-nccl-${OFI_NCCL_VERSION}
3433./autogen.sh
35- ./configure --with-libfabric=/opt/amazon/efa --with-cuda=/usr/local/cuda- ${CUDA_VERSION} /targets/x86_64-linux/ --with-nccl=/shared/openmpi/nccl-${NCCL_VERSION} /build/ --with-mpi=${MPI_HOME} --prefix /shared/openmpi/ofi-plugin
34+ ./configure --with-libfabric=/opt/amazon/efa --with-cuda=/usr/local/cuda/targets/x86_64-linux/ --with-nccl=/shared/openmpi/nccl-${NCCL_VERSION} /build/ --with-mpi=${MPI_HOME} --prefix /shared/openmpi/ofi-plugin
3635make
37- make install
36+ make install
You can’t perform that action at this time.
0 commit comments