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