NVIDIA / NVIDIA/cccl

[EPIC] Expose `cuda::ptx::` functions to Python in `cuda.ptx` module

Open
#6,459 2 comments 1 reaction 0 assignees View on GitHub
Dominant language
C++
Stars
2.5k
Forks
486
Avg merge
2d 6h
Merged PRs (30d)
295

Description

### Is this a duplicate?

- [x] I confirmed there appear to be no [duplicate issues](https://github.com/NVIDIA/cccl/issues) for this request and that I agree to the [Code of Conduct](CODE_OF_CONDUCT.md)

### Area

General CCCL

### Is your feature request related to a problem? Please describe.

As a Python developer using `numba-cuda` to author SOL kernels in Python, I'd like to have access to low-level building blocks for convenient access to PTX operations.

### Describe the solution you'd like

In CUDA C++, the facilities in [`cuda::ptx::`](https://nvidia.github.io/cccl/libcudacxx/ptx_api.html) give me access to these building blocks. I'd like the equivalent in a new `cuda.cccl` Python module, `cuda.ptx` (name TBD).

### Describe alternatives you've considered

_No response_

### Additional context

The implementation here should be morally equivalent to how we expose CUB C++ device-side APIs in `cuda.coop`. In an ideal world, [numbast ](https://github.com/NVIDIA/numbast), could be used to just fully automate generating these bindings as much as possible.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.