fix(nccl-tests): set LIBRARY_PATH so ld finds libnccl.so in nccl cache
This commit is contained in:
@@ -118,6 +118,7 @@ else
|
|||||||
-gencode=arch=compute_100,code=sm_100"
|
-gencode=arch=compute_100,code=sm_100"
|
||||||
echo "gencode: sm_70..sm_100 (CUDA 12)"
|
echo "gencode: sm_70..sm_100 (CUDA 12)"
|
||||||
fi
|
fi
|
||||||
|
LIBRARY_PATH="$NCCL_LIB_DIR${LIBRARY_PATH:+:$LIBRARY_PATH}" \
|
||||||
make MPI=0 \
|
make MPI=0 \
|
||||||
NVCC="$NVCC" \
|
NVCC="$NVCC" \
|
||||||
CUDA_HOME="$CUDA_HOME" \
|
CUDA_HOME="$CUDA_HOME" \
|
||||||
|
|||||||
Reference in New Issue
Block a user