1  /* PR middle-end/66199 */
       2  /* { dg-do run } */
       3  
       4  #pragma omp declare target
       5  int u[1024], v[1024], w[1024];
       6  #pragma omp end declare target
       7  
       8  __attribute__((noinline, noclone)) long
       9  f1 (long a, long b)
      10  {
      11    long d;
      12    #pragma omp target map(from: d)
      13    #pragma omp teams distribute simd default(none) firstprivate (a, b) shared(u, v, w)
      14    for (d = a; d < b; d++)
      15      u[d] = v[d] + w[d];
      16    return d;
      17  }
      18  
      19  __attribute__((noinline, noclone)) long
      20  f2 (long a, long b, long c)
      21  {
      22    long d, e;
      23    #pragma omp target map(from: d, e)
      24    #pragma omp teams distribute simd default(none) firstprivate (a, b, c) shared(u, v, w) linear(d) lastprivate(e)
      25    for (d = a; d < b; d++)
      26      {
      27        u[d] = v[d] + w[d];
      28        e = c + d * 5;
      29      }
      30    return d + e;
      31  }
      32  
      33  __attribute__((noinline, noclone)) long
      34  f3 (long a1, long b1, long a2, long b2)
      35  {
      36    long d1, d2;
      37    #pragma omp target map(from: d1, d2)
      38    #pragma omp teams distribute simd default(none) firstprivate (a1, b1, a2, b2) shared(u, v, w) lastprivate(d1, d2) collapse(2)
      39    for (d1 = a1; d1 < b1; d1++)
      40      for (d2 = a2; d2 < b2; d2++)
      41        u[d1 * 32 + d2] = v[d1 * 32 + d2] + w[d1 * 32 + d2];
      42    return d1 + d2;
      43  }
      44  
      45  __attribute__((noinline, noclone)) long
      46  f4 (long a1, long b1, long a2, long b2)
      47  {
      48    long d1, d2;
      49    #pragma omp target map(from: d1, d2)
      50    #pragma omp teams distribute simd default(none) firstprivate (a1, b1, a2, b2) shared(u, v, w) collapse(2)
      51    for (d1 = a1; d1 < b1; d1++)
      52      for (d2 = a2; d2 < b2; d2++)
      53        u[d1 * 32 + d2] = v[d1 * 32 + d2] + w[d1 * 32 + d2];
      54    return d1 + d2;
      55  }
      56  
      57  int
      58  main ()
      59  {
      60    if (f1 (0, 1024) != 1024
      61        || f2 (0, 1024, 17) != 1024 + (17 + 5 * 1023)
      62        || f3 (0, 32, 0, 32) != 64
      63        || f4 (0, 32, 0, 32) != 64)
      64      __builtin_abort ();
      65    return 0;
      66  }