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

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