diff --git a/contrib/build-wheel.sh b/contrib/build-wheel.sh index d765bf9ac3..f5bcff7bec 100755 --- a/contrib/build-wheel.sh +++ b/contrib/build-wheel.sh @@ -97,7 +97,7 @@ if [ "$CUDA_MAJOR" -ne 12 ] && [ "$CUDA_MAJOR" -ne 13 ]; then echo "Invalid CUDA_MAJOR: '$CUDA_MAJOR'" exit 1 fi -AUDITWHEEL_EXCLUDES="--exclude libcuda* --exclude libcufile* --exclude libssl* --exclude libcrypto* --exclude libefa* --exclude libhwloc* --exclude libfabric* --exclude libtorch* --exclude libc10* --exclude libdoca* --exclude libred_client* --exclude libred_async* --exclude liblz4*" +AUDITWHEEL_EXCLUDES="--exclude libcuda* --exclude libcufile* --exclude libssl* --exclude libcrypto* --exclude libefa* --exclude libhwloc* --exclude libfabric* --exclude libtorch* --exclude libc10* --exclude libdoca* --exclude libred_client* --exclude libred_async* --exclude liblz4* --exclude libibverbs* --exclude libmlx5*" PKG_NAME="nixl-cu${CUDA_MAJOR}" CU_TAG="cu$(nvcc --version | grep -Eo 'release [0-9]+\.[0-9]+' | cut -d' ' -f2 | tr -d .)"