100 lines · plain
1! RUN: bbc -emit-fir -hlfir=false %s -o - | FileCheck %s2! RUN: %flang_fc1 -emit-fir -flang-deprecated-no-hlfir %s -o - | FileCheck %s3 4! CHECK-LABEL: func @_QPsize_test() {5subroutine size_test()6 real, dimension(1:10, -10:10) :: a7 integer :: dim = 18 integer :: iSize9! CHECK: %[[VAL_c1:.*]] = arith.constant 1 : index10! CHECK: %[[VAL_c10:.*]] = arith.constant 10 : index11! CHECK: %[[VAL_cneg10:.*]] = arith.constant -10 : index12! CHECK: %[[VAL_c21:.*]] = arith.constant 21 : index13! CHECK: %[[VAL_0:.*]] = fir.alloca !fir.array<10x21xf32> {bindc_name = "a", uniq_name = "_QFsize_testEa"}14! CHECK: %[[VAL_1:.*]] = fir.address_of(@_QFsize_testEdim) : !fir.ref<i32>15! CHECK: %[[VAL_2:.*]] = fir.alloca i32 {bindc_name = "isize", uniq_name = "_QFsize_testEisize"}16! CHECK: %[[VAL_c2_i64:.*]] = arith.constant 2 : i6417! CHECK: %[[VAL_3:.*]] = fir.convert %[[VAL_c2_i64]] : (i64) -> index18! CHECK: %[[VAL_c1_i64:.*]] = arith.constant 1 : i6419! CHECK: %[[VAL_4:.*]] = fir.convert %[[VAL_c1_i64]] : (i64) -> index20! CHECK: %[[VAL_c5_i64:.*]] = arith.constant 5 : i6421! CHECK: %[[VAL_5:.*]] = fir.convert %[[VAL_c5_i64]] : (i64) -> index22! CHECK: %[[VAL_neg1_i64:.*]] = arith.constant -1 : i6423! CHECK: %[[VAL_6:.*]] = fir.convert %[[VAL_neg1_i64]] : (i64) -> index24! CHECK: %[[VAL_c1_i64_0:.*]] = arith.constant 1 : i6425! CHECK: %[[VAL_7:.*]] = fir.convert %[[VAL_c1_i64_0]] : (i64) -> index26! CHECK: %[[VAL_c1_i64_1:.*]] = arith.constant 1 : i6427! CHECK: %[[VAL_8:.*]] = fir.convert %[[VAL_c1_i64_1]] : (i64) -> index28! CHECK: %[[VAL_9:.*]] = fir.shape_shift %[[VAL_c1]], %[[VAL_c10]], %[[VAL_cneg10]], %[[VAL_c21]] : (index, index, index, index) -> !fir.shapeshift<2>29! CHECK: %[[VAL_10:.*]] = fir.slice %[[VAL_3]], %[[VAL_5]], %[[VAL_4]], %[[VAL_6]], %[[VAL_8]], %[[VAL_7]] : (index, index, index, index, index, index) -> !fir.slice<2>30! CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_0]](%[[VAL_9]]) [%[[VAL_10]]] : (!fir.ref<!fir.array<10x21xf32>>, !fir.shapeshift<2>, !fir.slice<2>) -> !fir.box<!fir.array<4x3xf32>>31! CHECK: %[[VAL_12:.*]] = fir.convert %[[VAL_1]] : (!fir.ref<i32>) -> i6432! CHECK: %[[c0_i64:.*]] = arith.constant 0 : i6433! CHECK: %[[VAL_13:.*]] = arith.cmpi eq, %[[VAL_12]], %[[c0_i64]] : i6434! CHECK: %[[VAL_14:.*]] = fir.if %[[VAL_13]] -> (i64) {35! CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_11]] : (!fir.box<!fir.array<4x3xf32>>) -> !fir.box<none>36! CHECK: %[[VAL_19:.*]] = fir.call @_FortranASize(%[[VAL_17]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, !fir.ref<i8>, i32) -> i6437! CHECK: fir.result %[[VAL_19]] : i6438! CHECK: } else {39! CHECK: %[[VAL_16:.*]] = fir.load %[[VAL_1]] : !fir.ref<i32>40! CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_11]] : (!fir.box<!fir.array<4x3xf32>>) -> !fir.box<none>41! CHECK: %[[VAL_20:.*]] = fir.call @_FortranASizeDim(%[[VAL_18]], %[[VAL_16]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, i32, !fir.ref<i8>, i32) -> i6442! CHECK: fir.result %[[VAL_20]] : i6443! CHECK: }44! CHECK: %[[VAL_21:.*]] = fir.convert %[[VAL_14]] : (i64) -> i3245! CHECK: fir.store %[[VAL_21]] to %[[VAL_2]] : !fir.ref<i32>46 iSize = size(a(2:5, -1:1), dim, 8)47end subroutine size_test48 49! CHECK-LABEL: func @_QPsize_optional_dim_1(50! CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?x?xf32>> {fir.bindc_name = "array"},51! CHECK-SAME: %[[VAL_1:.*]]: !fir.ref<i32> {fir.bindc_name = "dim", fir.optional},52! CHECK-SAME: %[[VAL_2:.*]]: !fir.ref<i64> {fir.bindc_name = "isize"}) {53subroutine size_optional_dim_1(array, dim, iSize)54 real, dimension(:,:) :: array55 integer, optional :: dim56 integer(8) :: iSize57 iSize = size(array, dim, 8)58! CHECK: %[[VAL_3:.*]] = fir.convert %[[VAL_1]] : (!fir.ref<i32>) -> i6459! CHECK: %[[VAL_4:.*]] = arith.constant 0 : i6460! CHECK: %[[VAL_5:.*]] = arith.cmpi eq, %[[VAL_3]], %[[VAL_4]] : i6461! CHECK: %[[VAL_6:.*]] = fir.if %[[VAL_5]] -> (i64) {62! CHECK: %[[VAL_9:.*]] = fir.convert %[[VAL_0]] : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<none>63! CHECK: %[[VAL_11:.*]] = fir.call @_FortranASize(%[[VAL_9]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, !fir.ref<i8>, i32) -> i6464! CHECK: fir.result %[[VAL_11]] : i6465! CHECK: } else {66! CHECK: %[[VAL_12:.*]] = fir.load %[[VAL_1]] : !fir.ref<i32>67! CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_0]] : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<none>68! CHECK: %[[VAL_17:.*]] = fir.call @_FortranASizeDim(%[[VAL_15]], %[[VAL_12]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, i32, !fir.ref<i8>, i32) -> i6469! CHECK: fir.result %[[VAL_17]] : i6470! CHECK: }71! CHECK: fir.store %[[VAL_6]] to %[[VAL_2]] : !fir.ref<i64>72end subroutine73 74! CHECK-LABEL: func @_QPsize_optional_dim_2(75! CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?x?xf32>> {fir.bindc_name = "array"},76! CHECK-SAME: %[[VAL_1:.*]]: !fir.ref<!fir.box<!fir.ptr<i32>>> {fir.bindc_name = "dim"},77! CHECK-SAME: %[[VAL_2:.*]]: !fir.ref<i64> {fir.bindc_name = "isize"}) {78subroutine size_optional_dim_2(array, dim, iSize)79 real, dimension(:,:) :: array80 integer, pointer :: dim81 integer(8) :: iSize82 iSize = size(array, dim, 8)83! CHECK: %[[VAL_3:.*]] = fir.load %[[VAL_1]] : !fir.ref<!fir.box<!fir.ptr<i32>>>84! CHECK: %[[VAL_4:.*]] = fir.box_addr %[[VAL_3]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>85! CHECK: %[[VAL_5:.*]] = fir.convert %[[VAL_4]] : (!fir.ptr<i32>) -> i6486! CHECK: %[[VAL_6:.*]] = arith.constant 0 : i6487! CHECK: %[[VAL_7:.*]] = arith.cmpi eq, %[[VAL_5]], %[[VAL_6]] : i6488! CHECK: %[[VAL_8:.*]] = fir.if %[[VAL_7]] -> (i64) {89! CHECK: %[[VAL_11:.*]] = fir.convert %[[VAL_0]] : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<none>90! CHECK: %[[VAL_13:.*]] = fir.call @_FortranASize(%[[VAL_11]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, !fir.ref<i8>, i32) -> i6491! CHECK: fir.result %[[VAL_13]] : i6492! CHECK: } else {93! CHECK: %[[VAL_14:.*]] = fir.load %[[VAL_4]] : !fir.ptr<i32>94! CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_0]] : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<none>95! CHECK: %[[VAL_19:.*]] = fir.call @_FortranASizeDim(%[[VAL_17]], %[[VAL_14]], %{{.*}}, %{{.*}}) {{.*}}: (!fir.box<none>, i32, !fir.ref<i8>, i32) -> i6496! CHECK: fir.result %[[VAL_19]] : i6497! CHECK: }98! CHECK: fir.store %[[VAL_8]] to %[[VAL_2]] : !fir.ref<i64>99end subroutine100