mxlib
c++ tools for analyzing astronomical data and other tasks by Jared R. Males. [git repo]
|
A template interface to cuBlas. More...
A template interface to cuBlas.
Definition in file templateCublas.hpp.
Go to the source code of this file.
Namespaces | |
mx | |
The mxlib c++ namespace. | |
Functions | |
template<typename floatT > | |
cublasStatus_t | mx::cuda::cublasTscal (cublasHandle_t handle, int n, const floatT *alpha, floatT *x, int incx) |
Multiplies a vector by a scalar, overwriting the vector with the result. More... | |
template<typename floatT > | |
cublasStatus_t | mx::cuda::cublasTaxpy (cublasHandle_t handle, int n, const floatT *alpha, const floatT *x, int incx, floatT *y, int incy) |
Multiplies a vector by a scalar, adding it to a second vector which is overwritten by the result. More... | |
template<typename dataT1 , typename dataT2 > | |
cudaError_t | mx::cuda::elementwiseXxY (dataT1 *x, dataT2 *y, int size) |
Calculates the element-wise product of two vectors, storing the result in the first. More... | |
template<typename floatT > | |
cublasStatus_t | mx::cuda::cublasTgemv (cublasHandle_t handle, cublasOperation_t trans, int m, int n, const floatT *alpha, const floatT *A, int lda, const floatT *x, int incx, const floatT *beta, floatT *y, int incy) |
Perform a matrix-vector multiplication. More... | |
template<typename floatT > | |
cublasStatus_t | mx::cuda::cublasTgemv (cublasHandle_t handle, cublasOperation_t trans, int m, int n, const floatT *alpha, const floatT *A, const floatT *x, const floatT *beta, floatT *y) |
Perform a matrix-vector multiplication for stride-less arrays. More... | |