cublas_impl_kernels.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	cublas_impl_kernels
	Templated kernel code for implenentations of several BLAS-type functions in CUDA.

Macros
#define	RESTRICT

Functions
template<typename DataType , typename ComputeType , unsigned int block_size>
__global__ void	cublas_impl_kernels::cublasTgemv_kernel (const bool trans, const int m, const int n, const DataType alpha, const DataType *RESTRICT A, const int lda, const DataType *RESTRICT x, const int incx, const DataType beta, DataType *RESTRICT y, const int incy)
	Performs the operation \( \boldsymbol{y} = \alpha \mathbf{A} \boldsymbol{x} + \beta \boldsymbol{y} \).
template<typename DataType >
__global__ void	cublas_impl_kernels::cublasTcopy_kernel (const int n, const DataType *RESTRICT x, const int incx, DataType *RESTRICT y, const int incy)
	Performs \( \boldsymbol{y} = \boldsymbol{x} \).
template<typename DataType >
__global__ void	cublas_impl_kernels::cublasTaxpy_kernel (const int n, const DataType alpha, const DataType *RESTRICT x, const int incx, DataType *RESTRICT y, const int incy)
	Performs \( \boldsymbol{y} = \alpha \boldsymbol{x} + \boldsymbol{y} \).
template<typename DataType , typename ComputeType , unsigned int block_size>
__global__ void	cublas_impl_kernels::cublasTdot_kernel (const int n, const DataType *RESTRICT x, const int incx, const DataType *RESTRICT y, const int incy, ComputeType *RESTRICT result)
	Computes \( a = \boldsymbol{x} \cdot \boldsymbol{y} \).
template<typename DataType , typename ComputeType , unsigned int block_size>
__global__ void	cublas_impl_kernels::cublasTnrm2_kernel (const int n, const DataType *RESTRICT x, const int incx, ComputeType *RESTRICT result)
	Computes \( a = \boldsymbol{x} \cdot \boldsymbol{x} \).
template<typename DataType >
__global__ void	cublas_impl_kernels::cublasTscal_kernel (const int n, const DataType alpha, DataType *RESTRICT x, const int incx)
	Performs \( \boldsymbol{x} = \alpha \boldsymbol{x} \).

Macro Definition Documentation

RESTRICT

#define RESTRICT

Definition at line 26 of file cublas_impl_kernels.h.