mxnet
Public Member Functions | Friends | List of all members
mxnet::rtc::CudaModule::Kernel Class Reference

Cuda kernel. More...

#include <rtc.h>

Collaboration diagram for mxnet::rtc::CudaModule::Kernel:
Collaboration graph

Public Member Functions

void Launch (const Context &ctx, const std::vector< dmlc::any > &args, uint32_t grid_dim_x, uint32_t grid_dim_y, uint32_t grid_dim_z, uint32_t block_dim_x, uint32_t block_dim_y, uint32_t block_dim_z, uint32_t shared_mem)
 Launch the kernel. More...
 
const std::vector< ArgType > & signature ()
 kernel interface signature More...
 

Friends

class CudaModule
 

Detailed Description

Cuda kernel.

Member Function Documentation

void mxnet::rtc::CudaModule::Kernel::Launch ( const Context ctx,
const std::vector< dmlc::any > &  args,
uint32_t  grid_dim_x,
uint32_t  grid_dim_y,
uint32_t  grid_dim_z,
uint32_t  block_dim_x,
uint32_t  block_dim_y,
uint32_t  block_dim_z,
uint32_t  shared_mem 
)

Launch the kernel.

const std::vector<ArgType>& mxnet::rtc::CudaModule::Kernel::signature ( )
inline

kernel interface signature

Friends And Related Function Documentation

friend class CudaModule
friend

The documentation for this class was generated from the following file: