Skip to content

Commit 681c662

Browse files
authored
[CI] Patch to fix wheel build for WinARM64 (dmlc#11702)
1 parent 7efa4fc commit 681c662

File tree

1 file changed

+22
-4
lines changed

1 file changed

+22
-4
lines changed

.github/workflows/python_wheels_winarm64.yml

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,33 +34,51 @@ 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+
Remove-Item -Path $wheelFile -Verbose
4563
46-
- name: Build XGBoost Python wheel for Win-ARM64
64+
- name: Build Python wheel xgboost-cpu for Win-ARM64
4765
run: |
4866
# Patch to rename pkg to xgboost-cpu
4967
python ops/script/pypi_variants.py --use-suffix=cpu --require-nccl-dep=na
5068
cd python-package
51-
mkdir -p wheelhouse
5269
pip wheel --no-deps -v . --wheel-dir wheelhouse/
5370
$wheelFile = Get-ChildItem wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
5471
python -m wheel tags --python-tag py3 --abi-tag none --platform win_arm64 --remove $wheelFile
5572
56-
- name: Upload Python wheel
73+
- name: Upload Python wheel xgboost-cpu
5774
if: github.ref == 'refs/heads/master' || contains(github.ref, 'refs/heads/release_')
5875
run: |
5976
$wheelFile = Get-ChildItem python-package/wheelhouse/*.whl | Select-Object -First 1 -ExpandProperty FullName
6077
python ops/pipeline/manage-artifacts.py upload `
6178
--s3-bucket xgboost-nightly-builds `
6279
--prefix ${{ env.BRANCH_NAME }}/${{ github.sha }} --make-public `
6380
$wheelFile
81+
6482
env:
6583
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID_IAM_S3_UPLOADER }}
6684
AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY_IAM_S3_UPLOADER }}

0 commit comments

Comments
 (0)