libgomp/testsuite/libgomp.c++/declare-mapper-2.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
49
50
51
52
53
54
55

// { dg-do run }

#include <cassert>

#define N 256

struct doublebuf
{
  int buf_a[N][N];
  int buf_b[N][N];
};

#pragma omp declare mapper(lo:doublebuf b) map(b.buf_a[0:N/2][0:N]) \
					   map(b.buf_b[0:N/2][0:N])

#pragma omp declare mapper(hi:doublebuf b) map(b.buf_a[N/2:N/2][0:N]) \
					   map(b.buf_b[N/2:N/2][0:N])

int main (int argc, char *argv[])
{
  doublebuf db;

  for (int i = 0; i < N; i++)
    for (int j = 0; j < N; j++)
      db.buf_a[i][j] = db.buf_b[i][j] = 0;

  #pragma omp target map(mapper(lo), tofrom:db)
  {
    for (int i = 0; i < N / 2; i++)
      for (int j = 0; j < N; j++)
	{
	  db.buf_a[i][j]++;
	  db.buf_b[i][j]++;
	}
  }

  #pragma omp target map(mapper(hi), tofrom:db)
  {
    for (int i = N / 2; i < N; i++)
      for (int j = 0; j < N; j++)
	{
	  db.buf_a[i][j]++;
	  db.buf_b[i][j]++;
	}
  }

  for (int i = 0; i < N; i++)
    for (int j = 0; j < N; j++)
      {
	assert (db.buf_a[i][j] == 1);
	assert (db.buf_b[i][j] == 1);
      }

  return 0;
}