cupyx.jit.rawkernel#

cupyx.jit.rawkernel(*, mode='cuda', device=False)[source]#

A decorator compiles a Python function into CUDA kernel.