![]() |
imate
C++/CUDA 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}
\). | |