Skip to content

Commit 06b46a6

Browse files
committed
Add Windows wheel build for x86_64 platform
Signed-off-by: Gigon Bae <[email protected]>
1 parent ef99071 commit 06b46a6

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

Diff for: ci/build_wheel.sh

+22
Original file line numberDiff line numberDiff line change
@@ -36,4 +36,26 @@ mkdir -p final_dist
3636
python -m auditwheel repair -w final_dist dist/*
3737
ls -1 final_dist | grep -vqz 'none'
3838

39+
# Build the wheel for Windows if the current platform is x86_64 (#763)
40+
if [ "$(arch)" = "x86_64" ]; then
41+
# Install the wheel package
42+
python -m pip install wheel
43+
44+
# Create directories
45+
mkdir -p unpacked converted_dist
46+
# Unpack the wheel
47+
wheel unpack final_dist/*_x86_64.whl -d unpacked
48+
49+
# Delete unnecessary files/folder (.so files, .libs folder)
50+
find unpacked -name '*.so*' -exec rm {} +
51+
find unpacked -maxdepth 2 -type d -name '*.libs' -exec rm -r {} +
52+
53+
# Pack the wheel again
54+
wheel pack --dest-dir converted_dist unpacked/*
55+
56+
# Change the platform tag to win_amd64 and store it in the final_dist folder
57+
wheel tags --remove --python-tag=py3 --abi-tag=none --platform-tag=win_amd64 converted_dist/*_x86_64.whl
58+
mv converted_dist/*-win_amd64.whl final_dist
59+
fi
60+
3961
RAPIDS_PY_WHEEL_NAME="${package_name}_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 final_dist

0 commit comments

Comments
 (0)