imate
C++/CUDA Reference
Loading...
Searching...
No Matches
cublas_impl.h File Reference
#include <cublas_v2.h>
Include dependency graph for cublas_impl.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  cublas_impl
 Templated implenentations of several BLAS-type functions in CUDA.
 

Macros

#define RESTRICT
 

Functions

template<typename DataType , typename ComputeType >
cudaError_t cublas_impl::cublasTgemv (cublasOperation_t trans, int m, int n, const DataType *RESTRICT alpha, const DataType *RESTRICT A, int lda, const DataType *RESTRICT x, int incx, const DataType *RESTRICT beta, DataType *RESTRICT y, int incy)
 Performs \( \boldsymbol{y} = \alpha \text{op}(\mathbf{A}) \boldsymbol{x} + \beta \boldsymbol{y} \).
 
template<typename DataType >
cudaError_t cublas_impl::cublasTcopy (int n, const DataType *RESTRICT x, int incx, DataType *RESTRICT y, int incy)
 Performs \( \boldsymbol{y} = \boldsymbol{x} \).
 
template<typename DataType >
cudaError_t cublas_impl::cublasTaxpy (int n, const DataType *RESTRICT alpha, const DataType *RESTRICT x, int incx, DataType *RESTRICT y, int incy)
 Performs \( \boldsymbol{y} = \alpha \boldsymbol{x} + \boldsymbol{y} \).
 
template<typename DataType , typename ComputeType >
cudaError_t cublas_impl::cublasTdot (int n, const DataType *RESTRICT x, int incx, const DataType *RESTRICT y, int incy, DataType *RESTRICT result)
 Computes \( a = \boldsymbol{x} \cdot \boldsymbol{y} \).
 
template<typename DataType , typename ComputeType >
cudaError_t cublas_impl::cublasTnrm2 (int n, const DataType *RESTRICT x, int incx, DataType *RESTRICT result)
 Computes \( a = \boldsymbol{x} \cdot \boldsymbol{x} \).
 
template<typename DataType >
cudaError_t cublas_impl::cublasTscal (int n, const DataType *RESTRICT alpha, DataType *RESTRICT x, int incx)
 Performs \( \boldsymbol{x} = \alpha \boldsymbol{x} \).
 

Macro Definition Documentation

◆ RESTRICT

#define RESTRICT

Definition at line 50 of file cublas_impl.h.