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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
| //#include "mini_blas.h"
| #ifdef __cplusplus
| #define PUT_IN_REGISTER
| #else
| #define PUT_IN_REGISTER register
| #endif
|
| void cpu_gemm_nn(int TA, int TB, int M, int N, int K, float ALPHA,
| float *A, int lda,
| float *B, int ldb,
| float BETA,
| float *C, int ldc)
| {
| int i,j,k;
| for(i = 0; i < M; ++i){
| for(k = 0; k < K; ++k){
| PUT_IN_REGISTER float A_PART = ALPHA * A[i * lda + k];
| for(j = 0; j < N; ++j){
| C[i*ldc+j] += A_PART*B[k*ldb+j];
| }
| }
| }
| }
|
| void cpu_gemm_nt(int TA, int TB, int M, int N, int K, float ALPHA,
| float *A, int lda,
| float *B, int ldb,
| float BETA,
| float *C, int ldc)
| {
| int i,j,k;
| for(i = 0; i < M; ++i){
| for(j = 0; j < N; ++j){
| PUT_IN_REGISTER float sum = 0;
| for(k = 0; k < K; ++k){
| sum += ALPHA*A[i*lda+k]*B[k+j*ldb];
| }
| C[i*ldc+j] += sum;
| }
| }
| }
|
| void cpu_gemm_tn(int TA, int TB, int M, int N, int K, float ALPHA,
| float *A, int lda,
| float *B, int ldb,
| float BETA,
| float *C, int ldc)
| {
| int i,j,k;
| for(i = 0; i < M; ++i){
| for(k = 0; k < K; ++k){
| PUT_IN_REGISTER float A_PART = ALPHA * A[k * lda + i];
| for(j = 0; j < N; ++j){
| C[i*ldc+j] += A_PART*B[k*ldb+j];
| }
| }
| }
| }
| void cpu_gemm_tt(int TA, int TB, int M, int N, int K, float ALPHA,
| float *A, int lda,
| float *B, int ldb,
| float BETA,
| float *C, int ldc)
| {
| int i,j,k;
| for(i = 0; i < M; ++i){
| for(j = 0; j < N; ++j){
| for(k = 0; k < K; ++k){
| C[i*ldc+j] += ALPHA*A[i+k*lda]*B[k+j*ldb];
| }
| }
| }
| }
|
|
| void cpu_gemm(int TA, int TB, int M, int N, int K, float ALPHA,
| float *A, int lda,
| float *B, int ldb,
| float BETA,
| float *C, int ldc)
| {
| int i, j;
| for(i = 0; i < M; ++i){
| for(j = 0; j < N; ++j){
| C[i*ldc + j] *= BETA;
| }
| }
| if(!TA && !TB)
| cpu_gemm_nn( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
| else if(TA && !TB)
| cpu_gemm_tn( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
| else if(!TA && TB)
| cpu_gemm_nt( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
| else
| cpu_gemm_tt( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc);
| }
|
|
