Skip to content

Instantly share code, notes, and snippets.

@pentschev

pentschev/cupy_simple_addition.py

Last active August 9, 2019 20:19
Show Gist options
  • Save pentschev/c046ed15a2f838bb9397e21cf18221cb to your computer and use it in GitHub Desktop.
Save pentschev/c046ed15a2f838bb9397e21cf18221cb to your computer and use it in GitHub Desktop.
Download ZIP
Blog Post - Parallelizing Custom CuPy Kernels with Dask - CuPy Simple Addition
Raw
cupy_simple_addition.py
import cupy
x = cupy.arange(4096 * 1024, dtype=cupy.float32).reshape((4096, 1024))
y = cupy.arange(1024, dtype=cupy.float32)
res_cupy = x + y
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment