Skip to content

Commit bbf837a

Browse files
authored
use CUDA-suffixed XGBoost wheels (#7550)
Contributes to rapidsai/build-planning#208 `xgboost` wheels currently follow this pattern: * `xgboost` (CUDA 12 support by default) * `xgboost-cu12` (always CUDA 12) * `xgboost-cu13` (always CUDA 13) See: * dmlc/xgboost#10807 (comment) * dmlc/xgboost#11677 This switches `cuml` to use those suffixed wheels, so the correct version will always be pulled in. As of this PR, `cuml-cu13[test]` will now depend on `xgboost-cu13` instead of `xgboost`. ## Notes for Reviewers ### XGBoost doesn't have arm64 wheels with CUDA 13 support See #7550 (comment) Authors: - James Lamb (https://github.com/jameslamb) Approvers: - Bradley Dice (https://github.com/bdice) URL: #7550
1 parent 4f78f6d commit bbf837a

File tree

1 file changed

+33
-5
lines changed

1 file changed

+33
-5
lines changed

dependencies.yaml

Lines changed: 33 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,7 @@ files:
155155
- py_run_cuml
156156
- py_version
157157
- test_notebooks
158+
- test_python_xgboost
158159
py_build_cuml:
159160
output: pyproject
160161
pyproject_dir: python/cuml
@@ -506,10 +507,39 @@ dependencies:
506507
# included in the "devcontainers" key. The libxgboost package depends
507508
# on librmm but we do not want to have a package depending on librmm
508509
# in devcontainers since it should be built from source.
509-
- &rapids-xgboost rapids-xgboost==26.2.*,>=0.0.0a0
510+
- rapids-xgboost==26.2.*,>=0.0.0a0
511+
specific:
510512
- output_types: [requirements, pyproject]
511-
packages:
512-
- &xgboost xgboost>=2.1.0
513+
matrices:
514+
- matrix:
515+
cuda: "12.*"
516+
cuda_suffixed: "true"
517+
packages:
518+
- xgboost-cu12>=2.1.0
519+
- matrix:
520+
arch: aarch64
521+
cuda: "13.*"
522+
cuda_suffixed: "true"
523+
packages:
524+
# TODO: always install xgboost-cu13 once there are xgboost-cu13 aarch64 wheels
525+
#
526+
# xgboost-cu13 originally shipped without aarch64 wheels
527+
# Using the unsuffixed name pulls in CUDA 12 packages for some versions and may
528+
# at some point in the future pull in CUDA 13 packages.
529+
#
530+
# ref: https://github.com/rapidsai/cuml/pull/7550#issuecomment-3603279934
531+
- &xgboost_unsuffixed xgboost>=2.1.0
532+
- matrix:
533+
arch: x86_64
534+
cuda: "13.*"
535+
cuda_suffixed: "true"
536+
packages:
537+
- xgboost-cu13>=2.1.0
538+
# this unsuffixed dependency is useful in environments where xgboost is built from source
539+
# (like RAPIDS DLFW builds)
540+
- matrix:
541+
packages:
542+
- *xgboost_unsuffixed
513543
test_notebooks:
514544
common:
515545
- output_types: [conda, requirements]
@@ -523,11 +553,9 @@ dependencies:
523553
- output_types: conda
524554
packages:
525555
- matplotlib-base
526-
- *rapids-xgboost
527556
- output_types: requirements
528557
packages:
529558
- matplotlib
530-
- *xgboost
531559
depends_on_cuda_python:
532560
specific:
533561
- output_types: [conda, requirements, pyproject]

0 commit comments

Comments
 (0)