-
Notifications
You must be signed in to change notification settings - Fork 91
/
gemm.h
65 lines (61 loc) · 1.46 KB
/
gemm.h
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#ifndef _DEFINED_SCOMPLEX
#define _DEFINED_SCOMPLEX
typedef struct
{
float real;
float imag;
} scomplex;
#endif
#ifndef _DEFINED_DCOMPLEX
#define _DEFINED_DCOMPLEX
typedef struct
{
double real;
double imag;
} dcomplex;
#endif
extern void
dgemm_nn(I m,
I n,
I k,
double alpha,
double *A,
I rs_A,
I cs_A,
double *B,
I rs_B,
I cs_B,
double beta,
double *C,
I rs_C,
I cs_C);
extern void
igemm_nn(I m,
I n,
I k,
I alpha,
I *A,
I rs_A,
I cs_A,
I *B,
I rs_B,
I cs_B,
I beta,
double *C,
I rs_C,
I cs_C);
extern void
zgemm_nn(I m,
I n,
I k,
dcomplex alpha,
dcomplex *A,
I rs_A,
I cs_A,
dcomplex *B,
I rs_B,
I cs_B,
dcomplex beta,
dcomplex *C,
I rs_C,
I cs_C);