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
|
#define AN 3
#define BM 2
#define AMBN 5
//#define PRINT
#include <time.h>
#include <fstream>
#include <cstdio>
/*
<test name='mm_small'>
procedure void mm(
in float[3][5] A = matrix([*,*],lambda i,j: random(-8,8)),
in float[5][2] B = matrix([*,*],lambda i,j: random(-8,8)),
out float[3][2] C = matrix([*,*],lambda i,j: 0))
</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++) {
C[i][j] = 0;
for(k = 0; k < AMBN; k++) {
C[i][j] += A[i][k] * B[k][j];
}
}
}
}
int main(int argc, char** argv) {
float A[3][5] = {{0,1,2,3,4},{5,6,7,8,9},{10,11,12,13,14}};
float B[5][2] = {{0,1},{2,3},{4,5},{6,7},{8,9}};
float C[3][2] = {{0,0},{0,0},{0,0}};
timespec start_time;
timespec end_time;
if (argc == 3) {
std::ifstream is(argv[1], std::ifstream::in | std::ifstream::binary);
is.read((char*)A, 15*sizeof(float));
is.read((char*)B, 10*sizeof(float));
is.close();
}
clock_gettime(CLOCK_REALTIME, &start_time);
for(int i = 0; i < 10000; i++) {
mm(A,B,C);
}
clock_gettime(CLOCK_REALTIME, &end_time);
if (argc == 3) {
std::ofstream os(argv[2], std::ofstream::out | std::ofstream::binary);
os.write((char*)C, 6*sizeof(float));
os.close();
}
#ifdef PRINT
std::printf("A:\n");
for(int i = 0; i < 3; i++) {
std::printf("[");
for(int j = 0; j < 5; j++) {
std::printf("%f,",A[i][j]);
}
std::printf("]\n");
}
std::printf("B:\n");
for(int i = 0; i < 5; i++) {
std::printf("[");
for(int j = 0; j < 2; j++) {
std::printf("%f,",B[i][j]);
}
std::printf("]\n");
}
std::printf("C:\n");
for(int i = 0; i < 3; i++) {
std::printf("[");
for(int j = 0; j < 2; j++) {
std::printf("%f,",C[i][j]);
}
std::printf("]\n");
}
#else
double time_diff = (end_time.tv_sec - start_time.tv_sec) + (end_time.tv_nsec - start_time.tv_nsec)/1000000000.0;
std::printf("(%f,)", time_diff);
#endif
return 0;
}
|
