Repository navigation
Forward-merge release/26.12 into main - #281
Conversation
Depend on cuda-core[cu12]/[cu13] in the respective wheel. cuda-core 1.3.0 (released 2026-11-11) refuses to import unless cuda-bindings meets a minimum version (>=12.9.8 for CUDA 12, >=13.5.0 for CUDA 13). Its metadata declares that minimum only under the `cu12`/`cu13` extras. nvforest wheels depend on plain `cuda-core`, so pip can install cuda-core 1.3.0 next to an older cuda-bindings, and then `import cuda.core` fails: ```text ImportError: cuda.core 1.3.0 requires cuda-bindings >= 12.9.8 for CUDA 12, but cuda-bindings 12.9.3 is installed. ``` This breaks cuML's nightly wheel tests in two ways: - In the oldest-deps jobs, cuML pins `cuda-bindings==12.9.3`. - In the CUDA 12 integrations job, `torch==2.10.0+cu129` pins `cuda-bindings==12.9.4`. Examples: - https://github.com/NVIDIA/cuml/actions/runs/43171195643/job/131024721358 - https://github.com/NVIDIA/cuml/actions/runs/43171195643/job/131024702175 This PR makes the wheels depend on `cuda-core[cu12]` / `cuda-core[cu13]`, so pip sees the minimum and falls back to cuda-core 1.2.x (the version before what was just released) when cuda-bindings is pinned below 12..9.8. dask-cuda already does this, and cuML's oldest-deps dask job, which gets cuda-core through dask-cuda, resolved cuda-core 1.2.2 and passed. We could bump the cuda-binding in the oldest job instead, but we'd still be stuck for CUDA 12 in general I think (because pytorch pins things exactly). I think with this we get the "best of all worlds" where you get the newer cuda-core if there is nothing in your env that keeps cuda-bindings pinned low. And if there is you get the older version of cuda-core. I don't know what the downside of sometimes having cuda-core 1.2.x and sometimes having 1.3.0 is. One downside is that through `cuda-bindings[all]`, the extra also installs the nvcc, nvrtc, nvjitlink and cufile wheels. This isn't great but for cuml these would anyway get installed. Authors: - Tim Head (https://github.com/betatim) Approvers: - James Lamb (https://github.com/jameslamb) URL: rapidsai#277
|
No actionable comments were generated in the recent review. Summary
Merge Risk:
|
|
/merge nosquash |
See #279