Skip to content

Instantly share code, notes, and snippets.

@pentschev

pentschev/map_blocks_dispatch_add_broadcast.py

Created August 9, 2019 20:26
Show Gist options
  • Save pentschev/5c37d037c683ae06cbd1727cce2fcd23 to your computer and use it in GitHub Desktop.
Save pentschev/5c37d037c683ae06cbd1727cce2fcd23 to your computer and use it in GitHub Desktop.
Download ZIP
Blog Post - Parallelizing Custom CuPy Kernels with Dask - Map Blocks
Raw
map_blocks_dispatch_add_broadcast.py
res = da.map_blocks(dispatch_add_broadcast, dx, dy, dtype=cupy.float32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment