libgomp/testsuite/libgomp.oacc-c-c++-common/routine-4.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3
   4 #define M 8
   5 #define N 32
   6
   7 #pragma acc routine vector
   8 void
   9 vector (int *a)
  10 {
  11   int i;
  12
  13 #pragma acc loop vector
  14   for (i = 0; i < N; i++)
  15     a[i] -= a[i];
  16 }
  17
  18 #pragma acc routine worker
  19 void
  20 worker (int *b)
  21 {
  22   int i, j;
  23
  24 #pragma acc loop worker
  25   for (i = 0; i < N; i++)
  26     {
  27 #pragma acc loop vector
  28       for (j = 0; j < M; j++)
  29         b[i * M + j] += b[i  * M + j];
  30     }
  31 }
  32
  33 #pragma acc routine gang
  34 void
  35 gang (int *a)
  36 {
  37   int i;
  38
  39 #pragma acc loop gang worker vector
  40   for (i = 0; i < N; i++)
  41     a[i] -= i;
  42 }
  43
  44 #pragma acc routine seq
  45 void
  46 seq (int *a)
  47 {
  48   int i;
  49
  50   for (i = 0; i < N; i++)
  51     a[i] += 1;
  52 }
  53
  54 int
  55 main(int argc, char **argv)
  56 {
  57   int i;
  58   int a[N];
  59   int b[M * N];
  60
  61   i = 0;
  62
  63   for (i = 0; i < N; i++)
  64     a[i] = 0;
  65
  66 #pragma acc parallel copy (a[0:N])
  67   {
  68 #pragma acc loop seq
  69     for (i = 0; i < N; i++)
  70       seq (&a[0]);
  71   }
  72
  73   for (i = 0; i < N; i++)
  74     {
  75       if (a[i] != N)
  76         abort ();
  77     }
  78
  79 #pragma acc parallel copy (a[0:N])
  80   {
  81 #pragma acc loop seq
  82     for (i = 0; i < N; i++)
  83       gang (&a[0]);
  84   }
  85
  86   for (i = 0; i < N; i++)
  87     {
  88       if (a[i] != N + (N * (-1 * i)))
  89         abort ();
  90     }
  91
  92   for (i = 0; i < N; i++)
  93     a[i] = i;
  94
  95 #pragma acc parallel copy (b[0:M*N])
  96   {
  97     worker (&b[0]);
  98   }
  99
 100   for (i = 0; i < N; i++)
 101     {
 102       if (a[i] != i)
 103         abort ();
 104     }
 105
 106   for (i = 0; i < N; i++)
 107     a[i] = i;
 108
 109 #pragma acc parallel copy (a[0:N])
 110   {
 111 #pragma acc loop
 112     for (i = 0; i < N; i++)
 113       vector (&a[0]);
 114   }
 115
 116   for (i = 0; i < N; i++)
 117     {
 118       if (a[i] != 0)
 119         abort ();
 120     }
 121
 122   return 0;
 123 }