summaryrefslogtreecommitdiff
path: root/libgomp/testsuite/libgomp.c/examples-4/declare_target-4.c
blob: f24143669517578d706602473c07e88753687ee9 (plain)
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
/* { dg-do run } */
/* { dg-additional-options "-DTESTITERS=20" { target { ! run_expensive_tests } } } */

#include <stdlib.h>

#define EPS 0.00001
#define N 1000
#ifndef TESTITERS
#define TESTITERS N
#endif

#pragma omp declare target
float Q[N][N];
float Pfun (const int i, const int k)
{
  return Q[i][k] * Q[k][i];
}
#pragma omp end declare target

void init ()
{
  int i, j;
  for (i = 0; i < N; i++)
    for (j = 0; j < N; j++)
      Q[i][j] = 0.001 * i * j;
}

float accum_ref (int k)
{
  int i;
  float tmp = 0.0;

  for (i = 0; i < N; i++)
    tmp += Pfun (i, k);

  return tmp;
}

float accum (int k)
{
  int i;
  float tmp = 0.0;

  #pragma omp target map(tofrom:tmp)
    #pragma omp parallel for reduction(+:tmp)
      for (i = 0; i < N; i++)
	tmp += Pfun (i, k);

  return tmp;
}

void check (float a, float b)
{
  float err = (b == 0.0) ? a : (a - b) / b;
  if (((err > 0) ? err : -err) > EPS)
    abort ();
}

int main ()
{
  int i;

  init ();

  #pragma omp target update to(Q)

  for (i = 0; i < TESTITERS; i++)
    check (accum (i), accum_ref (i));

  return 0;
}