Table Of Contents
Table Of Contents

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.