chainer.backends.cuda.raw

chainer.backends.cuda.raw(code, name, *args, **kwargs)[source]

Creates a raw kernel function.

This function uses memoize() to cache the resulting kernel object, i.e. the resulting kernel object is cached for each argument combination and CUDA device.

The arguments are the same as those for cupy.RawKernel.