Skip to content

WIP: build wheels with CUDA 13.0.x, test wheels against mix of CTK versions#604

Draft
jameslamb wants to merge 2 commits intorapidsai:mainfrom
jameslamb:test-oldest-ctk
Draft

WIP: build wheels with CUDA 13.0.x, test wheels against mix of CTK versions#604
jameslamb wants to merge 2 commits intorapidsai:mainfrom
jameslamb:test-oldest-ctk

Conversation

@jameslamb
Copy link
Member

Description

Contributes to rapidsai/build-planning#257

  • builds CUDA 13 wheels with the 13.0 CTK

Contributes to rapidsai/build-planning#256

  • updates wheel tests to cover a range of CTK versions (we previously, accidentally, were only testing the latest 12.x and 13.x)

Notes for Reviewers

How I tested this

This uses wheels from similar PRs from RAPIDS dependencies, at build and test time:

@jameslamb jameslamb added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Mar 5, 2026
@copy-pr-bot
Copy link

copy-pr-bot bot commented Mar 5, 2026

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@jameslamb
Copy link
Member Author

/ok to test

@jameslamb
Copy link
Member Author

/ok to test

packages:
# some components (like nvidia-cublas-cu12 and nvidia-cuda-nvcc-cu12) didn't have
# aarch64 wheels until CTK 12.3, so allow a slightly looser bound here
- cuda-toolkit>=12.2,<12.4
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

For this repo, it's nvidia-cuda-nvcc-cu12 that ends up being required but can't be installed on aarch64 + CUDA 12.2:

docker run --rm -it rapidsai/citestwheel:26.04-cuda12.2.2-ubuntu22.04-py3.11 bash

pip install 'cuda-toolkit[nvcc,nvrtc]==12.2.*'
# ERROR: ResolutionImpossible

pip install 'cuda-toolkit[nvcc,nvrtc]==12.3.0'
# ERROR: No matching distribution found for nvidia-cuda-nvcc-cu12

pip install 'cuda-toolkit[nvcc,nvrtc]==12.3.1'
# ERROR: No matching distribution found for nvidia-cuda-nvcc-cu12

pip install 'cuda-toolkit[nvcc,nvrtc]==12.3.2'
# Successfully installed cuda-toolkit-12.3.2 nvidia-cuda-nvcc-cu12-12.3.107 nvidia-cuda-nvrtc-cu12-12.3.107

But trying to keep this cuda_version: list consistent across all of RAPIDs, so including cuBLAS in the comment too.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant