mxnet.rtcΒΆ
Interface to runtime cuda kernel compile module.
Classes
CudaKernel (handle, name, is_ndarray, dtypes) |
Constructs CUDA kernel. |
CudaModule (source[, options, exports]) |
Compile and run CUDA code from Python. |
Interface to runtime cuda kernel compile module.
Classes
CudaKernel (handle, name, is_ndarray, dtypes) |
Constructs CUDA kernel. |
CudaModule (source[, options, exports]) |
Compile and run CUDA code from Python. |