1  extern
       2  #ifdef __cplusplus
       3  "C"
       4  #endif
       5  void abort (void);
       6  
       7  void
       8  fn1 (double *x, double *y, int z)
       9  {
      10    int i;
      11    for (i = 0; i < z; i++)
      12      {
      13        x[i] = i & 31;
      14        y[i] = (i & 63) - 30;
      15      }
      16  }
      17  
      18  double
      19  fn2 (int x)
      20  {
      21    double s = 0;
      22    double b[3 * x], c[3 * x], d[3 * x], e[3 * x];
      23    int i;
      24    fn1 (b, c, x);
      25    fn1 (e, d + x, x);
      26    #pragma omp target map(to: b, c[:x], d[x:x], e) map(tofrom: s)
      27      #pragma omp parallel for reduction(+:s)
      28        for (i = 0; i < x; i++)
      29  	s += b[i] * c[i] + d[x + i] + sizeof (b) - sizeof (c);
      30    return s;
      31  }
      32  
      33  double
      34  fn3 (int x)
      35  {
      36    double s = 0;
      37    double b[3 * x], c[3 * x], d[3 * x], e[3 * x];
      38    int i;
      39    fn1 (b, c, x);
      40    fn1 (e, d, x);
      41    #pragma omp target map(tofrom: s)
      42      #pragma omp parallel for reduction(+:s)
      43        for (i = 0; i < x; i++)
      44  	s += b[i] * c[i] + d[i];
      45    return s;
      46  }
      47  
      48  double
      49  fn4 (int x)
      50  {
      51    double s = 0;
      52    double b[3 * x], c[3 * x], d[3 * x], e[3 * x];
      53    int i;
      54    fn1 (b, c, x);
      55    fn1 (e, d + x, x);
      56    #pragma omp target data map(from: b, c[:x], d[x:x], e)
      57      {
      58        #pragma omp target update to(b, c[:x], d[x:x], e)
      59        #pragma omp target map(c[:x], d[x:x], s)
      60  	#pragma omp parallel for reduction(+:s)
      61  	  for (i = 0; i < x; i++)
      62  	    {
      63  	      s += b[i] * c[i] + d[x + i] + sizeof (b) - sizeof (c);
      64  	      b[i] = i + 0.5;
      65  	      c[i] = 0.5 - i;
      66  	      d[x + i] = 0.5 * i;
      67  	    }
      68      }
      69    for (i = 0; i < x; i++)
      70      if (b[i] != i + 0.5 || c[i] != 0.5 - i || d[x + i] != 0.5 * i)
      71        abort ();
      72    return s;
      73  }
      74  
      75  int
      76  main ()
      77  {
      78    double a = fn2 (128);
      79    if (a != 14080.0)
      80      abort ();
      81    double b = fn3 (128);
      82    if (a != b)
      83      abort ();
      84    double c = fn4 (256);
      85    if (c != 28160.0)
      86      abort ();
      87    return 0;
      88  }