194 lines · plain
1! Test lowering of vector subscripted designators in assignment2! left-hand sides.3! RUN: bbc -emit-hlfir -o - -I nw %s 2>&1 | FileCheck %s4 5subroutine test_simple(x, vector)6 integer(8) :: vector(10)7 real :: x(:)8 x(vector) = 42.9end subroutine10! CHECK-LABEL: func.func @_QPtest_simple(11! CHECK: %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}Evector12! CHECK: %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Ex13! CHECK: hlfir.region_assign {14! CHECK: %[[VAL_6:.*]] = arith.constant 4.200000e+01 : f3215! CHECK: hlfir.yield %[[VAL_6]] : f3216! CHECK: } to {17! CHECK: %[[VAL_7:.*]] = arith.constant 10 : index18! CHECK: %[[VAL_8:.*]] = fir.shape %[[VAL_7]] : (index) -> !fir.shape<1>19! CHECK: hlfir.elemental_addr %[[VAL_8]] unordered : !fir.shape<1> {20! CHECK: ^bb0(%[[VAL_9:.*]]: index):21! CHECK: %[[VAL_10:.*]] = hlfir.designate %[[VAL_4]]#0 (%[[VAL_9]]) : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>22! CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_10]] : !fir.ref<i64>23! CHECK: %[[VAL_12:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_11]]) : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>24! CHECK: hlfir.yield %[[VAL_12]] : !fir.ref<f32>25! CHECK: }26! CHECK: }27 28subroutine test_cleanup(x, vector, matrix)29 integer(8) :: vector(10), matrix(10, 5)30 real :: x(:)31 x(matmul(vector, matrix)) = 42.32end subroutine33! CHECK-LABEL: func.func @_QPtest_cleanup(34! CHECK: %[[VAL_6:.*]]:2 = hlfir.declare {{.*}}Ematrix35! CHECK: %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Evector36! CHECK: %[[VAL_10:.*]]:2 = hlfir.declare {{.*}}Ex37! CHECK: hlfir.region_assign {38! CHECK: %[[VAL_11:.*]] = arith.constant 4.200000e+01 : f3239! CHECK: hlfir.yield %[[VAL_11]] : f3240! CHECK: } to {41! CHECK: %[[VAL_12:.*]] = hlfir.matmul %[[VAL_9]]#0 %[[VAL_6]]#0 {fastmath = #arith.fastmath<contract>} : (!fir.ref<!fir.array<10xi64>>, !fir.ref<!fir.array<10x5xi64>>) -> !hlfir.expr<5xi64>42! CHECK: %[[VAL_13:.*]] = arith.constant 5 : index43! CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1>44! CHECK: hlfir.elemental_addr %[[VAL_14]] unordered : !fir.shape<1> {45! CHECK: ^bb0(%[[VAL_15:.*]]: index):46! CHECK: %[[VAL_16:.*]] = hlfir.apply %[[VAL_12]], %[[VAL_15]] : (!hlfir.expr<5xi64>, index) -> i6447! CHECK: %[[VAL_17:.*]] = hlfir.designate %[[VAL_10]]#0 (%[[VAL_16]]) : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>48! CHECK: hlfir.yield %[[VAL_17]] : !fir.ref<f32>49! CHECK: } cleanup {50! CHECK: hlfir.destroy %[[VAL_12]] : !hlfir.expr<5xi64>51! CHECK: }52! CHECK: }53 54subroutine test_nested_vectors(x, vector1, vector2, vector3)55 integer(8) :: vector1(10), vector2(8), vector3(6)56 real :: x(:)57 x(vector1(vector2(vector3))) = 42.58end subroutine59! CHECK-LABEL: func.func @_QPtest_nested_vectors(60! CHECK: %[[VAL_6:.*]]:2 = hlfir.declare {{.*}}Evector161! CHECK: %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Evector262! CHECK: %[[VAL_12:.*]]:2 = hlfir.declare {{.*}}Evector363! CHECK: %[[VAL_13:.*]]:2 = hlfir.declare {{.*}}Ex64! CHECK: hlfir.region_assign {65! CHECK: %[[VAL_14:.*]] = arith.constant 4.200000e+01 : f3266! CHECK: hlfir.yield %[[VAL_14]] : f3267! CHECK: } to {68! CHECK: %[[VAL_15:.*]] = arith.constant 6 : index69! CHECK: %[[VAL_16:.*]] = fir.shape %[[VAL_15]] : (index) -> !fir.shape<1>70! CHECK: %[[VAL_17:.*]] = hlfir.elemental %[[VAL_16]] unordered : (!fir.shape<1>) -> !hlfir.expr<6xi64> {71! CHECK: ^bb0(%[[VAL_18:.*]]: index):72! CHECK: %[[VAL_19:.*]] = hlfir.designate %[[VAL_12]]#0 (%[[VAL_18]]) : (!fir.ref<!fir.array<6xi64>>, index) -> !fir.ref<i64>73! CHECK: %[[VAL_20:.*]] = fir.load %[[VAL_19]] : !fir.ref<i64>74! CHECK: %[[VAL_21:.*]] = hlfir.designate %[[VAL_9]]#0 (%[[VAL_20]]) : (!fir.ref<!fir.array<8xi64>>, i64) -> !fir.ref<i64>75! CHECK: %[[VAL_22:.*]] = fir.load %[[VAL_21]] : !fir.ref<i64>76! CHECK: hlfir.yield_element %[[VAL_22]] : i6477! CHECK: }78! CHECK: %[[VAL_23:.*]] = arith.constant 6 : index79! CHECK: %[[VAL_24:.*]] = fir.shape %[[VAL_23]] : (index) -> !fir.shape<1>80! CHECK: %[[VAL_25:.*]] = hlfir.elemental %[[VAL_24]] unordered : (!fir.shape<1>) -> !hlfir.expr<6xi64> {81! CHECK: ^bb0(%[[VAL_26:.*]]: index):82! CHECK: %[[VAL_27:.*]] = hlfir.apply %[[VAL_28:.*]], %[[VAL_26]] : (!hlfir.expr<6xi64>, index) -> i6483! CHECK: %[[VAL_29:.*]] = hlfir.designate %[[VAL_6]]#0 (%[[VAL_27]]) : (!fir.ref<!fir.array<10xi64>>, i64) -> !fir.ref<i64>84! CHECK: %[[VAL_30:.*]] = fir.load %[[VAL_29]] : !fir.ref<i64>85! CHECK: hlfir.yield_element %[[VAL_30]] : i6486! CHECK: }87! CHECK: %[[VAL_31:.*]] = arith.constant 6 : index88! CHECK: %[[VAL_32:.*]] = fir.shape %[[VAL_31]] : (index) -> !fir.shape<1>89! CHECK: hlfir.elemental_addr %[[VAL_32]] unordered : !fir.shape<1> {90! CHECK: ^bb0(%[[VAL_33:.*]]: index):91! CHECK: %[[VAL_34:.*]] = hlfir.apply %[[VAL_35:.*]], %[[VAL_33]] : (!hlfir.expr<6xi64>, index) -> i6492! CHECK: %[[VAL_36:.*]] = hlfir.designate %[[VAL_13]]#0 (%[[VAL_34]]) : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>93! CHECK: hlfir.yield %[[VAL_36]] : !fir.ref<f32>94! CHECK: } cleanup {95! CHECK: hlfir.destroy %[[VAL_37:.*]] : !hlfir.expr<6xi64>96! CHECK: hlfir.destroy %[[VAL_38:.*]] : !hlfir.expr<6xi64>97! CHECK: }98! CHECK: }99 100subroutine test_substring(x, vector)101 integer(8) :: vector(10), ifoo, ibar102 external :: ifoo, ibar103 character(*) :: x(:)104 x(vector)(ifoo(): ibar()) = "hello"105end subroutine106! CHECK-LABEL: func.func @_QPtest_substring(107! CHECK: %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}Evector108! CHECK: %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Ex109! CHECK: hlfir.region_assign {110! CHECK: %[[VAL_6:.*]] = fir.address_of(@{{.*}}) : !fir.ref<!fir.char<1,5>>111! CHECK: %[[VAL_7:.*]] = arith.constant 5 : index112! CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_6]] typeparams %[[VAL_7]] {fortran_attrs = #fir.var_attrs<parameter>, uniq_name = "_QQclX68656C6C6F"} : (!fir.ref<!fir.char<1,5>>, index) -> (!fir.ref<!fir.char<1,5>>, !fir.ref<!fir.char<1,5>>)113! CHECK: hlfir.yield %[[VAL_8]]#0 : !fir.ref<!fir.char<1,5>>114! CHECK: } to {115! CHECK: %[[VAL_9:.*]] = arith.constant 10 : index116! CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_9]] : (index) -> !fir.shape<1>117! CHECK: %[[VAL_11:.*]] = fir.call @_QPifoo() {{.*}}: () -> i64118! CHECK: %[[VAL_12:.*]] = fir.call @_QPibar() {{.*}}: () -> i64119! CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_11]] : (i64) -> index120! CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_12]] : (i64) -> index121! CHECK: %[[VAL_15:.*]] = arith.constant 1 : index122! CHECK: %[[VAL_16:.*]] = arith.subi %[[VAL_14]], %[[VAL_13]] : index123! CHECK: %[[VAL_17:.*]] = arith.addi %[[VAL_16]], %[[VAL_15]] : index124! CHECK: %[[VAL_18:.*]] = arith.constant 0 : index125! CHECK: %[[VAL_19:.*]] = arith.cmpi sgt, %[[VAL_17]], %[[VAL_18]] : index126! CHECK: %[[VAL_20:.*]] = arith.select %[[VAL_19]], %[[VAL_17]], %[[VAL_18]] : index127! CHECK: hlfir.elemental_addr %[[VAL_10]] typeparams %[[VAL_20]] unordered : !fir.shape<1>, index {128! CHECK: ^bb0(%[[VAL_21:.*]]: index):129! CHECK: %[[VAL_22:.*]] = hlfir.designate %[[VAL_4]]#0 (%[[VAL_21]]) : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>130! CHECK: %[[VAL_23:.*]] = fir.load %[[VAL_22]] : !fir.ref<i64>131! CHECK: %[[VAL_24:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_23]]) substr %[[VAL_13]], %[[VAL_14]] typeparams %[[VAL_20]] : (!fir.box<!fir.array<?x!fir.char<1,?>>>, i64, index, index, index) -> !fir.boxchar<1>132! CHECK: hlfir.yield %[[VAL_24]] : !fir.boxchar<1>133! CHECK: }134! CHECK: }135 136subroutine test_hard_array_ref(x, vector1, vector2)137 integer(8) :: vector1(10), vector2(20), ifoo, ibar, ibaz138 external :: ifoo, ibar, ibaz139 real :: x(:, :, :, :, :)140 x(vector1, :, ifoo():ibar(), ibaz(), vector2) = 42.141end subroutine142! CHECK-LABEL: func.func @_QPtest_hard_array_ref(143! CHECK: %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Evector1144! CHECK: %[[VAL_8:.*]]:2 = hlfir.declare {{.*}}Evector2145! CHECK: %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Ex146! CHECK: hlfir.region_assign {147! CHECK: %[[VAL_10:.*]] = arith.constant 4.200000e+01 : f32148! CHECK: hlfir.yield %[[VAL_10]] : f32149! CHECK: } to {150! CHECK: %[[VAL_11:.*]] = arith.constant 10 : index151! CHECK: %[[VAL_12:.*]] = arith.constant 1 : index152! CHECK: %[[VAL_13:.*]] = arith.constant 1 : index153! CHECK: %[[VAL_14:.*]]:3 = fir.box_dims %[[VAL_9]]#1, %[[VAL_13]] : (!fir.box<!fir.array<?x?x?x?x?xf32>>, index) -> (index, index, index)154! CHECK: %[[VAL_15:.*]] = arith.constant 1 : index155! CHECK: %[[VAL_16:.*]] = arith.constant 0 : index156! CHECK: %[[VAL_17:.*]] = arith.subi %[[VAL_14]]#1, %[[VAL_12]] : index157! CHECK: %[[VAL_18:.*]] = arith.addi %[[VAL_17]], %[[VAL_15]] : index158! CHECK: %[[VAL_19:.*]] = arith.divsi %[[VAL_18]], %[[VAL_15]] : index159! CHECK: %[[VAL_20:.*]] = arith.cmpi sgt, %[[VAL_19]], %[[VAL_16]] : index160! CHECK: %[[VAL_21:.*]] = arith.select %[[VAL_20]], %[[VAL_19]], %[[VAL_16]] : index161! CHECK: %[[VAL_22:.*]] = fir.call @_QPifoo() {{.*}}: () -> i64162! CHECK: %[[VAL_23:.*]] = fir.call @_QPibar() {{.*}}: () -> i64163! CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_22]] : (i64) -> index164! CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_23]] : (i64) -> index165! CHECK: %[[VAL_26:.*]] = arith.constant 1 : index166! CHECK: %[[VAL_27:.*]] = arith.constant 0 : index167! CHECK: %[[VAL_28:.*]] = arith.subi %[[VAL_25]], %[[VAL_24]] : index168! CHECK: %[[VAL_29:.*]] = arith.addi %[[VAL_28]], %[[VAL_26]] : index169! CHECK: %[[VAL_30:.*]] = arith.divsi %[[VAL_29]], %[[VAL_26]] : index170! CHECK: %[[VAL_31:.*]] = arith.cmpi sgt, %[[VAL_30]], %[[VAL_27]] : index171! CHECK: %[[VAL_32:.*]] = arith.select %[[VAL_31]], %[[VAL_30]], %[[VAL_27]] : index172! CHECK: %[[VAL_33:.*]] = fir.call @_QPibaz() {{.*}}: () -> i64173! CHECK: %[[VAL_34:.*]] = arith.constant 20 : index174! CHECK: %[[VAL_35:.*]] = fir.shape %[[VAL_11]], %[[VAL_21]], %[[VAL_32]], %[[VAL_34]] : (index, index, index, index) -> !fir.shape<4>175! CHECK: hlfir.elemental_addr %[[VAL_35]] unordered : !fir.shape<4> {176! CHECK: ^bb0(%[[VAL_36:.*]]: index, %[[VAL_37:.*]]: index, %[[VAL_38:.*]]: index, %[[VAL_39:.*]]: index):177! CHECK: %[[VAL_40:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_36]]) : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>178! CHECK: %[[VAL_41:.*]] = fir.load %[[VAL_40]] : !fir.ref<i64>179! CHECK: %[[VAL_42:.*]] = arith.constant 1 : index180! CHECK: %[[VAL_43:.*]] = arith.subi %[[VAL_37]], %[[VAL_42]] : index181! CHECK: %[[VAL_44:.*]] = arith.muli %[[VAL_43]], %[[VAL_15]] : index182! CHECK: %[[VAL_45:.*]] = arith.addi %[[VAL_12]], %[[VAL_44]] : index183! CHECK: %[[VAL_46:.*]] = arith.constant 1 : index184! CHECK: %[[VAL_47:.*]] = arith.subi %[[VAL_38]], %[[VAL_46]] : index185! CHECK: %[[VAL_48:.*]] = arith.muli %[[VAL_47]], %[[VAL_26]] : index186! CHECK: %[[VAL_49:.*]] = arith.addi %[[VAL_24]], %[[VAL_48]] : index187! CHECK: %[[VAL_50:.*]] = hlfir.designate %[[VAL_8]]#0 (%[[VAL_39]]) : (!fir.ref<!fir.array<20xi64>>, index) -> !fir.ref<i64>188! CHECK: %[[VAL_51:.*]] = fir.load %[[VAL_50]] : !fir.ref<i64>189! CHECK: %[[VAL_52:.*]] = hlfir.designate %[[VAL_9]]#0 (%[[VAL_41]], %[[VAL_45]], %[[VAL_49]], %[[VAL_33]], %[[VAL_51]]) : (!fir.box<!fir.array<?x?x?x?x?xf32>>, i64, index, index, i64, i64) -> !fir.ref<f32>190! CHECK: hlfir.yield %[[VAL_52]] : !fir.ref<f32>191! CHECK: }192! CHECK: }193! CHECK: return194