Skip to content

async NDProcessor#1026

Merged
kushalkolar merged 12 commits into
ndwidgetfrom
async-ndp
Apr 9, 2026
Merged

async NDProcessor#1026
kushalkolar merged 12 commits into
ndwidgetfrom
async-ndp

Conversation

@kushalkolar

Copy link
Copy Markdown
Member

First time I'm doing async stuff, this is an adventure.

@kushalkolar kushalkolar requested a review from clewis7 as a code owner April 8, 2026 06:46
@kushalkolar kushalkolar changed the title async NDProcessor established, NOT TESTED async NDProcessor Apr 8, 2026
@kushalkolar

Copy link
Copy Markdown
Member Author

I just realized a few lines of code and GPU arrays can be fully async too, with the transfers to CPU only at the end for all of them :D

cupy, torch and jax all implement __cuda_array_interface__

Check if __cuda_array_interface__ is implemented, and if so treat that similar to a future. To resolve it just convert to cupy and send to CPU!

@kushalkolar kushalkolar merged commit b918626 into ndwidget Apr 9, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant