aboutsummaryrefslogtreecommitdiff
path: root/libgomp/testsuite/libgomp.c-c++-common/target-update-iterators-2.c
blob: 93438d01c97e1701ae7be02b5fdccf69a8044da2 (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
/* { dg-do run } */
/* { dg-require-effective-target offload_device_nonshared_as } */

/* Test target enter data and target update from the target using map
   iterators.  */

#include <stdlib.h>

#define DIM1 8
#define DIM2 15

void mkarray (int *x[])
{
  for (int i = 0; i < DIM1; i++)
    {
      x[i] = (int *) malloc (DIM2 * sizeof (int));
      for (int j = 0; j < DIM2; j++)
	x[i][j] = 0;
    }
}

int main (void)
{
  int *x[DIM1];
  int sum, expected;

  mkarray (x);

  #pragma omp target enter data map(alloc: x[:DIM1])
  #pragma omp target enter data map(iterator(i=0:DIM1), to: x[i][:DIM2])
  #pragma omp target map(from: expected)
    {
      expected = 0;
      for (int i = 0; i < DIM1; i++)
	for (int j = 0; j < DIM2; j++)
	  {
	    x[i][j] = (i + 1) * (j + 2);
	    expected += x[i][j];
	  }
    }

  /* Host copy of x should remain unchanged.  */
  sum = 0;
  for (int i = 0; i < DIM1; i++)
    for (int j = 0; j < DIM2; j++)
      sum += x[i][j];
  if (sum != 0)
    return 1;

  #pragma omp target update from(iterator(i=0:DIM1): x[i][:DIM2])

  /* Host copy should now be updated.  */
  sum = 0;
  for (int i = 0; i < DIM1; i++)
    for (int j = 0; j < DIM2; j++)
      sum += x[i][j];
  return sum - expected;
}