Skip to content

Update to numba-cuda>=0.16#462

Merged
rapids-bot[bot] merged 3 commits intorapidsai:branch-0.45from
pentschev:update-numba-cuda-stream
Jul 15, 2025
Merged

Update to numba-cuda>=0.16#462
rapids-bot[bot] merged 3 commits intorapidsai:branch-0.45from
pentschev:update-numba-cuda-stream

Conversation

@pentschev
Copy link
Copy Markdown
Member

@pentschev pentschev commented Jul 15, 2025

Update to numba-cuda>=0.16 to match cuDF, and update stream synchronization code in distributed-ucxx.

Closes rapidsai/dask-upstream-testing#71.

@pentschev pentschev self-assigned this Jul 15, 2025
@pentschev pentschev requested review from a team as code owners July 15, 2025 21:05
@pentschev pentschev requested a review from KyleFromNVIDIA July 15, 2025 21:05
@pentschev pentschev added feature request New feature or request non-breaking Introduces a non-breaking change labels Jul 15, 2025
@pentschev
Copy link
Copy Markdown
Member Author

/merge

@pentschev
Copy link
Copy Markdown
Member Author

I've preemptively triggered an auto-merge, without this PR many of our projects that test Dask+UCXX will fail nightlies because of the change to numba-cuda=0.16 that had some breaking API changes. @jakirkham if you're still around would appreciate if you could approve this on behalf of ucxx-python-codeowners, you're the only one who's not currently on PTO besides me.

@TomAugspurger
Copy link
Copy Markdown
Contributor

Gil got me added to ucxx-python-codeowners so we should be good to go here.

@rapids-bot rapids-bot bot merged commit d3aa4a3 into rapidsai:branch-0.45 Jul 15, 2025
54 checks passed
@pentschev pentschev deleted the update-numba-cuda-stream branch July 16, 2025 08:17
rapids-bot bot pushed a commit that referenced this pull request Jul 16, 2025
In #462 the Numba dependency was adjusted for `numba-cuda>=0.16`, but the `numba-cuda` package in PyPI is a metapackage while the required one is `numba-cuda[cu12]`. This change fixes the dependency for wheels installs.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Bradley Dice (https://github.com/bdice)

URL: #463
rapids-bot bot pushed a commit to rapidsai/rapids-dask-dependency that referenced this pull request Jul 16, 2025
This change applies rapidsai/ucxx#462 to the `protocol="ucx-old"` patch, without which Dask+UCX-Py tests currently fail.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Tom Augspurger (https://github.com/TomAugspurger)

URL: #118
rapids-bot bot pushed a commit that referenced this pull request Jul 17, 2025
- Partially reverts #462.
- Use numba-cuda>=0.15.2,<0.16

We are investigating an issue where the new CUDA bindings in numba-cuda 0.16 cause segfaults, blocking our CI. This temporarily downgrades to numba-cuda >=0.15.2,<0.16 until a fix can be made.

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Gil Forsyth (https://github.com/gforsyth)

URL: #466
rapids-bot bot pushed a commit that referenced this pull request Aug 8, 2025
The `numba-cuda` package was made a dependency in #462, but the conda recipe was missed.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Kyle Edwards (https://github.com/KyleFromNVIDIA)
  - James Lamb (https://github.com/jameslamb)
  - Bradley Dice (https://github.com/bdice)

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

Labels

feature request New feature or request non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Lots of warnings about Exception while handling op get_data

3 participants