libgomp/testsuite/libgomp.oacc-c-c++-common/kernels-private-vars-local-worker-2.c

   1 #include <assert.h>
   2
   3 /* Test of worker-private variables declared in a local scope, broadcasting
   4    to vector-partitioned mode.  Successive vector loops.  */
   5
   6 int
   7 main (int argc, char* argv[])
   8 {
   9   int x = 5, i, arr[32 * 32 * 32];
  10
  11   for (i = 0; i < 32 * 32 * 32; i++)
  12     arr[i] = i;
  13
  14   #pragma acc kernels copy(arr)
  15   {
  16     int j;
  17
  18     #pragma acc loop gang(num:32)
  19     for (i = 0; i < 32; i++)
  20       {
  21         #pragma acc loop worker(num:32)
  22         for (j = 0; j < 32; j++)
  23           {
  24             int k;
  25             int x = i ^ j * 3;
  26
  27             #pragma acc loop vector(length:32)
  28             for (k = 0; k < 32; k++)
  29               arr[i * 1024 + j * 32 + k] += x * k;
  30
  31             x = i | j * 5;
  32
  33             #pragma acc loop vector(length:32)
  34             for (k = 0; k < 32; k++)
  35               arr[i * 1024 + j * 32 + k] += x * k;
  36           }
  37       }
  38   }
  39
  40   for (i = 0; i < 32; i++)
  41     for (int j = 0; j < 32; j++)
  42       for (int k = 0; k < 32; k++)
  43         {
  44           int idx = i * 1024 + j * 32 + k;
  45           assert (arr[idx] == idx + (i ^ j * 3) * k + (i | j * 5) * k);
  46         }
  47
  48   return 0;
  49 }