libgomp/testsuite/libgomp.oacc-c-c++-common/kernels-private-vars-loop-worker-7.c

   1 #include <assert.h>
   2
   3 /* Test of worker-private variables declared on loop directive, broadcasting
   4    to vector-partitioned mode.  Array worker variable.  */
   5
   6 int
   7 main (int argc, char* argv[])
   8 {
   9   int i, arr[32 * 32 * 32];
  10   int pt[2];
  11
  12   for (i = 0; i < 32 * 32 * 32; i++)
  13     arr[i] = i;
  14
  15   /* "pt" is treated as "present_or_copy" on the kernels directive because it
  16      is an array variable.  */
  17   #pragma acc kernels copy(arr)
  18   {
  19     int j;
  20
  21     #pragma acc loop gang(num:32)
  22     for (i = 0; i < 32; i++)
  23       {
  24         /* But here, it is made private per-worker.  */
  25         #pragma acc loop worker(num:32) private(pt)
  26         for (j = 0; j < 32; j++)
  27           {
  28             int k;
  29
  30             pt[0] = i ^ j * 3;
  31
  32             #pragma acc loop vector(length:32)
  33             for (k = 0; k < 32; k++)
  34               arr[i * 1024 + j * 32 + k] += pt[0] * k;
  35
  36             pt[1] = i | j * 5;
  37
  38             #pragma acc loop vector(length:32)
  39             for (k = 0; k < 32; k++)
  40               arr[i * 1024 + j * 32 + k] += pt[1] * k;
  41           }
  42       }
  43   }
  44
  45   for (i = 0; i < 32; i++)
  46     for (int j = 0; j < 32; j++)
  47       for (int k = 0; k < 32; k++)
  48         {
  49           int idx = i * 1024 + j * 32 + k;
  50           assert (arr[idx] == idx + (i ^ j * 3) * k + (i | j * 5) * k);
  51         }
  52
  53   return 0;
  54 }