mxnet.rtc.CudaKernel¶

class mxnet.rtc.CudaKernel(handle, name, is_ndarray, dtypes)[source]¶

Constructs CUDA kernel. Should be created by CudaModule.get_kernel, not intended to be used by users.

__init__(handle, name, is_ndarray, dtypes)[source]¶: Initialize self. See help(type(self)) for accurate signature.

Methods

`__init__`(handle, name, is_ndarray, dtypes)	Initialize self.
`launch`(args, ctx, grid_dims, block_dims[, …])	Launch cuda kernel.