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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#include "stdlib.h"
#include "util.h"
#include "dataset.h"
void __attribute__((noinline)) matmul(const int coreid, const int ncores, const int lda, const data_t A[], const data_t B[], data_t C[] )
{
// ***************************** //
// **** ADD YOUR CODE HERE ***** //
// ***************************** //
//
// feel free to make a separate function for MI and MSI versions.
int j2, i2, k2, j, i, k;
int tmpC00, tmpC01, tmpC02, tmpC03, tmpC04, tmpC05, tmpC06, tmpC07;
int tmpC10, tmpC11, tmpC12, tmpC13, tmpC14, tmpC15, tmpC16, tmpC17;
int jBLOCK = 32;
int iBLOCK = 16;
int kBLOCK = 32;
static __thread int tB[4096]; //__thread
int startInd = coreid*(lda/ncores);
int endInd = (coreid+1)*(lda/ncores);
//tranpose B (block?)
for (i = 0; i < lda; i += 2) {
for (j = startInd; j < endInd; j += 2) {
tB[j*lda + i] = B[i*lda + j];
tB[(j + 1)*lda + i] = B[i*lda + j + 1];
tB[j*lda + i + 1] = B[(i + 1)*lda + j];
tB[(j + 1)*lda + i + 1] = B[(i + 1)*lda + j + 1];
}
barrier(ncores);
}
// compute C[j*n + i] += A[j*n + k] + Btranspose[i*n + k]
for ( j2 = 0; j2 < lda; j2 += jBLOCK )
for ( i2 = startInd; i2 < endInd; i2 += iBLOCK )
for ( j = j2; j < j2 + jBLOCK; j += 2 )
for ( k2 = 0; k2 < lda; k2 += kBLOCK )
for ( i = i2; i < i2 + iBLOCK; i += 8) {
tmpC00 = C[j*lda + i + 0]; tmpC10 = C[(j + 1)*lda + i + 0];
tmpC01 = C[j*lda + i + 1]; tmpC11 = C[(j + 1)*lda + i + 1];
tmpC02 = C[j*lda + i + 2]; tmpC12 = C[(j + 1)*lda + i + 2];
tmpC03 = C[j*lda + i + 3]; tmpC13 = C[(j + 1)*lda + i + 3];
tmpC04 = C[j*lda + i + 4]; tmpC14 = C[(j + 1)*lda + i + 4];
tmpC05 = C[j*lda + i + 5]; tmpC15 = C[(j + 1)*lda + i + 5];
tmpC06 = C[j*lda + i + 6]; tmpC16 = C[(j + 1)*lda + i + 6];
tmpC07 = C[j*lda + i + 7]; tmpC17 = C[(j + 1)*lda + i + 7];
for ( k = k2; k < k2 + kBLOCK; k += 4) {
tmpC00 += A[j*lda + k] * tB[(i + 0)*lda + k];
tmpC01 += A[j*lda + k] * tB[(i + 1)*lda + k];
tmpC02 += A[j*lda + k] * tB[(i + 2)*lda + k];
tmpC03 += A[j*lda + k] * tB[(i + 3)*lda + k];
tmpC04 += A[j*lda + k] * tB[(i + 4)*lda + k];
tmpC05 += A[j*lda + k] * tB[(i + 5)*lda + k];
tmpC06 += A[j*lda + k] * tB[(i + 6)*lda + k];
tmpC07 += A[j*lda + k] * tB[(i + 7)*lda + k];
tmpC10 += A[(j + 1)*lda + k] * tB[(i + 0)*lda + k];
tmpC11 += A[(j + 1)*lda + k] * tB[(i + 1)*lda + k];
tmpC12 += A[(j + 1)*lda + k] * tB[(i + 2)*lda + k];
tmpC13 += A[(j + 1)*lda + k] * tB[(i + 3)*lda + k];
tmpC14 += A[(j + 1)*lda + k] * tB[(i + 4)*lda + k];
tmpC15 += A[(j + 1)*lda + k] * tB[(i + 5)*lda + k];
tmpC16 += A[(j + 1)*lda + k] * tB[(i + 6)*lda + k];
tmpC17 += A[(j + 1)*lda + k] * tB[(i + 7)*lda + k];
tmpC00 += A[j*lda + k + 1] * tB[(i + 0)*lda + k + 1];
tmpC01 += A[j*lda + k + 1] * tB[(i + 1)*lda + k + 1];
tmpC02 += A[j*lda + k + 1] * tB[(i + 2)*lda + k + 1];
tmpC03 += A[j*lda + k + 1] * tB[(i + 3)*lda + k + 1];
tmpC04 += A[j*lda + k + 1] * tB[(i + 4)*lda + k + 1];
tmpC05 += A[j*lda + k + 1] * tB[(i + 5)*lda + k + 1];
tmpC06 += A[j*lda + k + 1] * tB[(i + 6)*lda + k + 1];
tmpC07 += A[j*lda + k + 1] * tB[(i + 7)*lda + k + 1];
tmpC10 += A[(j + 1)*lda + k + 1] * tB[(i + 0)*lda + k + 1];
tmpC11 += A[(j + 1)*lda + k + 1] * tB[(i + 1)*lda + k + 1];
tmpC12 += A[(j + 1)*lda + k + 1] * tB[(i + 2)*lda + k + 1];
tmpC13 += A[(j + 1)*lda + k + 1] * tB[(i + 3)*lda + k + 1];
tmpC14 += A[(j + 1)*lda + k + 1] * tB[(i + 4)*lda + k + 1];
tmpC15 += A[(j + 1)*lda + k + 1] * tB[(i + 5)*lda + k + 1];
tmpC16 += A[(j + 1)*lda + k + 1] * tB[(i + 6)*lda + k + 1];
tmpC17 += A[(j + 1)*lda + k + 1] * tB[(i + 7)*lda + k + 1];
tmpC00 += A[j*lda + k + 2] * tB[(i + 0)*lda + k + 2];
tmpC01 += A[j*lda + k + 2] * tB[(i + 1)*lda + k + 2];
tmpC02 += A[j*lda + k + 2] * tB[(i + 2)*lda + k + 2];
tmpC03 += A[j*lda + k + 2] * tB[(i + 3)*lda + k + 2];
tmpC04 += A[j*lda + k + 2] * tB[(i + 4)*lda + k + 2];
tmpC05 += A[j*lda + k + 2] * tB[(i + 5)*lda + k + 2];
tmpC06 += A[j*lda + k + 2] * tB[(i + 6)*lda + k + 2];
tmpC07 += A[j*lda + k + 2] * tB[(i + 7)*lda + k + 2];
tmpC10 += A[(j + 1)*lda + k + 2] * tB[(i + 0)*lda + k + 2];
tmpC11 += A[(j + 1)*lda + k + 2] * tB[(i + 1)*lda + k + 2];
tmpC12 += A[(j + 1)*lda + k + 2] * tB[(i + 2)*lda + k + 2];
tmpC13 += A[(j + 1)*lda + k + 2] * tB[(i + 3)*lda + k + 2];
tmpC14 += A[(j + 1)*lda + k + 2] * tB[(i + 4)*lda + k + 2];
tmpC15 += A[(j + 1)*lda + k + 2] * tB[(i + 5)*lda + k + 2];
tmpC16 += A[(j + 1)*lda + k + 2] * tB[(i + 6)*lda + k + 2];
tmpC17 += A[(j + 1)*lda + k + 2] * tB[(i + 7)*lda + k + 2];
tmpC00 += A[j*lda + k + 3] * tB[(i + 0)*lda + k + 3];
tmpC01 += A[j*lda + k + 3] * tB[(i + 1)*lda + k + 3];
tmpC02 += A[j*lda + k + 3] * tB[(i + 2)*lda + k + 3];
tmpC03 += A[j*lda + k + 3] * tB[(i + 3)*lda + k + 3];
tmpC04 += A[j*lda + k + 3] * tB[(i + 4)*lda + k + 3];
tmpC05 += A[j*lda + k + 3] * tB[(i + 5)*lda + k + 3];
tmpC06 += A[j*lda + k + 3] * tB[(i + 6)*lda + k + 3];
tmpC07 += A[j*lda + k + 3] * tB[(i + 7)*lda + k + 3];
tmpC10 += A[(j + 1)*lda + k + 3] * tB[(i + 0)*lda + k + 3];
tmpC11 += A[(j + 1)*lda + k + 3] * tB[(i + 1)*lda + k + 3];
tmpC12 += A[(j + 1)*lda + k + 3] * tB[(i + 2)*lda + k + 3];
tmpC13 += A[(j + 1)*lda + k + 3] * tB[(i + 3)*lda + k + 3];
tmpC14 += A[(j + 1)*lda + k + 3] * tB[(i + 4)*lda + k + 3];
tmpC15 += A[(j + 1)*lda + k + 3] * tB[(i + 5)*lda + k + 3];
tmpC16 += A[(j + 1)*lda + k + 3] * tB[(i + 6)*lda + k + 3];
tmpC17 += A[(j + 1)*lda + k + 3] * tB[(i + 7)*lda + k + 3];
}
C[j*lda + i + 0] = tmpC00; C[(j + 1)*lda + i + 0] = tmpC10;
C[j*lda + i + 1] = tmpC01; C[(j + 1)*lda + i + 1] = tmpC11;
C[j*lda + i + 2] = tmpC02; C[(j + 1)*lda + i + 2] = tmpC12;
C[j*lda + i + 3] = tmpC03; C[(j + 1)*lda + i + 3] = tmpC13;
C[j*lda + i + 4] = tmpC04; C[(j + 1)*lda + i + 4] = tmpC14;
C[j*lda + i + 5] = tmpC05; C[(j + 1)*lda + i + 5] = tmpC15;
C[j*lda + i + 6] = tmpC06; C[(j + 1)*lda + i + 6] = tmpC16;
C[j*lda + i + 7] = tmpC07; C[(j + 1)*lda + i + 7] = tmpC17;
barrier(ncores);
}
}
|
