[CI] [WheelNext] Build variant wheels #11531
Merged
+124
−0
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
trivialfis
reviewed
Jul 1, 2025
| python -m pip install -v \ | ||
| git+https://github.com/wheelnext/pep_xxx_wheel_variants.git@f3b287090f8a6f510b0e1723896e1c7e638f6bff#subdirectory=pep_xxx_wheel_variants | ||
| pip config set --site global.index-url https://variants-index.wheelnext.dev/ | ||
| variantlib make-variant -f python-package/dist/xgboost-*.whl \ |
Member
There was a problem hiding this comment.
Could you please provide some comments on what this line does? Skimming the pipeline, this follows a CUDA build, I don't see any potential variant here since there's no CPU build and NCCL is still installed.
Collaborator
Author
There was a problem hiding this comment.
This is an initial stage, where we build a single variant for the CUDA build. In the future, I plan to build the CPU variant and other variants.
trivialfis
approved these changes
Jul 1, 2025
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
WheelNext is a proposal to improve Python wheels to better support specialized hardware and reduce the size of individual wheels. In particular, we are keen to adopt the Wheel Variant proposal. Also see the presentation.
Use the prototype in https://github.com/wheelnext/pep_xxx_wheel_variants to build a variant wheel. A variant wheel is just like a regular wheel, but with extra variant metadata like
nvidia :: cuda :: 12.