Skip to content

Commit e8f041d

Browse files
authored
[backport] [CI] Patch to fix wheel build for WinARM64 (#11702) (#11703)
* [CI] Patch to fix wheel build for WinARM64 (#11702) * [CI] Additional fix for WinARM64 * [CI] Additional fix for WinARM64
1 parent 5546965 commit e8f041d

File tree

1 file changed

+28
-4
lines changed

1 file changed

+28
-4
lines changed

.github/workflows/python_wheels_winarm64.yml

Lines changed: 28 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,26 +34,50 @@ jobs:
3434
- name: Install build dependencies
3535
run: |
3636
python -m pip install --upgrade pip
37-
python -m pip install wheel setuptools awscli
37+
python -m pip install wheel setuptools awscli packaging
3838
3939
- name: Build XGBoost for Win-ARM64
4040
run: |
4141
mkdir build
4242
cd build
4343
cmake .. -G"Visual Studio 17 2022" -A ARM64
4444
cmake --build . --config Release -- /m /nodeReuse:false "/consoleloggerparameters:ShowCommandLine;Verbosity=minimal"
45+
46+
- name: Build Python wheel xgboost for Win-ARM64
47+
run: |
48+
cd python-package
49+
mkdir -p wheelhouse
50+
pip wheel --no-deps -v . --wheel-dir wheelhouse/
51+
$wheelFile = Get-ChildItem wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
52+
python -m wheel tags --python-tag py3 --abi-tag none --platform win_arm64 --remove $wheelFile
53+
54+
- name: Upload Python wheel xgboost
55+
if: github.ref == 'refs/heads/master' || contains(github.ref, 'refs/heads/release_')
56+
run: |
57+
$wheelFile = Get-ChildItem python-package/wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
58+
python ops/pipeline/manage-artifacts.py upload `
59+
--s3-bucket xgboost-nightly-builds `
60+
--prefix ${{ env.BRANCH_NAME }}/${{ github.sha }} --make-public `
61+
$wheelFile
62+
env:
63+
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID_IAM_S3_UPLOADER }}
64+
AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY_IAM_S3_UPLOADER }}
65+
66+
- name: Clean up
67+
run: |
68+
$wheelFile = Get-ChildItem python-package/wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
69+
Remove-Item -Path $wheelFile -Verbose
4570
46-
- name: Build XGBoost Python wheel for Win-ARM64
71+
- name: Build Python wheel xgboost-cpu for Win-ARM64
4772
run: |
4873
# Patch to rename pkg to xgboost-cpu
4974
python ops/script/pypi_variants.py --use-suffix=cpu --require-nccl-dep=na
5075
cd python-package
51-
mkdir -p wheelhouse
5276
pip wheel --no-deps -v . --wheel-dir wheelhouse/
5377
$wheelFile = Get-ChildItem wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
5478
python -m wheel tags --python-tag py3 --abi-tag none --platform win_arm64 --remove $wheelFile
5579
56-
- name: Upload Python wheel
80+
- name: Upload Python wheel xgboost-cpu
5781
if: github.ref == 'refs/heads/master' || contains(github.ref, 'refs/heads/release_')
5882
run: |
5983
$wheelFile = Get-ChildItem python-package/wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName

0 commit comments

Comments
 (0)