Skip to content

[CI] Adding ARM CUDA wheels for pypi#11827

Merged
hcho3 merged 10 commits into
dmlc:masterfrom
dantegd:arm-cuda-wheels
Dec 10, 2025
Merged

[CI] Adding ARM CUDA wheels for pypi#11827
hcho3 merged 10 commits into
dmlc:masterfrom
dantegd:arm-cuda-wheels

Conversation

@dantegd

@dantegd dantegd commented Nov 20, 2025

Copy link
Copy Markdown
Contributor

Closes #11563

@trivialfis

Copy link
Copy Markdown
Member

@hcho3 hcho3 changed the title Adding ARM CUDA wheels for pypi [CI] Adding ARM CUDA wheels for pypi Dec 10, 2025
@hcho3
hcho3 marked this pull request as ready for review December 10, 2025 05:18
@hcho3

hcho3 commented Dec 10, 2025

Copy link
Copy Markdown
Collaborator

I published new containers for ARM: dmlc/xgboost-devops#54, dmlc/xgboost-devops#56, dmlc/xgboost-devops#57

@hcho3

hcho3 commented Dec 10, 2025

Copy link
Copy Markdown
Collaborator

The PR is now ready for review.

@hcho3
hcho3 requested a review from trivialfis December 10, 2025 21:10
@hcho3
hcho3 merged commit b7feb5b into dmlc:master Dec 10, 2025
69 of 70 checks passed
hcho3 added a commit to hcho3/xgboost that referenced this pull request Dec 12, 2025
* FEA first commit for ARM CUDA wheels

* Enable building CUDA 12 wheel on ARM64

* fix

* typo in script

* Use matrix to de-duplicate jobs for x86_64, aarch64

* Removed unused build-cuda-arm64.sh

* typo

* Use correct artifact for mgpu test

* Use correct Docker image for arm64 gtest

---------

Co-authored-by: Hyunsu Cho <[email protected]>
hcho3 added a commit that referenced this pull request Dec 12, 2025
* [CI] Adding ARM CUDA wheels for pypi (#11827)

* FEA first commit for ARM CUDA wheels

* Enable building CUDA 12 wheel on ARM64

* fix

* typo in script

* Use matrix to de-duplicate jobs for x86_64, aarch64

* Removed unused build-cuda-arm64.sh

* typo

* Use correct artifact for mgpu test

* Use correct Docker image for arm64 gtest

---------

Co-authored-by: Hyunsu Cho <[email protected]>

* Backport part of #11796

---------

Co-authored-by: Dante Gama Dessavre <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[CI] Publishing ARM64 wheels on PyPI with GPU support

3 participants