1! { dg-do run }
2!
3! PR fortran/94690
4! PR middle-end/66199
5
6module m
7  implicit none
8  integer u(1024), v(1024), w(1024)
9  !$omp declare target (v, u, w)
10
11contains
12
13integer function f1 (a, b)
14  integer :: a, b, d
15  !$omp target map(from: d)
16  !$omp teams distribute simd default(none) firstprivate (a, b) shared(u, v, w)
17  do d = a, b-1
18    u(d) = v(d) + w(d)
19  end do
20  !$omp end teams distribute simd
21  !$omp end target
22  f1 = d
23end
24
25integer function f2 (a, b, c)
26  integer a, b, c, d, e
27  !$omp target map(from: d, e)
28  !$omp teams distribute simd default(none) firstprivate (a, b, c) shared(u, v, w) linear(d) lastprivate(e)
29  do d = a, b-1
30    u(d) = v(d) + w(d)
31    e = c + d * 5
32  end do
33  !$omp end teams distribute simd
34  !$omp end target
35  f2 = d + e
36end
37
38integer function f3 (a1, b1, a2, b2)
39  integer :: a1, b1, a2, b2, d1, d2
40  !$omp target map(from: d1, d2)
41  !$omp teams distribute simd default(none) firstprivate (a1, b1, a2, b2) shared(u, v, w) lastprivate(d1, d2) collapse(2)
42  do d1 = a1, b1-1
43    do d2 = a2, b2-1
44      u(d1 * 32 + d2) = v(d1 * 32 + d2) + w(d1 * 32 + d2)
45    end do
46  end do
47  !$omp end teams distribute simd
48  !$omp end target
49  f3 = d1 + d2
50end
51
52integer function f4 (a1, b1, a2, b2)
53  integer :: a1, b1, a2, b2, d1, d2
54  !$omp target map(from: d1, d2)
55  !$omp teams distribute simd default(none) firstprivate (a1, b1, a2, b2) shared(u, v, w) collapse(2)
56  do d1 = a1, b1-1
57    do d2 = a2, b2-1
58      u(d1 * 32 + d2) = v(d1 * 32 + d2) + w(d1 * 32 + d2)
59    end do
60  end do
61  !$omp end teams distribute simd
62  !$omp end target
63  f4 = d1 + d2
64end
65end module
66
67use m
68  if (f1 (0, 1024) /= 1024) stop 1
69  if (f2 (0, 1024, 17) /= 1024 + (17 + 5 * 1023)) stop 2
70  if (f3 (0, 32, 0, 32) /= 64) stop 3
71  if (f4 (0, 32, 0, 32) /= 64) stop 4
72end
73