Skip to content

Instantly share code, notes, and snippets.

@pentschev

pentschev/cupy_backed_dask_arrays.py

Created August 9, 2019 20:25
Show Gist options
  • Save pentschev/e468896e4e28c78b7e7161fc74460099 to your computer and use it in GitHub Desktop.
Save pentschev/e468896e4e28c78b7e7161fc74460099 to your computer and use it in GitHub Desktop.
Download ZIP
Blog Post - Parallelizing Custom CuPy Kernels with Dask - Create Dask CuPy-backed Dask Arrays
Raw
cupy_backed_dask_arrays.py
import dask.array as da
dx = da.from_array(x, chunks=(1024, 512), asarray=False)
dy = da.from_array(y, chunks=(1, 512), asarray=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment