-
Notifications
You must be signed in to change notification settings - Fork 1
/
j2d9pt-gol.c
37 lines (33 loc) · 1.37 KB
/
j2d9pt-gol.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
#define BENCH_DIM 2
#define BENCH_FPP 18
#define BENCH_RAD 1
#include "common.h"
double kernel_stencil(SB_TYPE *A1, int compsize, int timestep, bool scop)
{
double start_time = sb_time(), end_time = 0.0;
int dimsize = compsize + BENCH_RAD * 2;
SB_TYPE (*A)[dimsize][dimsize] = (SB_TYPE (*)[dimsize][dimsize])A1;
if (scop) {
#pragma scop
for (int t = 0; t < timestep; t++)
for (int i = BENCH_RAD; i < dimsize - BENCH_RAD; i++)
for (int j = BENCH_RAD; j < dimsize - BENCH_RAD; j++)
A[(t+1)%2][i][j] =
( 7.1f * A[t%2][i-1][j-1] + 5.1f * A[t%2][i-1][j] + 9.2f * A[t%2][i-1][j+1] +
12.1f * A[t%2][i][j-1] + 15.f * A[t%2][i][j] + 12.2f * A[t%2][i][j+1] +
9.1f * A[t%2][i+1][j-1] + 5.2f * A[t%2][i+1][j] + 7.2f * A[t%2][i+1][j+1]) / 118;
#pragma endscop
}
else {
for (int t = 0; t < timestep; t++)
#pragma omp parallel for
for (int i = BENCH_RAD; i < dimsize - BENCH_RAD; i++)
for (int j = BENCH_RAD; j < dimsize - BENCH_RAD; j++)
A[(t+1)%2][i][j] =
A[(t+1)%2][i][j] =
( 7.1f * A[t%2][i-1][j-1] + 5.1f * A[t%2][i-1][j] + 9.2f * A[t%2][i-1][j+1] +
12.1f * A[t%2][i][j-1] + 15.f * A[t%2][i][j] + 12.2f * A[t%2][i][j+1] +
9.1f * A[t%2][i+1][j-1] + 5.2f * A[t%2][i+1][j] + 7.2f * A[t%2][i+1][j+1]) / 118;
}
return (((end_time != 0.0) ? end_time : sb_time()) - start_time);
}