xref: /llvm-project/flang/test/Lower/loops.f90 (revision f35f863a88f83332bef9605ef4cfe4f05c066efb)
1! RUN: bbc -emit-fir -hlfir=false -o - %s | FileCheck %s
2
3! CHECK-LABEL: loop_test
4subroutine loop_test
5  ! CHECK: %[[VAL_2:.*]] = fir.alloca i16 {bindc_name = "i"}
6  ! CHECK: %[[VAL_3:.*]] = fir.alloca i16 {bindc_name = "i"}
7  ! CHECK: %[[VAL_4:.*]] = fir.alloca i16 {bindc_name = "i"}
8  ! CHECK: %[[VAL_5:.*]] = fir.alloca i8 {bindc_name = "k"}
9  ! CHECK: %[[VAL_6:.*]] = fir.alloca i8 {bindc_name = "j"}
10  ! CHECK: %[[VAL_7:.*]] = fir.alloca i8 {bindc_name = "i"}
11  ! CHECK: %[[VAL_8:.*]] = fir.alloca i32 {bindc_name = "k"}
12  ! CHECK: %[[VAL_9:.*]] = fir.alloca i32 {bindc_name = "j"}
13  ! CHECK: %[[VAL_10:.*]] = fir.alloca i32 {bindc_name = "i"}
14  ! CHECK: %[[VAL_11:.*]] = fir.alloca !fir.array<5x5x5xi32> {bindc_name = "a", uniq_name = "_QFloop_testEa"}
15  ! CHECK: %[[VAL_12:.*]] = fir.alloca i32 {bindc_name = "asum", uniq_name = "_QFloop_testEasum"}
16  ! CHECK: %[[VAL_13:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFloop_testEi"}
17  ! CHECK: %[[VAL_14:.*]] = fir.alloca i32 {bindc_name = "j", uniq_name = "_QFloop_testEj"}
18  ! CHECK: %[[VAL_15:.*]] = fir.alloca i32 {bindc_name = "k", uniq_name = "_QFloop_testEk"}
19  ! CHECK: %[[VAL_16:.*]] = fir.alloca f32 {bindc_name = "x", uniq_name = "_QFloop_testEx"}
20  ! CHECK: %[[VAL_17:.*]] = fir.alloca i32 {bindc_name = "xsum", uniq_name = "_QFloop_testExsum"}
21
22  integer(4) :: a(5,5,5), i, j, k, asum, xsum
23
24  i = 100
25  j = 200
26  k = 300
27
28  ! CHECK-COUNT-3: fir.do_loop {{.*}} unordered
29  do concurrent (i=1:5, j=1:5, k=1:5) ! shared(a)
30    ! CHECK: fir.coordinate_of
31    a(i,j,k) = 0
32  enddo
33  ! CHECK: fir.call @_FortranAioBeginExternalListOutput
34  print*, 'A:', i, j, k
35
36  ! CHECK-COUNT-3: fir.do_loop {{.*}} unordered
37  ! CHECK: fir.if
38  do concurrent (integer(1)::i=1:5, j=1:5, k=1:5, i.ne.j .and. k.ne.3) shared(a)
39    ! CHECK-COUNT-2: fir.coordinate_of
40    a(i,j,k) = a(i,j,k) + 1
41  enddo
42
43  ! CHECK-COUNT-3: fir.do_loop {{[^un]*}} -> (index, i32)
44  asum = 0
45  do i=1,5
46    do j=1,5
47      do k=1,5
48        ! CHECK: fir.coordinate_of
49        asum = asum + a(i,j,k)
50      enddo
51    enddo
52  enddo
53  ! CHECK: fir.call @_FortranAioBeginExternalListOutput
54  print*, 'B:', i, j, k, '-', asum
55
56  ! CHECK: fir.do_loop {{.*}} unordered
57  ! CHECK-COUNT-2: fir.if
58  do concurrent (integer(2)::i=1:5, i.ne.3)
59    if (i.eq.2 .or. i.eq.4) goto 5 ! fir.if
60    ! CHECK: fir.call @_FortranAioBeginExternalListOutput
61    print*, 'C:', i
62  5 continue
63  enddo
64
65  ! CHECK: fir.do_loop {{.*}} unordered
66  ! CHECK-COUNT-2: fir.if
67  do concurrent (integer(2)::i=1:5, i.ne.3)
68    if (i.eq.2 .or. i.eq.4) then ! fir.if
69      goto 6
70    endif
71    ! CHECK: fir.call @_FortranAioBeginExternalListOutput
72    print*, 'D:', i
73  6 continue
74  enddo
75
76  ! CHECK-NOT: fir.do_loop
77  ! CHECK-NOT: fir.if
78  do concurrent (integer(2)::i=1:5, i.ne.3)
79    goto (7, 7) i+1
80    ! CHECK: fir.call @_FortranAioBeginExternalListOutput
81    print*, 'E:', i
82  7 continue
83  enddo
84
85  xsum = 0.0
86  ! CHECK-NOT: fir.do_loop
87  do x = 1.5, 3.5, 0.3
88    xsum = xsum + 1
89  enddo
90  ! CHECK: fir.call @_FortranAioBeginExternalFormattedOutput
91  print '(" F:",X,F3.1,A,I2)', x, ' -', xsum
92end subroutine loop_test
93
94! CHECK-LABEL: c.func @_QPlis
95subroutine lis(n)
96  ! CHECK-DAG: fir.alloca i32 {bindc_name = "m"}
97  ! CHECK-DAG: fir.alloca i32 {bindc_name = "j"}
98  ! CHECK-DAG: fir.alloca i32 {bindc_name = "i"}
99  ! CHECK-DAG: fir.alloca i8 {bindc_name = "i"}
100  ! CHECK-DAG: fir.alloca i32 {bindc_name = "j", uniq_name = "_QFlisEj"}
101  ! CHECK-DAG: fir.alloca i32 {bindc_name = "k", uniq_name = "_QFlisEk"}
102  ! CHECK-DAG: fir.alloca !fir.box<!fir.ptr<!fir.array<?x?x?xi32>>> {bindc_name = "p", uniq_name = "_QFlisEp"}
103  ! CHECK-DAG: fir.alloca !fir.array<?x?x?xi32>, %{{.*}}, %{{.*}}, %{{.*}} {bindc_name = "a", fir.target, uniq_name = "_QFlisEa"}
104  ! CHECK-DAG: fir.alloca !fir.array<?x?xi32>, %{{.*}}, %{{.*}} {bindc_name = "r", uniq_name = "_QFlisEr"}
105  ! CHECK-DAG: fir.alloca !fir.array<?x?xi32>, %{{.*}}, %{{.*}} {bindc_name = "s", uniq_name = "_QFlisEs"}
106  ! CHECK-DAG: fir.alloca !fir.array<?x?xi32>, %{{.*}}, %{{.*}} {bindc_name = "t", uniq_name = "_QFlisEt"}
107  integer, target    :: a(n,n,n) ! operand via p
108  integer            :: r(n,n)   ! result, unspecified locality
109  integer            :: s(n,n)   ! shared locality
110  integer            :: t(n,n)   ! local locality
111  integer, pointer   :: p(:,:,:) ! local_init locality
112
113  p => a
114  ! CHECK:     fir.do_loop %arg1 = %c0{{.*}} to %{{.*}} step %c1{{.*}} unordered iter_args(%arg2 = %{{.*}}) -> (!fir.array<?x?xi32>) {
115  ! CHECK:       fir.do_loop %arg3 = %c0{{.*}} to %{{.*}} step %c1{{.*}} unordered iter_args(%arg4 = %arg2) -> (!fir.array<?x?xi32>) {
116  ! CHECK:       }
117  ! CHECK:     }
118  r = 0
119
120  ! CHECK:     fir.do_loop %arg1 = %{{.*}} to %{{.*}} step %{{.*}} unordered {
121  ! CHECK:       fir.do_loop %arg2 = %{{.*}} to %{{.*}} step %c1{{.*}} iter_args(%arg3 = %{{.*}}) -> (index, i32) {
122  ! CHECK:       }
123  ! CHECK:     }
124  do concurrent (integer(kind=1)::i=n:1:-1)
125    do j = 1,n
126      a(i,j,:) = 2*(i+j)
127      s(i,j)   = -i-j
128    enddo
129  enddo
130
131  ! CHECK:     fir.do_loop %arg1 = %{{.*}} to %{{.*}} step %c1{{.*}} unordered {
132  ! CHECK:       fir.do_loop %arg2 = %{{.*}} to %{{.*}} step %c1{{.*}} unordered {
133  ! CHECK:         fir.if %{{.*}} {
134  ! CHECK:           %[[V_95:[0-9]+]] = fir.alloca !fir.array<?x?xi32>, %{{.*}}, %{{.*}} {bindc_name = "t", pinned, uniq_name = "_QFlisEt"}
135  ! CHECK:           %[[V_96:[0-9]+]] = fir.alloca !fir.box<!fir.ptr<!fir.array<?x?x?xi32>>> {bindc_name = "p", pinned, uniq_name = "_QFlisEp"}
136  ! CHECK:           fir.store %{{.*}} to %[[V_96]] : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?x?xi32>>>>
137  ! CHECK:           fir.do_loop %arg3 = %{{.*}} to %{{.*}} step %c1{{.*}} iter_args(%arg4 = %{{.*}}) -> (index, i32) {
138  ! CHECK:             fir.do_loop %arg5 = %{{.*}} to %{{.*}} step %c1{{.*}} unordered {
139  ! CHECK:               fir.load %[[V_96]] : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?x?xi32>>>>
140  ! CHECK:               fir.convert %[[V_95]] : (!fir.ref<!fir.array<?x?xi32>>) -> !fir.ref<!fir.array<?xi32>>
141  ! CHECK:             }
142  ! CHECK:           }
143  ! CHECK:           fir.convert %[[V_95]] : (!fir.ref<!fir.array<?x?xi32>>) -> !fir.ref<!fir.array<?xi32>>
144  ! CHECK:         }
145  ! CHECK:       }
146  ! CHECK:     }
147  do concurrent (i=1:n,j=1:n,i.ne.j) local(t) local_init(p) shared(s)
148    do k=1,n
149      do concurrent (m=1:n)
150        t(k,m) = p(k,m,k)
151      enddo
152    enddo
153    r(i,j) = t(i,j) + s(i,j)
154  enddo
155
156  print*, sum(r) ! n=6 -> 210
157end
158
159! CHECK-LABEL: print_nothing
160subroutine print_nothing(k1, k2)
161  if (k1 > 0) then
162    ! CHECK: br [[header:\^bb[0-9]+]]
163    ! CHECK: [[header]]
164    do while (k1 > k2)
165      print*, k1, k2 ! no output
166      k2 = k2 + 1
167      ! CHECK: br [[header]]
168    end do
169  end if
170end
171
172  call loop_test
173  call lis(6)
174  call print_nothing(2, 2)
175end
176