cuda-python icon indicating copy to clipboard operation
cuda-python copied to clipboard

WIP: Introducing `cuda.py`: pythonic access to CUDA core functionalities

Open leofang opened this issue 1 year ago • 1 comments

xref: #70

This is a highly experimental feature currently for preview purposes only, not for production use.

The current focus is centering around a correct, robust, and future-proof design of pythonic CUDA APIs, aiming to boost Python CUDA developers' productivity, among other things. This PR can be considered as a reference/prototype implementation for correctness checks and as a playground, with performance improvements coming next once the design is converged and approved.

More to come in the near future!

TODOs before merging (into the cuda_py branch just forked from main):

  • [ ] Expand/update PR description
  • [ ] Add README
  • [ ] Sync up with the (internal) design doc
  • [ ] Polish & check in test files
  • [ ] Polish & check in example files
  • [x] Update DLpack support
  • [ ] Rerun tests with CUDA 11 driver & CTK
  • [ ] ~~Fix (?) dependencies on the Python bindings~~
  • [x] Add GPUMemoryView prototype
  • [ ] ~~Add documentation~~

A separate issue will be created to track needed changes after merging (but before a beta release).

leofang avatar Sep 03 '24 18:09 leofang

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

copy-pr-bot[bot] avatar Sep 03 '24 18:09 copy-pr-bot[bot]

Update: To unblock parallel developments I will merge this PR soon (~tomorrow) so that subsequent PRs can start targeting the main branch of this repo, after checking in the following changes:

  • ~~GPUMemoryView -> StridedMemoryView~~ (done in commit a41a4b7)
  • Add a sample code for StridedMemoryView

leofang avatar Oct 09 '24 03:10 leofang

Merging! cc @ksimpson-work

leofang avatar Oct 10 '24 04:10 leofang