All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
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...


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 ( )

kernel interface signature

Friends And Related Function Documentation

friend class CudaModule

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