Cuda runtime compile module. More...

#include <rtc.h>

Collaboration diagram for mxnet::rtc::CudaModule:

Classes
struct	ArgType
	cuda kernel argument descriptor More...

class	Kernel
	Cuda kernel. More...

Public Member Functions
	CudaModule (const char *source, const std::vector< std::string > &options, const std::vector< std::string > &exports)
	CudaModule constructor. More...

std::shared_ptr< Kernel >	GetKernel (const std::string &name, const std::vector< ArgType > &signature)
	Get cuda kernal from module by name. More...

Detailed Description

Cuda runtime compile module.

Constructor & Destructor Documentation

mxnet::rtc::CudaModule::CudaModule	(	const char *	source,
		const std::vector< std::string > &	options,
		const std::vector< std::string > &	exports
	)

inline

CudaModule constructor.

Parameters

source	cuda source code.
exports	export symbols before mangling.

std::shared_ptr<Kernel> mxnet::rtc::CudaModule::GetKernel	(	const std::string &	name,
		const std::vector< ArgType > &	signature
	)

Get cuda kernal from module by name.

Parameters

name	kernel name
signature	kernel signature

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