kernelkit.kernel.copy_to_symbol

kernelkit.kernel.copy_to_symbol(module: RawModule, name: str, array)

Copy array to address on GPU, e.g. constant memory

Inspired by: cupy/cupy#1703 See also: https://docs.chainer.org/en/v1.3.1/_modules/cupy/creation/from_data.html

Parameters:

modulecp.RawModule

CUDA module

namestr

Name of the symbol to copy to (e.g. constant memory)

arraynp.ndarray

Array to copy to the symbol