-
Notifications
You must be signed in to change notification settings - Fork 10
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
Add __cuda_array_interface__
to itk::CudaImage
#10
Comments
Hi @thewtex For supporting |
Also here's some code from CuPy for consuming/producing |
LucasGandel
added a commit
to LucasGandel/ITKCudaCommon
that referenced
this issue
Aug 9, 2024
Adapt ITK's PyBuffer approach to inject __cuda_array_interface__ in wrapped code. Closes RTKConsortium#10
6 tasks
Soon addressed by #42. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
A
__cuda_array_interface__
method could prevent CPU/GPU memory transfer whenitk.CudaImage
is used withcupy
,pytorch
, etc.@gigony you shared a nice reference of how this is implemented in cuCIM, but I am having a hard time finding it in a quick grep of the cucim repository. Any pointers?
Edit: Found it :-):
https://github.com/rapidsai/cucim/blob/4061b7e00c5b10f18f04f3c356b32a42c081de4f/python/pybind11/cucim_py.cpp#L628-L631
CC @SimonRit @LucasGandel
The text was updated successfully, but these errors were encountered: