Skip to content

Commit

Permalink
Exclude libcuda.so, see what happens with libnvidia-ml.so
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Apr 27, 2024
1 parent d61f7ca commit 5aada80
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion ci/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -15,5 +15,9 @@ PACKAGE_CUDA_SUFFIX="-${RAPIDS_PY_CUDA_SUFFIX}"
sed -i -E "s/^name = \"${package_name}(.*)?\"$/name = \"${package_name}${PACKAGE_CUDA_SUFFIX}\"/g" ${pyproject_file}

python -m pip wheel "${package_dir}"/ -w "${package_dir}"/dist -vvv --no-deps --disable-pip-version-check
python -m auditwheel repair -w ${package_dir}/final_dist ${package_dir}/dist/*

# We must avoid bundling libcuda.so. What about libnvidia-ml.so? We had long
# discussions about this in the past, and at the time we settled on excluding
# it since we were going to be non-compliant without libcuda.so anyway.
python -m auditwheel repair -w ${package_dir}/final_dist --exclude "libcuda.so" ${package_dir}/dist/*
RAPIDS_PY_WHEEL_NAME="ucx_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 cpp ${package_dir}/final_dist

0 comments on commit 5aada80

Please sign in to comment.