vectorization-inl.h File Reference

#include <mxnet/base.h>
#include <sstream>
#include <string>
#include <vector>
#include <algorithm>
#include "../rtc.h"
#include "../../utils.h"

Include dependency graph for vectorization-inl.h:

Go to the source code of this file.

Namespaces
	mxnet
	namespace of mxnet
	mxnet::common
	mxnet::common::cuda
	common utils for cuda
	mxnet::common::cuda::rtc

Functions
template<typename Params >
void	mxnet::common::cuda::rtc::VectorizedKernelRTCLauncher (const std::string &parameters, const std::string &kernel_name, const std::string &code, int nvec, const index_t lead_dim, const index_t other_dim, mshadow::Stream< gpu > *s, const Params params, const std::vector< TBlob > &inputs, const std::vector< TBlob > &outputs, const int dev_id, const int lead_input_num=0, const index_t blocks=0)
	Launcher helper for the kernels using vectorization. More...

Variables
const char	mxnet::common::cuda::rtc::vectorization_support_string []