mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Classes | Public Member Functions | List of all members
mxnet::rtc::CudaModule Class Reference

Cuda runtime compile module. More...

#include <rtc.h>

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

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< KernelGetKernel (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
sourcecuda source code.
exportsexport symbols before mangling.

Member Function Documentation

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
namekernel name
signaturekernel signature
Returns
shared pointer to cuda kernel

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