Table Of Contents
Table Of Contents

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.