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
|
#define AN 3
#define BM 2
#define AMBN 5
/*
<test name='mm_small'>
with {evendist2:lambda i,j: random(-8,8), zero2:lambda i,j: 0}
procedure void mm(
in float[3][5] A = matrix([*,*],evendist2),
in float[5][2] B = matrix([*,*],evendist2),
out float[3][2] C = matrix([*,*],zero2))
</test>
*/
void mm(float A[AN][AMBN], float B[AMBN][BM], float C[AN][BM]) {
int i;
int j;
int k;
for(i = 0; i < AN; i++) {
for(j = 0; j < BM; j++) {
for(k = 0; k < AMBN; k++) {
C[i][j] += A[i][k] * B[k][j];
}
}
}
}
|
