mxnet::rtc::CudaModule::Kernel Class Reference

Cuda kernel. More...

#include <rtc.h>

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

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:

• include/mxnet/rtc.h