imate
C++/CUDA Reference
|
A collection of templates to wrapper cublas functions. More...
Functions | |
template<> | |
cublasStatus_t | cublasXgemv< float > (cublasHandle_t handle, cublasOperation_t trans, int m, int n, const float *alpha, const float *A, int lda, const float *x, int incx, const float *beta, float *y, int incy) |
A template wrapper for cublasSgemv . More... | |
template<> | |
cublasStatus_t | cublasXgemv< double > (cublasHandle_t handle, cublasOperation_t trans, int m, int n, const double *alpha, const double *A, int lda, const double *x, int incx, const double *beta, double *y, int incy) |
A template wrapper for cublasDgemv . More... | |
template<> | |
cublasStatus_t | cublasXcopy< float > (cublasHandle_t handle, int n, const float *x, int incx, float *y, int incy) |
A template wrapper for cublasScopy . More... | |
template<> | |
cublasStatus_t | cublasXcopy< double > (cublasHandle_t handle, int n, const double *x, int incx, double *y, int incy) |
A template wrapper for cublasDcopy . More... | |
template<> | |
cublasStatus_t | cublasXaxpy< float > (cublasHandle_t handle, int n, const float *alpha, const float *x, int incx, float *y, int incy) |
A template wrapper for cublasSaxpy . More... | |
template<> | |
cublasStatus_t | cublasXaxpy< double > (cublasHandle_t handle, int n, const double *alpha, const double *x, int incx, double *y, int incy) |
A template wrapper for cublasDaxpy . More... | |
template<> | |
cublasStatus_t | cublasXdot< float > (cublasHandle_t handle, int n, const float *x, int incx, const float *y, int incy, float *result) |
A template wrapper for cublasSdot . More... | |
template<> | |
cublasStatus_t | cublasXdot< double > (cublasHandle_t handle, int n, const double *x, int incx, const double *y, int incy, double *result) |
A template wrapper for cublasDdot . More... | |
template<> | |
cublasStatus_t | cublasXnrm2< float > (cublasHandle_t handle, int n, const float *x, int incx, float *result) |
A template wrapper to cublasSnrm2 . More... | |
template<> | |
cublasStatus_t | cublasXnrm2< double > (cublasHandle_t handle, int n, const double *x, int incx, double *result) |
A template wrapper to cublasDnrm2 . More... | |
template<> | |
cublasStatus_t | cublasXscal< float > (cublasHandle_t handle, int n, const float *alpha, float *x, int incx) |
A template wrapper for cublasSscal . More... | |
template<> | |
cublasStatus_t | cublasXscal< double > (cublasHandle_t handle, int n, const double *alpha, double *x, int incx) |
A template wrapper for cublasDscal . More... | |
template<typename DataType > | |
cublasStatus_t | cublasXgemv (cublasHandle_t handle, cublasOperation_t trans, int m, int n, const DataType *alpha, const DataType *A, int lda, const DataType *x, int incx, const DataType *beta, DataType *y, int incy) |
template<typename DataType > | |
cublasStatus_t | cublasXcopy (cublasHandle_t handle, int n, const DataType *x, int incx, DataType *y, int incy) |
template<typename DataType > | |
cublasStatus_t | cublasXaxpy (cublasHandle_t handle, int n, const DataType *alpha, const DataType *x, int incx, DataType *y, int incy) |
template<typename DataType > | |
cublasStatus_t | cublasXdot (cublasHandle_t handle, int n, const DataType *x, int incx, const DataType *y, int incy, DataType *result) |
template<typename DataType > | |
cublasStatus_t | cublasXnrm2 (cublasHandle_t handle, int n, const DataType *x, int incx, DataType *result) |
template<typename DataType > | |
cublasStatus_t | cublasXscal (cublasHandle_t handle, int n, const DataType *alpha, DataType *x, int incx) |
A collection of templates to wrapper cublas functions.
cu
file is wrapped inside the namepsace clause. This is not necessary in general, however, it is needed to avoid the old gcc compiler error (this is a gcc bug) which complains "no instance of function template matches
the argument list const float". cublasStatus_t cublas_interface::cublasXaxpy | ( | cublasHandle_t | handle, |
int | n, | ||
const DataType * | alpha, | ||
const DataType * | x, | ||
int | incx, | ||
DataType * | y, | ||
int | incy | ||
) |
Referenced by cuVectorOperations< DataType >::subtract_scaled_vector().
cublasStatus_t cublas_interface::cublasXaxpy< double > | ( | cublasHandle_t | handle, |
int | n, | ||
const double * | alpha, | ||
const double * | x, | ||
int | incx, | ||
double * | y, | ||
int | incy | ||
) |
A template wrapper for cublasDaxpy
.
Definition at line 156 of file cublas_interface.cu.
References cublasDaxpy().
cublasStatus_t cublas_interface::cublasXaxpy< float > | ( | cublasHandle_t | handle, |
int | n, | ||
const float * | alpha, | ||
const float * | x, | ||
int | incx, | ||
float * | y, | ||
int | incy | ||
) |
A template wrapper for cublasSaxpy
.
Definition at line 135 of file cublas_interface.cu.
References cublasSaxpy().
cublasStatus_t cublas_interface::cublasXcopy | ( | cublasHandle_t | handle, |
int | n, | ||
const DataType * | x, | ||
int | incx, | ||
DataType * | y, | ||
int | incy | ||
) |
Referenced by cuVectorOperations< DataType >::copy_scaled_vector(), and cuVectorOperations< DataType >::copy_vector().
cublasStatus_t cublas_interface::cublasXcopy< double > | ( | cublasHandle_t | handle, |
int | n, | ||
const double * | x, | ||
int | incx, | ||
double * | y, | ||
int | incy | ||
) |
A template wrapper for cublasDcopy
.
Definition at line 115 of file cublas_interface.cu.
References cublasDcopy().
cublasStatus_t cublas_interface::cublasXcopy< float > | ( | cublasHandle_t | handle, |
int | n, | ||
const float * | x, | ||
int | incx, | ||
float * | y, | ||
int | incy | ||
) |
A template wrapper for cublasScopy
.
Definition at line 95 of file cublas_interface.cu.
References cublasScopy().
cublasStatus_t cublas_interface::cublasXdot | ( | cublasHandle_t | handle, |
int | n, | ||
const DataType * | x, | ||
int | incx, | ||
const DataType * | y, | ||
int | incy, | ||
DataType * | result | ||
) |
Referenced by cuVectorOperations< DataType >::inner_product().
cublasStatus_t cublas_interface::cublasXdot< double > | ( | cublasHandle_t | handle, |
int | n, | ||
const double * | x, | ||
int | incx, | ||
const double * | y, | ||
int | incy, | ||
double * | result | ||
) |
A template wrapper for cublasDdot
.
Definition at line 198 of file cublas_interface.cu.
References cublasDdot().
cublasStatus_t cublas_interface::cublasXdot< float > | ( | cublasHandle_t | handle, |
int | n, | ||
const float * | x, | ||
int | incx, | ||
const float * | y, | ||
int | incy, | ||
float * | result | ||
) |
A template wrapper for cublasSdot
.
Definition at line 177 of file cublas_interface.cu.
References cublasSdot().
cublasStatus_t cublas_interface::cublasXgemv | ( | cublasHandle_t | handle, |
cublasOperation_t | trans, | ||
int | m, | ||
int | n, | ||
const DataType * | alpha, | ||
const DataType * | A, | ||
int | lda, | ||
const DataType * | x, | ||
int | incx, | ||
const DataType * | beta, | ||
DataType * | y, | ||
int | incy | ||
) |
Referenced by cuMatrixOperations< DataType >::dense_matvec(), cuMatrixOperations< DataType >::dense_matvec_plus(), cuMatrixOperations< DataType >::dense_transposed_matvec(), and cuMatrixOperations< DataType >::dense_transposed_matvec_plus().
cublasStatus_t cublas_interface::cublasXgemv< double > | ( | cublasHandle_t | handle, |
cublasOperation_t | trans, | ||
int | m, | ||
int | n, | ||
const double * | alpha, | ||
const double * | A, | ||
int | lda, | ||
const double * | x, | ||
int | incx, | ||
const double * | beta, | ||
double * | y, | ||
int | incy | ||
) |
A template wrapper for cublasDgemv
.
Definition at line 67 of file cublas_interface.cu.
cublasStatus_t cublas_interface::cublasXgemv< float > | ( | cublasHandle_t | handle, |
cublasOperation_t | trans, | ||
int | m, | ||
int | n, | ||
const float * | alpha, | ||
const float * | A, | ||
int | lda, | ||
const float * | x, | ||
int | incx, | ||
const float * | beta, | ||
float * | y, | ||
int | incy | ||
) |
A template wrapper for cublasSgemv
.
Definition at line 40 of file cublas_interface.cu.
cublasStatus_t cublas_interface::cublasXnrm2 | ( | cublasHandle_t | handle, |
int | n, | ||
const DataType * | x, | ||
int | incx, | ||
DataType * | result | ||
) |
Referenced by cuVectorOperations< DataType >::euclidean_norm().
cublasStatus_t cublas_interface::cublasXnrm2< double > | ( | cublasHandle_t | handle, |
int | n, | ||
const double * | x, | ||
int | incx, | ||
double * | result | ||
) |
A template wrapper to cublasDnrm2
.
Definition at line 238 of file cublas_interface.cu.
References cublasDnrm2().
cublasStatus_t cublas_interface::cublasXnrm2< float > | ( | cublasHandle_t | handle, |
int | n, | ||
const float * | x, | ||
int | incx, | ||
float * | result | ||
) |
A template wrapper to cublasSnrm2
.
Definition at line 219 of file cublas_interface.cu.
References cublasSnrm2().
cublasStatus_t cublas_interface::cublasXscal | ( | cublasHandle_t | handle, |
int | n, | ||
const DataType * | alpha, | ||
DataType * | x, | ||
int | incx | ||
) |
Referenced by cuVectorOperations< DataType >::copy_scaled_vector(), and cuVectorOperations< DataType >::normalize_vector_in_place().
cublasStatus_t cublas_interface::cublasXscal< double > | ( | cublasHandle_t | handle, |
int | n, | ||
const double * | alpha, | ||
double * | x, | ||
int | incx | ||
) |
A template wrapper for cublasDscal
.
Definition at line 276 of file cublas_interface.cu.
References cublasDscal().
cublasStatus_t cublas_interface::cublasXscal< float > | ( | cublasHandle_t | handle, |
int | n, | ||
const float * | alpha, | ||
float * | x, | ||
int | incx | ||
) |
A template wrapper for cublasSscal
.
Definition at line 257 of file cublas_interface.cu.
References cublasSscal().