! RUN: %flang_fc1 -emit-hlfir %s -o - | FileCheck %s ! CHECK-LABEL: func @_QPiall_test_1( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}) -> i8 { integer(1) function iall_test_1(a) integer(1) :: a(:) ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[RES_ALLOCA:.*]] = fir.alloca i8 ! CHECK-DAG: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] {{.*}} : (!fir.ref) -> (!fir.ref, !fir.ref) ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index ! CHECK-DAG: %[[CONV_ARG:.*]] = fir.convert %[[ARG_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_C0:.*]] = fir.convert %[[C0]] : (index) -> i32 ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box iall_test_1 = iall(a) ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll1(%[[CONV_ARG]], %{{.*}}, %{{.*}}, %[[CONV_C0]], %[[CONV_ABSENT]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i8 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i8, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_2( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}) -> i16 { integer(2) function iall_test_2(a) integer(2) :: a(:) ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[RES_ALLOCA:.*]] = fir.alloca i16 ! CHECK-DAG: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] {{.*}} : (!fir.ref) -> (!fir.ref, !fir.ref) ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index ! CHECK-DAG: %[[CONV_ARG:.*]] = fir.convert %[[ARG_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_C0:.*]] = fir.convert %[[C0]] : (index) -> i32 ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box iall_test_2 = iall(a) ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll2(%[[CONV_ARG]], %{{.*}}, %{{.*}}, %[[CONV_C0]], %[[CONV_ABSENT]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i16 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i16, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_4( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}) -> i32 { integer function iall_test_4(a) integer :: a(:) ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[RES_ALLOCA:.*]] = fir.alloca i32 ! CHECK-DAG: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] {{.*}} : (!fir.ref) -> (!fir.ref, !fir.ref) ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index ! CHECK-DAG: %[[CONV_ARG:.*]] = fir.convert %[[ARG_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_C0:.*]] = fir.convert %[[C0]] : (index) -> i32 ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box iall_test_4 = iall(a) ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll4(%[[CONV_ARG]], %{{.*}}, %{{.*}}, %[[CONV_C0]], %[[CONV_ABSENT]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i32 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i32, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_8( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}) -> i64 { integer(8) function iall_test_8(a) integer(8) :: a(:) ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[RES_ALLOCA:.*]] = fir.alloca i64 ! CHECK-DAG: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] {{.*}} : (!fir.ref) -> (!fir.ref, !fir.ref) ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index ! CHECK-DAG: %[[CONV_ARG:.*]] = fir.convert %[[ARG_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_C0:.*]] = fir.convert %[[C0]] : (index) -> i32 ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box iall_test_8 = iall(a) ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll8(%[[CONV_ARG]], %{{.*}}, %{{.*}}, %[[CONV_C0]], %[[CONV_ABSENT]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i64 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i64, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_16( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}) -> i128 { integer(16) function iall_test_16(a) integer(16) :: a(:) ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[RES_ALLOCA:.*]] = fir.alloca i128 ! CHECK-DAG: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] {{.*}} : (!fir.ref) -> (!fir.ref, !fir.ref) ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index ! CHECK-DAG: %[[CONV_ARG:.*]] = fir.convert %[[ARG_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_C0:.*]] = fir.convert %[[C0]] : (index) -> i32 ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box iall_test_16 = iall(a) ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll16(%[[CONV_ARG]], %{{.*}}, %{{.*}}, %[[CONV_C0]], %[[CONV_ABSENT]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i128 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i128, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test2( ! CHECK-SAME: %[[ARG0:.*]]: !fir.box>{{.*}}, %[[ARG1:.*]]: !fir.box>{{.*}}) { subroutine iall_test2(a,r) integer :: a(:,:) integer :: r(:) ! CHECK-DAG: %[[BOX_ALLOCA:.*]] = fir.alloca !fir.box>> ! CHECK-DAG: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK-DAG: %[[ARG0_DECL:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[SCOPE]] arg 1 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[ARG1_DECL:.*]]:2 = hlfir.declare %[[ARG1]] dummy_scope %[[SCOPE]] arg 2 {{.*}} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) ! CHECK-DAG: %[[C2:.*]] = arith.constant 2 : i32 ! CHECK-DAG: %[[ABSENT:.*]] = fir.absent !fir.box ! CHECK-DAG: %[[CONV_BOX:.*]] = fir.convert %[[BOX_ALLOCA]] : (!fir.ref>>>) -> !fir.ref> ! CHECK-DAG: %[[CONV_ARG0:.*]] = fir.convert %[[ARG0_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK-DAG: %[[CONV_ABSENT:.*]] = fir.convert %[[ABSENT]] : (!fir.box) -> !fir.box r = iall(a,dim=2) ! CHECK: fir.call @_FortranAIAllDim(%[[CONV_BOX]], %[[CONV_ARG0]], %[[C2]], %{{.*}}, %{{.*}}, %[[CONV_ABSENT]]) fastmath : (!fir.ref>, !fir.box, i32, !fir.ref, i32, !fir.box) -> () ! CHECK: %[[BOX_LOAD:.*]] = fir.load %[[BOX_ALLOCA]] : !fir.ref>>> ! CHECK: %[[ADDR:.*]] = fir.box_addr %[[BOX_LOAD]] : (!fir.box>>) -> !fir.heap> ! CHECK: %[[TMP_DECL:.*]]:2 = hlfir.declare %[[ADDR]]({{.*}}) {uniq_name = ".tmp.intrinsic_result"} : (!fir.heap>, !fir.shapeshift<1>) -> (!fir.box>, !fir.heap>) ! CHECK: %[[EXPR:.*]] = hlfir.as_expr %[[TMP_DECL]]#0 move %{{.*}} : (!fir.box>, i1) -> !hlfir.expr ! CHECK: hlfir.assign %[[EXPR]] to %[[ARG1_DECL]]#0 : !hlfir.expr, !fir.box> ! CHECK: hlfir.destroy %[[EXPR]] : !hlfir.expr end subroutine ! CHECK-LABEL: func @_QPiall_test_optional( ! CHECK-SAME: %[[MASK:.*]]: !fir.box>>{{.*}}, %[[X:.*]]: !fir.box>{{.*}}) -> i32 { integer function iall_test_optional(mask, x) integer :: x(:) logical, optional :: mask(:) ! CHECK: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK: %[[RES_ALLOCA:.*]] = fir.alloca i32 ! CHECK: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] ! CHECK: %[[MASK_DECL:.*]]:2 = hlfir.declare %[[MASK]] dummy_scope %[[SCOPE]] arg 1 {fortran_attrs = #fir.var_attrs, {{.*}}} ! CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] dummy_scope %[[SCOPE]] arg 2 {{.*}} iall_test_optional = iall(x, mask=mask) ! CHECK: %[[CONV_X:.*]] = fir.convert %[[X_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK: %[[CONV_MASK:.*]] = fir.convert %[[MASK_DECL]]#1 : (!fir.box>>) -> !fir.box ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll4(%[[CONV_X]], %{{.*}}, %{{.*}}, %{{.*}}, %[[CONV_MASK]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i32 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i32, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_optional_2( ! CHECK-SAME: %[[MASK:.*]]: !fir.ref>>>>{{.*}}, %[[X:.*]]: !fir.box>{{.*}}) -> i32 { integer function iall_test_optional_2(mask, x) integer :: x(:) logical, pointer :: mask(:) ! CHECK: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK: %[[RES_ALLOCA:.*]] = fir.alloca i32 ! CHECK: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] ! CHECK: %[[MASK_DECL:.*]]:2 = hlfir.declare %[[MASK]] dummy_scope %[[SCOPE]] arg 1 {fortran_attrs = #fir.var_attrs, {{.*}}} ! CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] dummy_scope %[[SCOPE]] arg 2 {{.*}} iall_test_optional_2 = iall(x, mask=mask) ! CHECK: %[[MASK_LOAD1:.*]] = fir.load %[[MASK_DECL]]#0 : !fir.ref>>>> ! CHECK: %[[MASK_ADDR:.*]] = fir.box_addr %[[MASK_LOAD1]] : (!fir.box>>>) -> !fir.ptr>> ! CHECK: %[[MASK_PTR_CONV:.*]] = fir.convert %[[MASK_ADDR]] : (!fir.ptr>>) -> i64 ! CHECK: %[[C0_I64:.*]] = arith.constant 0 : i64 ! CHECK: %[[IS_PRESENT:.*]] = arith.cmpi ne, %[[MASK_PTR_CONV]], %[[C0_I64]] : i64 ! CHECK: %[[MASK_LOAD2:.*]] = fir.load %[[MASK_DECL]]#0 : !fir.ref>>>> ! CHECK: %[[ABSENT:.*]] = fir.absent !fir.box>>> ! CHECK: %[[SELECTED:.*]] = arith.select %[[IS_PRESENT]], %[[MASK_LOAD2]], %[[ABSENT]] : !fir.box>>> ! CHECK: %[[CONV_X:.*]] = fir.convert %[[X_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK: %[[CONV_MASK:.*]] = fir.convert %[[SELECTED]] : (!fir.box>>>) -> !fir.box ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll4(%[[CONV_X]], %{{.*}}, %{{.*}}, %{{.*}}, %[[CONV_MASK]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i32 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i32, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_optional_3( ! CHECK-SAME: %[[MASK:.*]]: !fir.ref>>{{.*}}, %[[X:.*]]: !fir.box>{{.*}}) -> i32 { integer function iall_test_optional_3(mask, x) integer :: x(:) logical, optional :: mask(10) ! CHECK: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK: %[[RES_ALLOCA:.*]] = fir.alloca i32 ! CHECK: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] ! CHECK: %[[MASK_DECL:.*]]:2 = hlfir.declare %[[MASK]]({{.*}}) dummy_scope %[[SCOPE]] arg 1 {fortran_attrs = #fir.var_attrs, {{.*}}} ! CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] dummy_scope %[[SCOPE]] arg 2 {{.*}} iall_test_optional_3 = iall(x, mask=mask) ! CHECK: %[[IS_PRESENT:.*]] = fir.is_present %[[MASK_DECL]]#0 : (!fir.ref>>) -> i1 ! CHECK: %[[EMBOX:.*]] = fir.embox %[[MASK_DECL]]#0(%{{.*}}) : (!fir.ref>>, !fir.shape<1>) -> !fir.box>> ! CHECK: %[[ABSENT:.*]] = fir.absent !fir.box>> ! CHECK: %[[SELECTED:.*]] = arith.select %[[IS_PRESENT]], %[[EMBOX]], %[[ABSENT]] : !fir.box>> ! CHECK: %[[CONV_X:.*]] = fir.convert %[[X_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK: %[[CONV_MASK:.*]] = fir.convert %[[SELECTED]] : (!fir.box>>) -> !fir.box ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll4(%[[CONV_X]], %{{.*}}, %{{.*}}, %{{.*}}, %[[CONV_MASK]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i32 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i32, !fir.ref end function ! CHECK-LABEL: func @_QPiall_test_optional_4( ! CHECK-SAME: %[[X:.*]]: !fir.box>{{.*}}, %[[USE_MASK:.*]]: !fir.ref>{{.*}}) -> i32 { integer function iall_test_optional_4(x, use_mask) ! Test that local allocatable tracked in local variables ! are dealt as optional argument correctly. integer :: x(:) logical :: use_mask logical, allocatable :: mask(:) ! CHECK: %[[SCOPE:.*]] = fir.dummy_scope : !fir.dscope ! CHECK: %[[RES_ALLOCA:.*]] = fir.alloca i32 ! CHECK: %[[RES_DECL:.*]]:2 = hlfir.declare %[[RES_ALLOCA]] ! CHECK: %[[MASK_ALLOCA:.*]] = fir.alloca !fir.box>>> ! CHECK: %[[MASK_DECL:.*]]:2 = hlfir.declare %[[MASK_ALLOCA]] {fortran_attrs = #fir.var_attrs, {{.*}}} ! CHECK: %[[USE_MASK_DECL:.*]]:2 = hlfir.declare %[[USE_MASK]] dummy_scope %[[SCOPE]] arg 2 {{.*}} ! CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] dummy_scope %[[SCOPE]] arg 1 {{.*}} if (use_mask) then allocate(mask(size(x, 1))) call set_mask(mask) ! CHECK: fir.call @_QPset_mask(%{{.*}}) fastmath : (!fir.ref>>) -> () end if iall_test_optional_4 = iall(x, mask=mask) ! CHECK: %[[MASK_LOAD1:.*]] = fir.load %[[MASK_DECL]]#0 : !fir.ref>>>> ! CHECK: %[[MASK_ADDR1:.*]] = fir.box_addr %[[MASK_LOAD1]] : (!fir.box>>>) -> !fir.heap>> ! CHECK: %[[MASK_PTR_CONV:.*]] = fir.convert %[[MASK_ADDR1]] : (!fir.heap>>) -> i64 ! CHECK: %[[C0_I64:.*]] = arith.constant 0 : i64 ! CHECK: %[[IS_ALLOCATED:.*]] = arith.cmpi ne, %[[MASK_PTR_CONV]], %[[C0_I64]] : i64 ! CHECK: %[[MASK_LOAD2:.*]] = fir.load %[[MASK_DECL]]#0 : !fir.ref>>>> ! CHECK: %[[MASK_ADDR2:.*]] = fir.box_addr %[[MASK_LOAD2]] : (!fir.box>>>) -> !fir.heap>> ! CHECK: %[[EMBOX:.*]] = fir.embox %[[MASK_ADDR2]](%{{.*}}) : (!fir.heap>>, !fir.shapeshift<1>) -> !fir.box>> ! CHECK: %[[ABSENT:.*]] = fir.absent !fir.box>> ! CHECK: %[[SELECTED:.*]] = arith.select %[[IS_ALLOCATED]], %[[EMBOX]], %[[ABSENT]] : !fir.box>> ! CHECK: %[[CONV_X:.*]] = fir.convert %[[X_DECL]]#1 : (!fir.box>) -> !fir.box ! CHECK: %[[CONV_MASK:.*]] = fir.convert %[[SELECTED]] : (!fir.box>>) -> !fir.box ! CHECK: %[[RESULT:.*]] = fir.call @_FortranAIAll4(%[[CONV_X]], %{{.*}}, %{{.*}}, %{{.*}}, %[[CONV_MASK]]) fastmath : (!fir.box, !fir.ref, i32, i32, !fir.box) -> i32 ! CHECK: hlfir.assign %[[RESULT]] to %[[RES_DECL]]#0 : i32, !fir.ref end function