128 lines · plain
1! RUN: %flang_fc1 -emit-hlfir -o - %s | FileCheck %s2 3! Simple tests for structured concurrent loops with loop-control.4 5pure function bar(n, m)6 implicit none7 integer, intent(in) :: n, m8 integer :: bar9 bar = n + m10end function11 12!CHECK-LABEL: sub113subroutine sub1(n)14 implicit none15 integer :: n, m, i, j, k16 integer, dimension(n) :: a17!CHECK: %[[N_DECL:.*]]:2 = hlfir.declare %{{.*}} dummy_scope %{{.*}} {uniq_name = "_QFsub1En"}18!CHECK: %[[A_DECL:.*]]:2 = hlfir.declare %{{.*}}(%{{.*}}) {uniq_name = "_QFsub1Ea"}19 20!CHECK: %[[LB1:.*]] = arith.constant 1 : i3221!CHECK: %[[LB1_CVT:.*]] = fir.convert %[[LB1]] : (i32) -> index22!CHECK: %[[UB1:.*]] = fir.load %{{.*}}#0 : !fir.ref<i32>23!CHECK: %[[UB1_CVT:.*]] = fir.convert %[[UB1]] : (i32) -> index24 25!CHECK: %[[LB2:.*]] = arith.constant 1 : i3226!CHECK: %[[LB2_CVT:.*]] = fir.convert %[[LB2]] : (i32) -> index27!CHECK: %[[UB2:.*]] = fir.call @_QPbar(%{{.*}}, %{{.*}}) proc_attrs<pure> fastmath<contract> : (!fir.ref<i32>, !fir.ref<i32>) -> i3228!CHECK: %[[UB2_CVT:.*]] = fir.convert %[[UB2]] : (i32) -> index29 30!CHECK: %[[LB3:.*]] = arith.constant 5 : i3231!CHECK: %[[LB3_CVT:.*]] = fir.convert %[[LB3]] : (i32) -> index32!CHECK: %[[UB3:.*]] = arith.constant 10 : i3233!CHECK: %[[UB3_CVT:.*]] = fir.convert %[[UB3]] : (i32) -> index34 35!CHECK: fir.do_concurrent36!CHECK: %[[I:.*]] = fir.alloca i32 {bindc_name = "i"}37!CHECK: %[[I_DECL:.*]]:2 = hlfir.declare %[[I]]38!CHECK: %[[J:.*]] = fir.alloca i32 {bindc_name = "j"}39!CHECK: %[[J_DECL:.*]]:2 = hlfir.declare %[[J]]40!CHECK: %[[K:.*]] = fir.alloca i32 {bindc_name = "k"}41!CHECK: %[[K_DECL:.*]]:2 = hlfir.declare %[[K]]42 43!CHECK: fir.do_concurrent.loop (%[[I_IV:.*]], %[[J_IV:.*]], %[[K_IV:.*]]) =44!CHECK-SAME: (%[[LB1_CVT]], %[[LB2_CVT]], %[[LB3_CVT]]) to45!CHECK-SAME: (%[[UB1_CVT]], %[[UB2_CVT]], %[[UB3_CVT]]) step46!CHECK-SAME: (%{{.*}}, %{{.*}}, %{{.*}}) {47!CHECK: %[[I_IV_CVT:.*]] = fir.convert %[[I_IV]] : (index) -> i3248!CHECK: fir.store %[[I_IV_CVT]] to %[[I_DECL]]#0 : !fir.ref<i32>49!CHECK: %[[J_IV_CVT:.*]] = fir.convert %[[J_IV]] : (index) -> i3250!CHECK: fir.store %[[J_IV_CVT]] to %[[J_DECL]]#0 : !fir.ref<i32>51!CHECK: %[[K_IV_CVT:.*]] = fir.convert %[[K_IV]] : (index) -> i3252!CHECK: fir.store %[[K_IV_CVT]] to %[[K_DECL]]#0 : !fir.ref<i32>53 54!CHECK: %[[N_VAL:.*]] = fir.load %[[N_DECL]]#0 : !fir.ref<i32>55!CHECK: %[[I_VAL:.*]] = fir.load %[[I_DECL]]#0 : !fir.ref<i32>56!CHECK: %[[I_VAL_CVT:.*]] = fir.convert %[[I_VAL]] : (i32) -> i6457!CHECK: %[[A_ELEM:.*]] = hlfir.designate %[[A_DECL]]#0 (%[[I_VAL_CVT]])58!CHECK: hlfir.assign %[[N_VAL]] to %[[A_ELEM]] : i32, !fir.ref<i32>59 do concurrent(i=1:n, j=1:bar(n*m, n/m), k=5:10)60 a(i) = n61 end do62end subroutine63 64!CHECK-LABEL: sub265subroutine sub2(n)66 implicit none67 integer :: n, m, i, j68 integer, dimension(n) :: a69!CHECK: %[[LB1:.*]] = arith.constant 1 : i3270!CHECK: %[[LB1_CVT:.*]] = fir.convert %[[LB1]] : (i32) -> index71!CHECK: %[[UB1:.*]] = fir.load %{{.*}}#0 : !fir.ref<i32>72!CHECK: %[[UB1_CVT:.*]] = fir.convert %[[UB1]] : (i32) -> index73!CHECK: fir.do_concurrent74!CHECK: fir.do_concurrent.loop (%{{.*}}) = (%[[LB1_CVT]]) to (%[[UB1_CVT]]) step (%{{.*}})75 76!CHECK: %[[LB2:.*]] = arith.constant 1 : i3277!CHECK: %[[LB2_CVT:.*]] = fir.convert %[[LB2]] : (i32) -> index78!CHECK: %[[UB2:.*]] = fir.call @_QPbar(%{{.*}}, %{{.*}}) proc_attrs<pure> fastmath<contract> : (!fir.ref<i32>, !fir.ref<i32>) -> i3279!CHECK: %[[UB2_CVT:.*]] = fir.convert %[[UB2]] : (i32) -> index80!CHECK: fir.do_concurrent81!CHECK: fir.do_concurrent.loop (%{{.*}}) = (%[[LB2_CVT]]) to (%[[UB2_CVT]]) step (%{{.*}})82 do concurrent(i=1:n)83 do concurrent(j=1:bar(n*m, n/m))84 a(i) = n85 end do86 end do87end subroutine88 89!CHECK-LABEL: unstructured90subroutine unstructured(inner_step)91 integer(4) :: i, j, inner_step92 93!CHECK-NOT: cf.br94!CHECK-NOT: cf.cond_br95!CHECK: %[[LB1:.*]] = arith.constant 1 : i3296!CHECK: %[[LB1_CVT:.*]] = fir.convert %c1_i32 : (i32) -> i1697!CHECK: %[[UB1:.*]] = arith.constant 5 : i3298!CHECK: %[[UB1_CVT:.*]] = fir.convert %c5_i32 : (i32) -> i1699!CHECK: %[[STP1:.*]] = arith.constant 1 : i16100 101!CHECK-NOT: cf.br102!CHECK-NOT: cf.cond_br103!CHECK: %[[LB2:.*]] = arith.constant 3 : i32104!CHECK: %[[LB2_CVT:.*]] = fir.convert %[[LB2]] : (i32) -> i16105!CHECK: %[[UB2:.*]] = arith.constant 9 : i32106!CHECK: %[[UB2_CVT:.*]] = fir.convert %[[UB2]] : (i32) -> i16107!CHECK: %[[STP2:.*]] = fir.load %{{.*}}#0 : !fir.ref<i32>108!CHECK: %[[STP2_CVT:.*]] = fir.convert %[[STP2]] : (i32) -> i16109!CHECK: fir.store %[[STP2_CVT]] to %{{.*}} : !fir.ref<i16>110!CHECK: cf.br ^[[I_LOOP_HEADER:.*]]111 112!CHECK: ^[[I_LOOP_HEADER]]:113!CHECK-NEXT: %{{.*}} = fir.load %{{.*}} : !fir.ref<i16>114!CHECK-NEXT: %{{.*}} = arith.constant 0 : i16115!CHECK-NEXT: %{{.*}} = arith.cmpi sgt, %{{.*}}, %{{.*}}: i16116!CHECK-NEXT: cf.cond_br %{{.*}}, ^[[J_LOOP_HEADER:.*]], ^{{.*}}117 118!CHECK: ^[[J_LOOP_HEADER]]:119!CHECK-NEXT: %[[RANGE:.*]] = arith.subi %[[UB2_CVT]], %[[LB2_CVT]] : i16120!CHECK-NEXT: %{{.*}} = arith.addi %[[RANGE]], %[[STP2_CVT]] : i16121!CHECK-NEXT: %{{.*}} = arith.divsi %{{.*}}, %[[STP2_CVT]] : i16122 do concurrent (integer(2)::i=1:5, j=3:9:inner_step, i.ne.3)123 goto (7, 7) i+1124 print*, 'E:', i, j125 7 continue126 enddo127end subroutine unstructured128