97 lines · plain
1! Test lowering of ALL intrinsic to HLFIR2! RUN: bbc -emit-hlfir -o - %s 2>&1 | FileCheck %s3 4! simple 1 argument ALL5subroutine all1(a, s)6 logical :: a(:), s7 s = ALL(a)8end subroutine9! CHECK-LABEL: func.func @_QPall1(10! CHECK: %[[ARG0:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>11! CHECK: %[[ARG1:.*]]: !fir.ref<!fir.logical<4>>12! CHECK-DAG: %[[MASK:.*]]:2 = hlfir.declare %[[ARG0]]13! CHECK-DAG: %[[OUT:.*]]:2 = hlfir.declare %[[ARG1]]14! CHECK-NEXT: %[[EXPR:.*]] = hlfir.all %[[MASK]]#0 : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> !fir.logical<4>15! CHECK-NEXT: hlfir.assign %[[EXPR]] to %[[OUT]]#0 : !fir.logical<4>, !fir.ref<!fir.logical<4>>16! CHECK-NEXT: return17! CHECK-NEXT: }18 19! all with by-ref DIM argument20subroutine all2(a, s, d)21 logical :: a(:,:), s(:)22 integer :: d23s = ALL(a, d)24end subroutine25! CHECK-LABEL: func.func @_QPall2(26! CHECK: %[[ARG0:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "a"}27! CHECK: %[[ARG1:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>> {fir.bindc_name = "s"}28! CHECK: %[[ARG2:.*]]: !fir.ref<i32> {fir.bindc_name = "d"}29! CHECK-DAG: %[[MASK:.*]]:2 = hlfir.declare %[[ARG0]]30! CHECK-DAG: %[[DIM_REF:.*]]:2 = hlfir.declare %[[ARG2]]31! CHECK-DAG: %[[OUT:.*]]:2 = hlfir.declare %[[ARG1]]32! CHECK-NEXT: %[[DIM:.*]] = fir.load %[[DIM_REF]]#0 : !fir.ref<i32>33! CHECK-NEXT: %[[EXPR:.*]] = hlfir.all %[[MASK]]#0 dim %[[DIM]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> !hlfir.expr<?x!fir.logical<4>>34! CHECK-NEXT: hlfir.assign %[[EXPR]] to %[[OUT]]#0 : !hlfir.expr<?x!fir.logical<4>>, !fir.box<!fir.array<?x!fir.logical<4>>>35! CHECK-NEXT: hlfir.destroy %[[EXPR]]36! CHECK-NEXT: return37! CHECK-NEXT: }38 39! all with DIM argument by-val, mask isn't boxed40subroutine all3(s)41 logical :: s(2)42 logical :: a(2,2) = reshape((/.true.,.false.,.true.,.false./), shape(a))43s = ALL(a, 1)44end subroutine45! CHECK-LABEL: func.func @_QPall3(46! CHECK: %[[ARG0:.*]]: !fir.ref<!fir.array<2x!fir.logical<4>>> {fir.bindc_name = "s"}47! CHECK-DAG: %[[ADDR:.*]] = fir.address_of{{.*}} : !fir.ref<!fir.array<2x2x!fir.logical<4>>>48! CHECK-DAG: %[[MASK_SHAPE:.*]] = fir.shape {{.*}} -> !fir.shape<2>49! CHECK-DAG: %[[MASK:.*]]:2 = hlfir.declare %[[ADDR]](%[[MASK_SHAPE]])50! CHECK-DAG: %[[OUT_SHAPE:.*]] = fir.shape {{.*}} -> !fir.shape<1>51! CHECK-DAG: %[[OUT:.*]]:2 = hlfir.declare %[[ARG0]](%[[OUT_SHAPE]])52! CHECK-DAG: %[[C1:.*]] = arith.constant 1 : i3253! CHECK-DAG: %[[EXPR:.*]] = hlfir.all %[[MASK]]#0 dim %[[C1]] : (!fir.ref<!fir.array<2x2x!fir.logical<4>>>, i32) -> !hlfir.expr<2x!fir.logical<4>>54! CHECK-DAG: hlfir.assign %[[EXPR]] to %[[OUT]]55! CHECK-NEXT: hlfir.destroy %[[EXPR]] : !hlfir.expr<2x!fir.logical<4>>56! CHECK-NEXT: return57! CHECK-NEXT: }58 59! all with DIM from pointer60subroutine all4(a, s, d)61 integer, pointer :: d62 logical :: a(:,:), s(:)63 s = ALL(a, (d))64end subroutine65! CHECK-LABEL: func.func @_QPall4(66! CHECK: %[[ARG0:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>> {fir.bindc_name = "a"}67! CHECK: %[[ARG1:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>> {fir.bindc_name = "s"}68! CHECK: %[[ARG2:.*]]: !fir.ref<!fir.box<!fir.ptr<i32>>> {fir.bindc_name = "d"}69! CHECK-DAG: %[[ARRAY:.*]]:2 = hlfir.declare %[[ARG0]]70! CHECK-DAG: %[[OUT:.*]]:2 = hlfir.declare %[[ARG1]]71! CHECK-DAG: %[[DIM:.*]]:2 = hlfir.declare %[[ARG2]]72! CHECK-NEXT: %[[DIM_BOX:.*]] = fir.load %[[DIM]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>73! CHECK-NEXT: %[[DIM_ADDR:.*]] = fir.box_addr %[[DIM_BOX]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>74! CHECK-NEXT: %[[DIM0:.*]] = fir.load %[[DIM_ADDR]] : !fir.ptr<i32>75! CHECK-NEXT: %[[DIM1:.*]] = hlfir.no_reassoc %[[DIM0]] : i3276! CHECK-NEXT: %[[EXPR:.*]] = hlfir.all %[[ARRAY]]#0 dim %[[DIM1]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> !hlfir.expr<?x!fir.logical<4>>77! CHECK-NEXT: hlfir.assign %[[EXPR]] to %[[OUT]]#0 : !hlfir.expr<?x!fir.logical<4>>, !fir.box<!fir.array<?x!fir.logical<4>>>78! CHECK-NEXT: hlfir.destroy %[[EXPR]] : !hlfir.expr<?x!fir.logical<4>>79! CHECK-NEXT: return80! CHECK-NEXT: }81 82subroutine all5(a, s)83 logical, allocatable :: a(:)84 logical :: s85 s = ALL(a)86end subroutine87! CHECK-LABEL: func.func @_QPall5(88! CHECK: %[[ARG0:.*]]: !fir.ref<!fir.box<!fir.heap<!fir.array<?x!fir.logical<4>>>>>89! CHECK: %[[ARG1:.*]]: !fir.ref<!fir.logical<4>>90! CHECK-DAG: %[[MASK:.*]]:2 = hlfir.declare %[[ARG0]]91! CHECK-DAG: %[[OUT:.*]]:2 = hlfir.declare %[[ARG1]]92! CHECK-NEXT: %[[MASK_LOADED:.*]] = fir.load %[[MASK]]#093! CHECK-NEXT: %[[EXPR:.*]] = hlfir.all %[[MASK_LOADED]] : (!fir.box<!fir.heap<!fir.array<?x!fir.logical<4>>>>) -> !fir.logical<4>94! CHECK-NEXT: hlfir.assign %[[EXPR]] to %[[OUT]]#0 : !fir.logical<4>, !fir.ref<!fir.logical<4>>95! CHECK-NEXT: return96! CHECK-NEXT: }97