imate
C++/CUDA Reference
Loading...
Searching...
No Matches
cublas_impl.cu File Reference
#include "./cublas_impl.h"
#include "./cublas_impl_kernels.h"
#include "../_cu_arithmetics/cu_arithmetics.h"
#include <cuda_runtime.h>
#include "../_cu_definitions/cu_types.h"
#include <stdexcept>

Go to the source code of this file.

Namespaces

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

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} \).