CuPy:Kernels

lastmod 2024-07-19created 2024-07-191 edits

How to write user-defined kernels in cupy.

Basics of elementwise kernels

Type-generic kernels

Raw argument specifiers

Texture memory

Reduction kernels

Raw kernels

Kernel arguments

Custom user types

Raw modules

Kernel fusion

JIT kernel definition

See also

• cupy
• CUDA:Kernel

Favorite site

• User-Defined Kernels — CuPy 13.2.0 documentation