libgomp/testsuite/libgomp.oacc-c-c++-common/data-2.c

   1 /* Test 'acc enter/exit data' regions.  */
   2
   3 /* { dg-do run } */
   4 /* { dg-xfail-run-if "TODO" { openacc_nvidia_accel_selected } } */
   5
   6 #include <stdlib.h>
   7
   8 int
   9 main (int argc, char **argv)
  10 {
  11   int N = 128; //1024 * 1024;
  12   float *a, *b, *c, *d, *e;
  13   int i;
  14   int nbytes;
  15
  16   nbytes = N * sizeof (float);
  17
  18   a = (float *) malloc (nbytes);
  19   b = (float *) malloc (nbytes);
  20   c = (float *) malloc (nbytes);
  21   d = (float *) malloc (nbytes);
  22   e = (float *) malloc (nbytes);
  23
  24   for (i = 0; i < N; i++)
  25     {
  26       a[i] = 3.0;
  27       b[i] = 0.0;
  28     }
  29
  30 #pragma acc enter data copyin (a[0:N]) copyin (b[0:N]) copyin (N) async
  31 #pragma acc parallel present (a[0:N], b[0:N]) async wait
  32 #pragma acc loop
  33   for (i = 0; i < N; i++)
  34     b[i] = a[i];
  35
  36 #pragma acc exit data copyout (a[0:N]) copyout (b[0:N]) wait async
  37 #pragma acc wait
  38
  39   for (i = 0; i < N; i++)
  40     {
  41       if (a[i] != 3.0)
  42         abort ();
  43
  44       if (b[i] != 3.0)
  45         abort ();
  46     }
  47
  48   for (i = 0; i < N; i++)
  49     {
  50       a[i] = 3.0;
  51       b[i] = 0.0;
  52     }
  53
  54 #pragma acc enter data copyin (a[0:N]) async
  55 #pragma acc enter data copyin (b[0:N]) async wait
  56 #pragma acc enter data copyin (N) async wait
  57 #pragma acc parallel async wait
  58 #pragma acc loop
  59   for (i = 0; i < N; i++)
  60     b[i] = a[i];
  61
  62 #pragma acc exit data copyout (a[0:N]) copyout (b[0:N]) delete (N) wait async
  63 #pragma acc wait
  64
  65   for (i = 0; i < N; i++)
  66     {
  67       if (a[i] != 3.0)
  68         abort ();
  69
  70       if (b[i] != 3.0)
  71         abort ();
  72     }
  73
  74   for (i = 0; i < N; i++)
  75     {
  76       a[i] = 2.0;
  77       b[i] = 0.0;
  78     }
  79
  80 #pragma acc enter data copyin (a[0:N]) copyin (b[0:N]) copyin (N) async (1)
  81 #pragma acc parallel present (a[0:N], b[0:N])  async (1)
  82 #pragma acc loop
  83   for (i = 0; i < N; i++)
  84     b[i] = a[i];
  85
  86 #pragma acc exit data copyout (a[0:N]) copyout (b[0:N]) wait (1) async (1)
  87 #pragma acc wait (1)
  88
  89   for (i = 0; i < N; i++)
  90     {
  91       if (a[i] != 2.0)
  92         abort ();
  93
  94       if (b[i] != 2.0)
  95         abort ();
  96     }
  97
  98   for (i = 0; i < N; i++)
  99     {
 100       a[i] = 3.0;
 101       b[i] = 0.0;
 102       c[i] = 0.0;
 103       d[i] = 0.0;
 104     }
 105
 106 #pragma acc enter data copyin (a[0:N]) copyin (b[0:N]) copyin (c[0:N]) copyin (d[0:N]) copyin (N) async (1)
 107
 108 #pragma acc parallel present (a[0:N], b[0:N]) async (1) wait (1)
 109 #pragma acc loop
 110   for (i = 0; i < N; i++)
 111     b[i] = (a[i] * a[i] * a[i]) / a[i];
 112
 113 #pragma acc parallel present (a[0:N], c[0:N]) async (2) wait (1)
 114 #pragma acc loop
 115   for (i = 0; i < N; i++)
 116     c[i] = (a[i] + a[i] + a[i] + a[i]) / a[i];
 117
 118 #pragma acc parallel present (a[0:N], d[0:N]) async (3) wait (1)
 119 #pragma acc loop
 120   for (i = 0; i < N; i++)
 121     d[i] = ((a[i] * a[i] + a[i]) / a[i]) - a[i];
 122
 123 #pragma acc exit data copyout (a[0:N]) copyout (b[0:N]) copyout (c[0:N]) copyout (d[0:N]) wait (1, 2, 3) async (1)
 124 #pragma acc wait (1)
 125
 126   for (i = 0; i < N; i++)
 127     {
 128       if (a[i] != 3.0)
 129         abort ();
 130
 131       if (b[i] != 9.0)
 132         abort ();
 133
 134       if (c[i] != 4.0)
 135         abort ();
 136
 137       if (d[i] != 1.0)
 138         abort ();
 139     }
 140
 141   for (i = 0; i < N; i++)
 142     {
 143       a[i] = 2.0;
 144       b[i] = 0.0;
 145       c[i] = 0.0;
 146       d[i] = 0.0;
 147       e[i] = 0.0;
 148     }
 149
 150 #pragma acc enter data copyin (a[0:N]) copyin (b[0:N]) copyin (c[0:N]) copyin (d[0:N]) copyin (e[0:N]) copyin (N) async (1)
 151
 152 #pragma acc parallel present (a[0:N], b[0:N]) async (1) wait (1)
 153   for (int ii = 0; ii < N; ii++)
 154     b[ii] = (a[ii] * a[ii] * a[ii]) / a[ii];
 155
 156 #pragma acc parallel present (a[0:N], c[0:N]) async (2) wait (1)
 157   for (int ii = 0; ii < N; ii++)
 158     c[ii] = (a[ii] + a[ii] + a[ii] + a[ii]) / a[ii];
 159
 160 #pragma acc parallel present (a[0:N], d[0:N]) async (3) wait (1)
 161   for (int ii = 0; ii < N; ii++)
 162     d[ii] = ((a[ii] * a[ii] + a[ii]) / a[ii]) - a[ii];
 163
 164 #pragma acc parallel present (a[0:N], b[0:N], c[0:N], d[0:N], e[0:N]) \
 165   wait (1) async (4)
 166   for (int ii = 0; ii < N; ii++)
 167     e[ii] = a[ii] + b[ii] + c[ii] + d[ii];
 168
 169 #pragma acc exit data copyout (a[0:N]) copyout (b[0:N]) copyout (c[0:N]) \
 170   copyout (d[0:N]) copyout (e[0:N]) wait (1, 2, 3, 4) async (1)
 171 #pragma acc wait (1)
 172
 173   for (i = 0; i < N; i++)
 174     {
 175       if (a[i] != 2.0)
 176         abort ();
 177
 178       if (b[i] != 4.0)
 179         abort ();
 180
 181       if (c[i] != 4.0)
 182         abort ();
 183
 184       if (d[i] != 1.0)
 185         abort ();
 186
 187       if (e[i] != 11.0)
 188         abort ();
 189     }
 190
 191   return 0;
 192 }