libgomp/testsuite/libgomp.oacc-c-c++-common/loop-red-wv-1.c

   1 #include <stdio.h>
   2 #include <openacc.h>
   3 #include <gomp-constants.h>
   4
   5 #define N (32*32*32+17)
   6 int main ()
   7 {
   8   int ix;
   9   int ondev = 0;
  10   int t = 0, h = 0;
  11   int workersize, vectorsize;
  12
  13 #pragma acc parallel num_workers(32) vector_length(32) copy(ondev) \
  14             copyout(workersize, vectorsize)
  15   {
  16 #pragma acc loop worker vector reduction (+:t)
  17     for (unsigned ix = 0; ix < N; ix++)
  18       {
  19         int val = ix;
  20
  21         if (acc_on_device (acc_device_not_host))
  22           {
  23             int g, w, v;
  24
  25             g = __builtin_goacc_parlevel_id (GOMP_DIM_GANG);
  26             w = __builtin_goacc_parlevel_id (GOMP_DIM_WORKER);
  27             v = __builtin_goacc_parlevel_id (GOMP_DIM_VECTOR);
  28             val = (g << 16) | (w << 8) | v;
  29             ondev = 1;
  30           }
  31         t += val;
  32       }
  33     workersize = __builtin_goacc_parlevel_size (GOMP_DIM_WORKER);
  34     vectorsize = __builtin_goacc_parlevel_size (GOMP_DIM_VECTOR);
  35   }
  36
  37   for (ix = 0; ix < N; ix++)
  38     {
  39       int val = ix;
  40       if(ondev)
  41         {
  42           int g = 0;
  43           int w = (ix / vectorsize) % workersize;
  44           int v = ix % vectorsize;
  45
  46           val = (g << 16) | (w << 8) | v;
  47         }
  48       h += val;
  49     }
  50   if (t != h)
  51     {
  52       printf ("t=%x expected %x\n", t, h);
  53       return 1;
  54     }
  55
  56   return 0;
  57 }