cupyx.jit.rawkernel

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

A decorator compiles a Python function into CUDA kernel.