libgomp/testsuite/libgomp.c-c++-common/baseptrs-6.c

   1 /* { dg-do run } */
   2
   3 #include <stdlib.h>
   4 #include <assert.h>
   5
   6 #define N 1024
   7 #define M 64
   8
   9 int main (void)
  10 {
  11   int *a[N];
  12
  13   for (int i = 0; i < N; i++)
  14     a[i] = (int *) calloc (M, sizeof (int));
  15
  16   /* 'target enter data'/'target exit data' with array of pointers.  */
  17 #pragma omp target enter data map(alloc: a[0:N])
  18
  19   for (int i = 0; i < N; i++)
  20   {
  21 #pragma omp target enter data map(to: a[i][0:M])
  22   }
  23
  24 #pragma omp target map(alloc: a)
  25   {
  26     for (int i = 0; i < N; i++)
  27       for (int j = 0; j < M; j++)
  28         a[i][j] = i + j;
  29   }
  30
  31 for (int i = 0; i < N; i++)
  32   {
  33 #pragma omp target exit data map(release: a[i]) map(from: a[i][0:M])
  34   }
  35
  36 #pragma omp target exit data map(release: a, a[0:N])
  37
  38   /* 'target data' with array of pointers.  */
  39 #pragma omp target data map(alloc: a[0:N])
  40   {
  41 #pragma omp target data map(tofrom: a[5][0:M])
  42     {
  43 #pragma omp target map(alloc: a)
  44       {
  45         for (int i = 0; i < M; i++)
  46           a[5][i]++;
  47       }
  48     }
  49   }
  50
  51   /* 'target' with array of pointers.  */
  52 #pragma omp target data map(alloc: a[0:N])
  53   {
  54 #pragma omp target map(tofrom: a[7][0:M])
  55     {
  56       for (int i = 0; i < M; i++)
  57         a[7][i] += 2;
  58     }
  59   }
  60
  61   for (int i = 0; i < N; i++)
  62     for (int j = 0; j < M; j++)
  63       assert (a[i][j] == i + j + (i == 5) + 2 * (i == 7));
  64
  65   for (int i = 0; i < N; i++)
  66     free (a[i]);
  67
  68   return 0;
  69 }