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
f1(long a,long b)9 f1 (long a, long b)
10 {
11   long d;
12   #pragma omp target map(from: d)
13   #pragma omp teams distribute parallel for 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
f2(long a,long b,long c)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 parallel for 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
f3(long a1,long b1,long a2,long b2)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 parallel for 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
f4(long a1,long b1,long a2,long b2)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 parallel for 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
main()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 }
67