Skip to content

Commit 938958c

Browse files
committed
Reference CUDA directory in integration test through normalized linked name
1 parent 1c26fae commit 938958c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ module load ${1}
77
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'
10-
CUDA_VERSION='11.4'
1110
OFI_NCCL_VERSION='1.1.1'
1211
MPI_HOME=$(which mpirun | awk -F '/bin' '{print $1}')
1312
NVCC_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
3231
tar xvfz v${OFI_NCCL_VERSION}.tar.gz
3332
cd 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
3635
make
37-
make install
36+
make install

0 commit comments

Comments
 (0)