Skip to content

[WIP] ENH: dask+cupy, dask+sparse etc. namespaces#270

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.

Sign up for GitHub

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

Jump to bottom
Draft
wants to merge 1 commit into
base:main
Choose a base branch
from
Draft

[WIP] ENH: dask+cupy, dask+sparse etc. namespaces #270

wants to merge 1 commit into from
+81 −7

Conversation

crusaderky
Copy link
Contributor

@crusaderkycrusaderky commented Mar 6, 2025•
edited
Loading

Very early prototype inspired by https://github.com/mdhaber/marray/

FYI @mdhaber@lucascolley

Demo

>>>importdask.arrayasda, cupyascp>>>fromarray_api_compatimportarray_namespace>>>x=da.from_array(cp.asarray(0)) >>>xp=array_namespace(x) >>>xp<module'array_api_compat.dask.array.cupy'>>>>y=xp.arange(10) >>>ydask.array<arange, shape=(10,), dtype=int64, chunksize=(10,), chunktype=cupy.ndarray>>>>type(y.compute()) cupy.ndarray>>>array_namespace(y) isxpTrue
@crusaderkycrusaderkyforce-pushed the dask_wrap_namespace branch 3 times, most recently from 7f39b03 to 592e31aCompareMarch 6, 2025 00:35
@crusaderkycrusaderky mentioned this pull request Mar 6, 2025
Merged
@crusaderkycrusaderkyforce-pushed the dask_wrap_namespace branch 3 times, most recently from 57d168e to f8aa551CompareApril 1, 2025 08:30
@crusaderkycrusaderkyforce-pushed the dask_wrap_namespace branch 2 times, most recently from 330c006 to 927c9a3CompareApril 10, 2025 09:45
@ev-brev-br added the enhancement New feature or request label Apr 15, 2025
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancementNew feature or request
2 participants
@crusaderky@ev-br
close