@@ -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