libgomp/testsuite/libgomp.oacc-c-c++-common/kernels-private-vars-loop-vector-2.c

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