-
Notifications
You must be signed in to change notification settings - Fork 19
/
cublasWrappers.cuh
36 lines (29 loc) · 1.21 KB
/
cublasWrappers.cuh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#ifndef CUBLASWRAPPERS_CUH
#define CUBLASWRAPPERS_CUH
#include "cublas_v2.h"
/**************/
/* CUBLASTDOT */
/**************/
cublasStatus_t cublasTdot(cublasHandle_t, int, const float *, int, const float *, int, float *);
cublasStatus_t cublasTdot(cublasHandle_t, int, const double *, int, const double *, int, double *);
/***************/
/* CUBLASTNRM2 */
/***************/
cublasStatus_t cublasTnrm2(cublasHandle_t, int, const float *, int, float *);
cublasStatus_t cublasTnrm2(cublasHandle_t, int, const double *, int, double *);
/***************/
/* CUBLASTSCAL */
/***************/
cublasStatus_t cublasTscal(cublasHandle_t, int, const float *, float *, int);
cublasStatus_t cublasTscal(cublasHandle_t, int, const double *, double *, int);
/***************/
/* CUBLASTAXPY */
/***************/
cublasStatus_t cublasTaxpy(cublasHandle_t, int, const float *, const float *, int, float *, int);
cublasStatus_t cublasTaxpy(cublasHandle_t, int, const double *, const double *, int, double *, int);
/***************/
/* CUBLASTCOPY */
/***************/
cublasStatus_t cublasTcopy(cublasHandle_t, int, const float *, int, float *, int);
cublasStatus_t cublasTcopy(cublasHandle_t, int, const double *, int, double *, int);
#endif