-
-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Closed
Description
Dask contains parallel algorithms for parallel arrays based on the numpy syntax. While commonly used with numpy arrays they also work with other libraries that mimic subsets of the numpy ndarray API.
The CuPy library implements a NumPy API compatible GPU ndarray. It would be interesting to see how these two libraries interact, where performance bottlenecks are, and under what conditions they may be effectively used together.
To experiment with this one would want to do a few things:
- Register cupy operations for
concatenateandtensordot. Examples doing this for thesparselibrary are at the top of dask/array/core.py - Create a GPU array using
da.from_array, orda.from_delayedandda.concatenate/stack. See http://dask.pydata.org/en/latest/array-creation.html - If you want to use distributed computing then ensure that cupy ndarrays can be efficiently serialized, probably by implementing
__setstate__and__getstate__methods for them if they don't already exist.
This should be sufficient to get something up and running.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels