Skip to content

Commit

Permalink
Add Windows wheel build for x86_64 platform
Browse files Browse the repository at this point in the history
Signed-off-by: Gigon Bae <gbae@nvidia.com>
  • Loading branch information
gigony committed Aug 21, 2024
1 parent ef99071 commit 2ec710e
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions ci/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,4 +36,26 @@ mkdir -p final_dist
python -m auditwheel repair -w final_dist dist/*
ls -1 final_dist | grep -vqz 'none'

# Build the wheel for Windows if the current platform is x86_64 (#763)
if [ "$(arch)" = "x86_64" ]; then
# Install the wheel package
python -m pip install wheel

# Create directories
mkdir -p unpacked converted_dist
# Unpack the wheel
python -m wheel unpack final_dist/*_x86_64.whl -d unpacked

# Delete unnecessary files/folder (.so files, .libs folder)
find unpacked -name '*.so*' -exec rm {} +
find unpacked -maxdepth 2 -type d -name '*.libs' -exec rm -r {} +

# Pack the wheel again
python -m wheel pack --dest-dir converted_dist unpacked/*

# Change the platform tag to win_amd64 and store it in the final_dist folder
python -m wheel tags --remove --python-tag=py3 --abi-tag=none --platform-tag=win_amd64 converted_dist/*_x86_64.whl
mv converted_dist/*-win_amd64.whl final_dist
fi

RAPIDS_PY_WHEEL_NAME="${package_name}_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 final_dist

0 comments on commit 2ec710e

Please sign in to comment.