Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Accept NumPy types in cuda.cooperative #3914

Open
brycelelbach opened this issue Feb 22, 2025 · 1 comment
Open

Accept NumPy types in cuda.cooperative #3914

brycelelbach opened this issue Feb 22, 2025 · 1 comment
Assignees
Labels
cuda.cooperative For all items related to the cuda.cooperative Python module

Comments

@brycelelbach
Copy link
Collaborator

Some cuda.cooperative interfaces only accept Numba types, not NumPy dtypes (e.g. only numba.int32 works, not numpy.int32). This is quite annoying, as many of the host-side APIs we use to create and manage arrays only accept NumPy dtypes, not Numba types.

Everywhere that we accept a type, NumPy types should be supported.

@github-project-automation github-project-automation bot moved this to Todo in CCCL Feb 22, 2025
@brycelelbach brycelelbach added the cuda.cooperative For all items related to the cuda.cooperative Python module label Feb 22, 2025
@tpn tpn self-assigned this Feb 24, 2025
@leofang
Copy link
Member

leofang commented Feb 27, 2025

FWIW, NumPy data types (ex: np.int32, which is a valid Python type) and dtypes (ex: np.dtype(np.int32)) should be both accepted.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cuda.cooperative For all items related to the cuda.cooperative Python module
Projects
Status: Todo
Development

No branches or pull requests

3 participants