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

   1 #include <assert.h>
   2
   3 /* Test of vector-private variables declared on loop directive.  */
   4
   5 int
   6 main (int argc, char* argv[])
   7 {
   8   int x, 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(x)
  26             for (k = 0; k < 32; k++)
  27               {
  28                 x = i ^ j * 3;
  29                 arr[i * 1024 + j * 32 + k] += x * k;
  30               }
  31
  32             #pragma acc loop vector(length:32) private(x)
  33             for (k = 0; k < 32; k++)
  34               {
  35                 x = i | j * 5;
  36                 arr[i * 1024 + j * 32 + k] += x * k;
  37               }
  38           }
  39       }
  40   }
  41
  42   for (i = 0; i < 32; i++)
  43     for (int j = 0; j < 32; j++)
  44       for (int k = 0; k < 32; k++)
  45         {
  46           int idx = i * 1024 + j * 32 + k;
  47           assert (arr[idx] == idx + (i ^ j * 3) * k + (i | j * 5) * k);
  48         }
  49
  50   return 0;
  51 }