-
Notifications
You must be signed in to change notification settings - Fork 2
/
trmm.c
48 lines (43 loc) · 1.03 KB
/
trmm.c
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
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <omp.h>
#include <sys/time.h>
double clock()
{
struct timeval Tp;
int stat;
stat = gettimeofday(&Tp, NULL);
if (stat != 0)
printf("Error return from gettimeofday: %d", stat);
return (Tp.tv_sec + Tp.tv_usec * 1.0e-6);
}
void kernel()
{
static float A[2000 - 1 + 2][2000 - 1 + 2] = { 0 }, B[2000 - 1 + 2][2600 - 1 + 2] = { 0 }, alpha = { 0 };
#pragma omp parallel
{
#pragma omp atomic write
alpha = 1.500000;
#pragma omp for
for (int i = 0; i <= 2000 - 1; i++) {
for (int j = 0; j <= 2600 - 1; j++) {
for (int k = i + 1; k <= 2000 - 1; k++) {
#pragma omp atomic
B[i][j] += A[k][i] * B[k][j];
}
#pragma omp atomic write
B[i][j] = alpha * B[i][j];
}
}
}
}
int main()
{
double start = 0.0, end = 0.0;
start = clock();
kernel();
end = clock();
printf("Total time taken = %fs\n", end - start);
return 0;
}