libgomp/testsuite/libgomp.oacc-c-c++-common/f-asyncwait-3.c

   1 /* { dg-do run } */
   2
   3 /* Based on asyncwait-3.f90.  */
   4
   5 #include <stdlib.h>
   6
   7 #define N 64
   8
   9 int
  10 main (void)
  11 {
  12   int *a, *b, *c;
  13
  14   a = (int *)malloc (N * sizeof (*a));
  15   b = (int *)malloc (N * sizeof (*b));
  16   c = (int *)malloc (N * sizeof (*c));
  17
  18 #pragma acc parallel copy (a[0:N]) async (0)
  19 #pragma acc loop
  20   for (int i = 0; i < N; ++i)
  21     a[i] = 1;
  22
  23 #pragma acc parallel copy (b[0:N]) async (1)
  24 #pragma acc loop
  25   for (int i = 0; i < N; ++i)
  26     b[i] = 1;
  27
  28 #pragma acc wait (0, 1)
  29
  30 #pragma acc parallel copy (a[0:N], b[0:N], c[0:N])
  31 #pragma acc loop
  32   for (int i = 0; i < N; ++i)
  33     c[i] = a[i] + b[i];
  34
  35   for (int i = 0; i < N; ++i)
  36     if (c[i] != 2)
  37       abort ();
  38
  39 #pragma acc kernels copy (a[0:N]) async (0)
  40 #pragma acc loop
  41   for (int i = 0; i < N; ++i)
  42     a[i] = 1;
  43
  44 #pragma acc kernels copy (b[0:N]) async (1)
  45 #pragma acc loop
  46   for (int i = 0; i < N; ++i)
  47     b[i] = 1;
  48
  49 #pragma acc wait (0, 1)
  50
  51 #pragma acc kernels copy (a[0:N], b[0:N], c[0:N])
  52 #pragma acc loop
  53   for (int i = 0; i < N; ++i)
  54     c[i] = a[i] + b[i];
  55
  56   for (int i = 0; i < N; ++i)
  57     if (c[i] != 2)
  58       abort ();
  59
  60   free (a);
  61   free (b);
  62   free (c);
  63 }