Skip to content

Support operations with side-effects (state) in cuda.compute#7008

Merged
shwina merged 9 commits intoNVIDIA:mainfrom
shwina:add-tests-and-bench-for-stateful-ops
Dec 19, 2025
Merged

Support operations with side-effects (state) in cuda.compute#7008
shwina merged 9 commits intoNVIDIA:mainfrom
shwina:add-tests-and-bench-for-stateful-ops

Conversation

@shwina
Copy link
Contributor

@shwina shwina commented Dec 18, 2025

Description

Merge after #6999

numba-cuda 0.23 introduces the ability to define kernels and device functions with side-effects (NVIDIA/numba-cuda#666).

This PR:

  • bumps our dependency to 0.23
  • fixes our function caching mechanism to handle functions referencing global arrays (state)
  • adds tests, benchmarks and examples for stateful operations

Checklist

  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@shwina shwina requested review from a team as code owners December 18, 2025 10:16
@github-project-automation github-project-automation bot moved this to Todo in CCCL Dec 18, 2025
@cccl-authenticator-app cccl-authenticator-app bot moved this from Todo to In Review in CCCL Dec 18, 2025
@shwina shwina force-pushed the add-tests-and-bench-for-stateful-ops branch from 7d0e2c6 to 3c51dab Compare December 18, 2025 10:19
@github-actions

This comment has been minimized.

@shwina shwina force-pushed the add-tests-and-bench-for-stateful-ops branch from 3c51dab to b14c101 Compare December 18, 2025 16:21
@github-actions

This comment has been minimized.

@shwina shwina force-pushed the add-tests-and-bench-for-stateful-ops branch from b14c101 to 355222a Compare December 19, 2025 09:59
@github-actions

This comment has been minimized.

@shwina shwina force-pushed the add-tests-and-bench-for-stateful-ops branch from 0a07c48 to 3f13f42 Compare December 19, 2025 13:32
Copy link
Contributor

@NaderAlAwar NaderAlAwar left a comment

Choose a reason for hiding this comment

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

great work!

@github-actions
Copy link
Contributor

🥳 CI Workflow Results

🟩 Finished in 57m 53s: Pass: 100%/48 | Total: 11h 10m | Max: 35m 01s

See results here.

@shwina shwina merged commit 2cd7c5f into NVIDIA:main Dec 19, 2025
62 of 63 checks passed
@github-project-automation github-project-automation bot moved this from In Review to Done in CCCL Dec 19, 2025
fbusato pushed a commit to fbusato/cccl that referenced this pull request Dec 23, 2025
…A#7008)

* Move algorithm cache to a central registry

* Add bench_select.py

* Add tests for stateful select and transform

* For the purposes of caching, hash DeviceArrayLike objects by pointer, shape, and dtype

* Update select benchmark

* Bump numba-cuda dependency to 0.23.0

* Add select example

* Lint

* Remove duplicate cache registry

---------

Co-authored-by: Ashwin Srinath <shwina@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Archived in project

Development

Successfully merging this pull request may close these issues.

2 participants