456 lines · plain
1// Test hlfir.associate/hlfir.end_associate operation code generation to FIR.2 3// RUN: fir-opt %s -bufferize-hlfir | FileCheck %s4 5func.func @associate_int() {6 %c42_i32 = arith.constant 42 : i327 %0:3 = hlfir.associate %c42_i32 {uniq_name = "x"} : (i32) -> (!fir.ref<i32>, !fir.ref<i32>, i1)8 fir.call @take_i4(%0#0) : (!fir.ref<i32>) -> ()9 hlfir.end_associate %0#1, %0#2 : !fir.ref<i32>, i110 return11}12// CHECK-LABEL: func.func @associate_int() {13// CHECK: %[[VAL_0:.*]] = fir.alloca i32 {bindc_name = "x"}14// CHECK: %[[VAL_1:.*]] = arith.constant 42 : i3215// CHECK: fir.store %[[VAL_1]] to %[[VAL_0]] : !fir.ref<i32>16// CHECK: %[[VAL_2:.*]] = arith.constant false17// CHECK: fir.call @take_i4(%[[VAL_0]]) : (!fir.ref<i32>) -> ()18// CHECK-NOT: fir.freemem19 20 21func.func @associate_real() {22 %cst = arith.constant 4.200000e-01 : f3223 %0:3 = hlfir.associate %cst {uniq_name = "x"} : (f32) -> (!fir.ref<f32>, !fir.ref<f32>, i1)24 fir.call @take_r4(%0#0) : (!fir.ref<f32>) -> ()25 hlfir.end_associate %0#1, %0#2 : !fir.ref<f32>, i126 return27}28// CHECK-LABEL: func.func @associate_real() {29// CHECK: %[[VAL_0:.*]] = fir.alloca f32 {bindc_name = "x"}30// CHECK: %[[VAL_1:.*]] = arith.constant 4.200000e-01 : f3231// CHECK: fir.store %[[VAL_1]] to %[[VAL_0]] : !fir.ref<f32>32// CHECK: %[[VAL_2:.*]] = arith.constant false33// CHECK: fir.call @take_r4(%[[VAL_0]]) : (!fir.ref<f32>) -> ()34// CHECK-NOT: fir.freemem35 36 37func.func @associate_logical() {38 %true = arith.constant true39 %0 = fir.convert %true : (i1) -> !fir.logical<4>40 %1:3 = hlfir.associate %0 {uniq_name = "x"} : (!fir.logical<4>) -> (!fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>, i1)41 fir.call @take_l4(%1#0) : (!fir.ref<!fir.logical<4>>) -> ()42 hlfir.end_associate %1#1, %1#2 : !fir.ref<!fir.logical<4>>, i143 return44}45// CHECK-LABEL: func.func @associate_logical() {46// CHECK: %[[VAL_0:.*]] = fir.alloca !fir.logical<4> {bindc_name = "x"}47// CHECK: %[[VAL_1:.*]] = arith.constant true48// CHECK: %[[VAL_2:.*]] = fir.convert %[[VAL_1]] : (i1) -> !fir.logical<4>49// CHECK: fir.store %[[VAL_2]] to %[[VAL_0]] : !fir.ref<!fir.logical<4>>50// CHECK: %[[VAL_3:.*]] = arith.constant false51// CHECK: fir.call @take_l4(%[[VAL_0]]) : (!fir.ref<!fir.logical<4>>) -> ()52// CHECK-NOT: fir.freemem53 54 55func.func @associate_char(%arg0: !fir.boxchar<1> ) {56 %c1 = arith.constant 1 : index57 %0:2 = fir.unboxchar %arg0 : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)58 %1:2 = hlfir.declare %0#0 typeparams %0#1 {uniq_name = "x"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)59 %2 = arith.addi %0#1, %c1 : index60 %3 = fir.undefined !fir.ref<!fir.char<1>>61 %4:2 = hlfir.declare %3 typeparams %c1 {fortran_attrs = #fir.var_attrs<parameter>, uniq_name = "char_literal"} : (!fir.ref<!fir.char<1>>, index) -> (!fir.ref<!fir.char<1>>, !fir.ref<!fir.char<1>>)62 %5 = hlfir.concat %1#0, %4#0 len %2 : (!fir.boxchar<1>, !fir.ref<!fir.char<1>>, index) -> !hlfir.expr<!fir.char<1,?>>63 %6 = hlfir.no_reassoc %5 : !hlfir.expr<!fir.char<1,?>>64 %7:3 = hlfir.associate %6 typeparams %2 {uniq_name = "x"} : (!hlfir.expr<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>, i1)65 fir.call @take_c(%7#0) : (!fir.boxchar<1>) -> ()66 hlfir.end_associate %7#1, %7#2 : !fir.ref<!fir.char<1,?>>, i167 return68}69 70// CHECK-LABEL: func.func @associate_char(71// CHECK-SAME: %[[VAL_0:.*]]: !fir.boxchar<1>) {72// CHECK: %[[VAL_1:.*]] = arith.constant 1 : index73// CHECK: %[[VAL_2:.*]]:2 = fir.unboxchar %[[VAL_0]] : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)74// CHECK: %[[VAL_3:.*]]:2 = hlfir.declare %[[VAL_2]]#0 typeparams %[[VAL_2]]#1 {uniq_name = "x"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)75// CHECK: %[[VAL_4:.*]] = arith.addi %[[VAL_2]]#1, %[[VAL_1]] : index76// CHECK: %[[VAL_5:.*]] = fir.undefined !fir.ref<!fir.char<1>>77// CHECK: %[[VAL_6:.*]]:2 = hlfir.declare %[[VAL_5]] typeparams %[[VAL_1]] {fortran_attrs = #fir.var_attrs<parameter>, uniq_name = "char_literal"} : (!fir.ref<!fir.char<1>>, index) -> (!fir.ref<!fir.char<1>>, !fir.ref<!fir.char<1>>)78// CHECK: %[[VAL_7:.*]] = arith.addi %[[VAL_2]]#1, %[[VAL_1]] : index79// CHECK: %[[VAL_8:.*]] = fir.alloca !fir.char<1,?>(%[[VAL_7]] : index) {bindc_name = ".chrtmp"}80// CHECK: %[[VAL_9:.*]] = arith.constant 1 : i6481// CHECK: %[[VAL_10:.*]] = fir.convert %[[VAL_2]]#1 : (index) -> i6482// CHECK: %[[VAL_11:.*]] = arith.muli %[[VAL_9]], %[[VAL_10]] : i6483// CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_8]] : (!fir.ref<!fir.char<1,?>>) -> !llvm.ptr84// CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_3]]#1 : (!fir.ref<!fir.char<1,?>>) -> !llvm.ptr85// CHECK: "llvm.intr.memmove"(%[[VAL_13]], %[[VAL_14]], %[[VAL_11]]) <{isVolatile = false}> : (!llvm.ptr, !llvm.ptr, i64) -> ()86// CHECK: %[[VAL_15:.*]] = arith.constant 1 : index87// CHECK: %[[VAL_16:.*]] = arith.subi %[[VAL_7]], %[[VAL_15]] : index88// CHECK: fir.do_loop %[[VAL_17:.*]] = %[[VAL_2]]#1 to %[[VAL_16]] step %[[VAL_15]] {89// CHECK: %[[VAL_18:.*]] = arith.subi %[[VAL_17]], %[[VAL_2]]#1 : index90// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_6]]#0 : (!fir.ref<!fir.char<1>>) -> !fir.ref<!fir.array<1x!fir.char<1>>>91// CHECK: %[[VAL_20:.*]] = fir.coordinate_of %[[VAL_19]], %[[VAL_18]] : (!fir.ref<!fir.array<1x!fir.char<1>>>, index) -> !fir.ref<!fir.char<1>>92// CHECK: %[[VAL_21:.*]] = fir.load %[[VAL_20]] : !fir.ref<!fir.char<1>>93// CHECK: %[[VAL_22:.*]] = fir.convert %[[VAL_8]] : (!fir.ref<!fir.char<1,?>>) -> !fir.ref<!fir.array<?x!fir.char<1>>>94// CHECK: %[[VAL_23:.*]] = fir.coordinate_of %[[VAL_22]], %[[VAL_17]] : (!fir.ref<!fir.array<?x!fir.char<1>>>, index) -> !fir.ref<!fir.char<1>>95// CHECK: fir.store %[[VAL_21]] to %[[VAL_23]] : !fir.ref<!fir.char<1>>96// CHECK: }97// CHECK: %[[VAL_24:.*]]:2 = hlfir.declare %[[VAL_8]] typeparams %[[VAL_7]] {uniq_name = "tmp"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)98// CHECK: %[[VAL_25:.*]] = arith.constant false99// CHECK: %[[VAL_26:.*]] = fir.undefined tuple<!fir.boxchar<1>, i1>100// CHECK: %[[VAL_27:.*]] = fir.insert_value %[[VAL_26]], %[[VAL_25]], [1 : index] : (tuple<!fir.boxchar<1>, i1>, i1) -> tuple<!fir.boxchar<1>, i1>101// CHECK: %[[VAL_28:.*]] = fir.insert_value %[[VAL_27]], %[[VAL_24]]#0, [0 : index] : (tuple<!fir.boxchar<1>, i1>, !fir.boxchar<1>) -> tuple<!fir.boxchar<1>, i1>102// CHECK: %[[VAL_29:.*]] = hlfir.no_reassoc %[[VAL_24]]#0 : !fir.boxchar<1>103// CHECK: %[[VAL_30:.*]] = fir.undefined tuple<!fir.boxchar<1>, i1>104// CHECK: %[[VAL_31:.*]] = fir.insert_value %[[VAL_30]], %[[VAL_25]], [1 : index] : (tuple<!fir.boxchar<1>, i1>, i1) -> tuple<!fir.boxchar<1>, i1>105// CHECK: %[[VAL_32:.*]] = fir.insert_value %[[VAL_31]], %[[VAL_29]], [0 : index] : (tuple<!fir.boxchar<1>, i1>, !fir.boxchar<1>) -> tuple<!fir.boxchar<1>, i1>106// CHECK: %[[VAL_33:.*]] = fir.box_addr %[[VAL_29]] : (!fir.boxchar<1>) -> !fir.ref<!fir.char<1,?>>107// CHECK: fir.call @take_c(%[[VAL_29]]) : (!fir.boxchar<1>) -> ()108// CHECK: return109// CHECK: }110 111func.func @test_end_associate_box(%var: !fir.box<!fir.array<?xf64>>) {112 %true = arith.constant 1 : i1113 hlfir.end_associate %var, %true : !fir.box<!fir.array<?xf64>>, i1114 return115}116// CHECK-LABEL: func.func @test_end_associate_box(117// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?xf64>>) {118// CHECK: %[[VAL_1:.*]] = arith.constant true119// CHECK: %[[VAL_2:.*]] = fir.box_addr %[[VAL_0]] : (!fir.box<!fir.array<?xf64>>) -> !fir.heap<!fir.array<?xf64>>120// CHECK: fir.freemem %[[VAL_2]] : !fir.heap<!fir.array<?xf64>>121 122 123func.func @test_end_associate_boxchar(%var: !fir.boxchar<2>) {124 %true = arith.constant 1 : i1125 hlfir.end_associate %var, %true : !fir.boxchar<2>, i1126 return127}128// CHECK-LABEL: func.func @test_end_associate_boxchar(129// CHECK-SAME: %[[VAL_0:.*]]: !fir.boxchar<2>) {130// CHECK: %[[VAL_1:.*]] = arith.constant true131// CHECK: %[[VAL_2:.*]] = fir.box_addr %[[VAL_0]] : (!fir.boxchar<2>) -> !fir.heap<!fir.char<2,?>>132// CHECK: fir.freemem %[[VAL_2]] : !fir.heap<!fir.char<2,?>>133 134 135func.func @test_end_associate_box_dynamic(%var: !fir.box<!fir.array<?xf64>>, %must_free: i1) {136 hlfir.end_associate %var, %must_free : !fir.box<!fir.array<?xf64>>, i1137 return138}139// CHECK-LABEL: func.func @test_end_associate_box_dynamic(140// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?xf64>>,141// CHECK-SAME: %[[VAL_1:.*]]: i1) {142// CHECK: fir.if %[[VAL_1]] {143// CHECK: %[[VAL_2:.*]] = fir.box_addr %[[VAL_0]] : (!fir.box<!fir.array<?xf64>>) -> !fir.heap<!fir.array<?xf64>>144// CHECK: fir.freemem %[[VAL_2]] : !fir.heap<!fir.array<?xf64>>145// CHECK: }146 147func.func private @bar(!fir.ref<!fir.array<?xi32>>) -> ()148func.func @test_result_box_addr(%x : !fir.box<!fir.array<?xi32>>) {149 %true = arith.constant 1 : i1150 %expr = hlfir.as_expr %x move %true : (!fir.box<!fir.array<?xi32>>, i1) -> !hlfir.expr<?xi32>151 %y:3 = hlfir.associate %expr {uniq_name = "y"}: (!hlfir.expr<?xi32>) -> (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?xi32>>, i1)152 fir.call @bar(%y#1) : (!fir.ref<!fir.array<?xi32>>) -> ()153 return154}155// CHECK-LABEL: func.func @test_result_box_addr(156// CHECK-SAME: %[[X:.*]]: !fir.box<!fir.array<?xi32>>) {157// CHECK: %[[ADDR:.*]] = fir.box_addr %[[X]] : (!fir.box<!fir.array<?xi32>>) -> !fir.ref<!fir.array<?xi32>>158// CHECK: fir.call @bar(%[[ADDR]]) : (!fir.ref<!fir.array<?xi32>>) -> ()159 160func.func private @bar2(!fir.ref<!fir.array<10xi32>>) -> ()161func.func @test_result_convert(%x : !fir.heap<!fir.array<10xi32>>) {162 %true = arith.constant 1 : i1163 %expr = hlfir.as_expr %x move %true : (!fir.heap<!fir.array<10xi32>>, i1) -> !hlfir.expr<10xi32>164 %y:3 = hlfir.associate %expr {uniq_name = "y"}: (!hlfir.expr<10xi32>) -> (!fir.ref<!fir.array<10xi32>>, !fir.ref<!fir.array<10xi32>>, i1)165 fir.call @bar2(%y#1) : (!fir.ref<!fir.array<10xi32>>) -> ()166 return167}168// CHECK-LABEL: func.func @test_result_convert(169// CHECK-SAME: %[[X:.*]]: !fir.heap<!fir.array<10xi32>>) {170// CHECK: fir.convert171// CHECK: %[[ADDR:.*]] = fir.convert %[[X]] : (!fir.heap<!fir.array<10xi32>>) -> !fir.ref<!fir.array<10xi32>>172// CHECK: fir.call @bar2(%[[ADDR]]) : (!fir.ref<!fir.array<10xi32>>) -> ()173 174 175func.func @test_0dim_box(%x : !fir.ref<!fir.box<!fir.heap<i32>>>) {176 %0 = fir.load %x : !fir.ref<!fir.box<!fir.heap<i32>>>177 %1:2 = hlfir.declare %0 {uniq_name = ".tmp.intrinsic_result"} : (!fir.box<!fir.heap<i32>>) -> (!fir.box<i32>, !fir.box<i32>)178 %true = arith.constant true179 %2 = hlfir.as_expr %1#0 move %true : (!fir.box<i32>, i1) -> !hlfir.expr<i32>180 %3:3 = hlfir.associate %2 {adapt.valuebyref} : (!hlfir.expr<i32>) -> (!fir.ref<i32>, !fir.ref<i32>, i1)181 return182}183// CHECK-LABEL: func.func @test_0dim_box(184// CHECK-SAME: %[[VAL_0:.*]]: !fir.ref<!fir.box<!fir.heap<i32>>>) {185// CHECK: %[[VAL_1:.*]] = fir.load %[[VAL_0]] : !fir.ref<!fir.box<!fir.heap<i32>>>186// CHECK: %[[VAL_2:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = ".tmp.intrinsic_result"} : (!fir.box<!fir.heap<i32>>) -> (!fir.box<i32>, !fir.box<i32>)187// CHECK: %[[VAL_3:.*]] = arith.constant true188// CHECK: %[[VAL_4:.*]] = fir.undefined tuple<!fir.box<i32>, i1>189// CHECK: %[[VAL_5:.*]] = fir.insert_value %[[VAL_4]], %[[VAL_3]], [1 : index] : (tuple<!fir.box<i32>, i1>, i1) -> tuple<!fir.box<i32>, i1>190// CHECK: %[[VAL_6:.*]] = fir.insert_value %[[VAL_5]], %[[VAL_2]]#0, [0 : index] : (tuple<!fir.box<i32>, i1>, !fir.box<i32>) -> tuple<!fir.box<i32>, i1>191// CHECK: %[[VAL_7:.*]] = fir.box_addr %[[VAL_2]]#0 : (!fir.box<i32>) -> !fir.ref<i32>192// CHECK: %[[VAL_8:.*]] = fir.box_addr %[[VAL_2]]#1 : (!fir.box<i32>) -> !fir.ref<i32>193// CHECK: return194// CHECK: }195 196// test that we support a hlfir.associate operation where the expr is also used in a hlfir.shape_of op197func.func @test_shape_of(%arg0: !fir.ref<!fir.array<4x3xi32>>) {198 %c4 = arith.constant 4 : index199 %c3 = arith.constant 3 : index200 %shape = fir.shape %c3, %c4 : (index, index) -> !fir.shape<2>201 // %0 = hlfir.transpose %arg0 : (!fir.ref<!fir.array<4x3xi32>>) -> !hlfir.expr<3x4xi32>202 %0 = hlfir.elemental %shape unordered : (!fir.shape<2>) -> !hlfir.expr<3x4xi32> {203 ^bb0(%arg1: index, %arg2: index):204 %4 = hlfir.designate %arg0 (%arg2, %arg1) : (!fir.ref<!fir.array<4x3xi32>>, index, index) -> !fir.ref<i32>205 %5 = fir.load %4 : !fir.ref<i32>206 hlfir.yield_element %5 : i32207 }208 %1 = hlfir.shape_of %0 : (!hlfir.expr<3x4xi32>) -> !fir.shape<2>209 %2:3 = hlfir.associate %0(%1) {adapt.valuebyref} : (!hlfir.expr<3x4xi32>, !fir.shape<2>) -> (!fir.ref<!fir.array<3x4xi32>>, !fir.ref<!fir.array<3x4xi32>>, i1)210 // ...211 hlfir.end_associate %2#1, %2#2 : !fir.ref<!fir.array<3x4xi32>>, i1212 return213}214// CHECK-LABEL: func.func @test_shape_of(215// CHECK-SAME: %[[VAL_0:.*]]: !fir.ref<!fir.array<4x3xi32>>) {216// CHECK: %[[VAL_1:.*]] = arith.constant 4 : index217// CHECK: %[[VAL_2:.*]] = arith.constant 3 : index218// CHECK: %[[VAL_3:.*]] = fir.shape %[[VAL_2]], %[[VAL_1]] : (index, index) -> !fir.shape<2>219// CHECK: %[[VAL_4:.*]] = fir.allocmem !fir.array<3x4xi32> {bindc_name = ".tmp.array", uniq_name = ""}220// CHECK: %[[VAL_5:.*]]:2 = hlfir.declare %[[VAL_4]](%[[VAL_3]]) {uniq_name = ".tmp.array"} : (!fir.heap<!fir.array<3x4xi32>>, !fir.shape<2>) -> (!fir.heap<!fir.array<3x4xi32>>, !fir.heap<!fir.array<3x4xi32>>)221// CHECK: %[[VAL_6:.*]] = arith.constant true222// CHECK: %[[VAL_7:.*]] = arith.constant 1 : index223// CHECK: fir.do_loop %[[VAL_8:.*]] = %[[VAL_7]] to %[[VAL_1]] step %[[VAL_7]] unordered {224// CHECK: fir.do_loop %[[VAL_9:.*]] = %[[VAL_7]] to %[[VAL_2]] step %[[VAL_7]] unordered {225// CHECK: %[[VAL_10:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_8]], %[[VAL_9]]) : (!fir.ref<!fir.array<4x3xi32>>, index, index) -> !fir.ref<i32>226// CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_10]] : !fir.ref<i32>227// CHECK: %[[VAL_12:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_9]], %[[VAL_8]]) : (!fir.heap<!fir.array<3x4xi32>>, index, index) -> !fir.ref<i32>228// CHECK: hlfir.assign %[[VAL_11]] to %[[VAL_12]] temporary_lhs : i32, !fir.ref<i32>229// CHECK: }230// CHECK: }231// CHECK: %[[VAL_13:.*]] = fir.undefined tuple<!fir.heap<!fir.array<3x4xi32>>, i1>232// CHECK: %[[VAL_14:.*]] = fir.insert_value %[[VAL_13]], %[[VAL_6]], [1 : index] : (tuple<!fir.heap<!fir.array<3x4xi32>>, i1>, i1) -> tuple<!fir.heap<!fir.array<3x4xi32>>, i1>233// CHECK: %[[VAL_15:.*]] = fir.insert_value %[[VAL_14]], %[[VAL_5]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<3x4xi32>>, i1>, !fir.heap<!fir.array<3x4xi32>>) -> tuple<!fir.heap<!fir.array<3x4xi32>>, i1>234// CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_5]]#0 : (!fir.heap<!fir.array<3x4xi32>>) -> !fir.ref<!fir.array<3x4xi32>>235// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_5]]#0 : (!fir.heap<!fir.array<3x4xi32>>) -> !fir.ref<!fir.array<3x4xi32>>236// CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_17]] : (!fir.ref<!fir.array<3x4xi32>>) -> !fir.heap<!fir.array<3x4xi32>>237// CHECK: fir.freemem %[[VAL_18]] : !fir.heap<!fir.array<3x4xi32>>238// CHECK: return239// CHECK: }240 241func.func @test_multiple_associations(%arg0: !hlfir.expr<1x2xi32>) {242 %c1 = arith.constant 1 : index243 %c2 = arith.constant 2 : index244 %shape = fir.shape %c1, %c2 : (index, index) -> !fir.shape<2>245 %0:3 = hlfir.associate %arg0(%shape) {uniq_name = "associate 0"} : (!hlfir.expr<1x2xi32>, !fir.shape<2>) -> (!fir.ref<!fir.array<1x2xi32>>, !fir.ref<!fir.array<1x2xi32>>, i1)246 %1:3 = hlfir.associate %arg0(%shape) {uniq_name = "associate 1"} : (!hlfir.expr<1x2xi32>, !fir.shape<2>) -> (!fir.ref<!fir.array<1x2xi32>>, !fir.ref<!fir.array<1x2xi32>>, i1)247 hlfir.end_associate %0#1, %0#2 : !fir.ref<!fir.array<1x2xi32>>, i1248 hlfir.end_associate %1#1, %1#2 : !fir.ref<!fir.array<1x2xi32>>, i1249 return250}251// CHECK-LABEL: func.func @test_multiple_associations(252// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<1x2xi32>) {253// CHECK: %[[VAL_1:.*]] = arith.constant 1 : index254// CHECK: %[[VAL_2:.*]] = arith.constant 2 : index255// CHECK: %[[VAL_3:.*]] = fir.shape %[[VAL_1]], %[[VAL_2]] : (index, index) -> !fir.shape<2>256// associate 0:257// CHECK: %[[VAL_4:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<1x2xi32>) -> !fir.shape<2>258// CHECK: %[[VAL_5:.*]] = arith.constant 1 : index259// CHECK: %[[VAL_6:.*]] = arith.constant 2 : index260// CHECK: %[[VAL_7:.*]] = fir.allocmem !fir.array<1x2xi32> {bindc_name = ".tmp", uniq_name = ""}261// CHECK: %[[VAL_9:.*]]:2 = hlfir.declare %[[VAL_7]](%[[VAL_4]]) {uniq_name = ".tmp"} : (!fir.heap<!fir.array<1x2xi32>>, !fir.shape<2>) -> (!fir.heap<!fir.array<1x2xi32>>, !fir.heap<!fir.array<1x2xi32>>)262// CHECK: hlfir.assign %[[VAL_0]] to %[[VAL_9]]#0 temporary_lhs : !hlfir.expr<1x2xi32>, !fir.heap<!fir.array<1x2xi32>>263// CHECK: %[[VAL_8:.*]] = arith.constant true264// CHECK: %[[VAL_10:.*]] = fir.undefined tuple<!fir.heap<!fir.array<1x2xi32>>, i1>265// CHECK: %[[VAL_11:.*]] = fir.insert_value %[[VAL_10]], %[[VAL_8]], [1 : index] : (tuple<!fir.heap<!fir.array<1x2xi32>>, i1>, i1) -> tuple<!fir.heap<!fir.array<1x2xi32>>, i1>266// CHECK: %[[VAL_12:.*]] = fir.insert_value %[[VAL_11]], %[[VAL_9]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<1x2xi32>>, i1>, !fir.heap<!fir.array<1x2xi32>>) -> tuple<!fir.heap<!fir.array<1x2xi32>>, i1>267// CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_9]]#0 : (!fir.heap<!fir.array<1x2xi32>>) -> !fir.ref<!fir.array<1x2xi32>>268// CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_9]]#0 : (!fir.heap<!fir.array<1x2xi32>>) -> !fir.ref<!fir.array<1x2xi32>>269// associate 1:270// CHECK: %[[VAL_15:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<1x2xi32>) -> !fir.shape<2>271// CHECK: %[[VAL_16:.*]] = arith.constant 1 : index272// CHECK: %[[VAL_17:.*]] = arith.constant 2 : index273// CHECK: %[[VAL_18:.*]] = fir.allocmem !fir.array<1x2xi32> {bindc_name = ".tmp", uniq_name = ""}274// CHECK: %[[VAL_20:.*]]:2 = hlfir.declare %[[VAL_18]](%[[VAL_15]]) {uniq_name = ".tmp"} : (!fir.heap<!fir.array<1x2xi32>>, !fir.shape<2>) -> (!fir.heap<!fir.array<1x2xi32>>, !fir.heap<!fir.array<1x2xi32>>)275// CHECK: hlfir.assign %[[VAL_0]] to %[[VAL_20]]#0 temporary_lhs : !hlfir.expr<1x2xi32>, !fir.heap<!fir.array<1x2xi32>>276// CHECK: %[[VAL_19:.*]] = arith.constant true277// CHECK: %[[VAL_21:.*]] = fir.undefined tuple<!fir.heap<!fir.array<1x2xi32>>, i1>278// CHECK: %[[VAL_22:.*]] = fir.insert_value %[[VAL_21]], %[[VAL_19]], [1 : index] : (tuple<!fir.heap<!fir.array<1x2xi32>>, i1>, i1) -> tuple<!fir.heap<!fir.array<1x2xi32>>, i1>279// CHECK: %[[VAL_23:.*]] = fir.insert_value %[[VAL_22]], %[[VAL_20]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<1x2xi32>>, i1>, !fir.heap<!fir.array<1x2xi32>>) -> tuple<!fir.heap<!fir.array<1x2xi32>>, i1>280// CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_20]]#0 : (!fir.heap<!fir.array<1x2xi32>>) -> !fir.ref<!fir.array<1x2xi32>>281// CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_20]]#0 : (!fir.heap<!fir.array<1x2xi32>>) -> !fir.ref<!fir.array<1x2xi32>>282// end associate 0:283// CHECK: %[[VAL_26:.*]] = fir.convert %[[VAL_14]] : (!fir.ref<!fir.array<1x2xi32>>) -> !fir.heap<!fir.array<1x2xi32>>284// CHECK: fir.freemem %[[VAL_26]] : !fir.heap<!fir.array<1x2xi32>>285// end associate 1:286// CHECK: %[[VAL_27:.*]] = fir.convert %[[VAL_25]] : (!fir.ref<!fir.array<1x2xi32>>) -> !fir.heap<!fir.array<1x2xi32>>287// CHECK: fir.freemem %[[VAL_27]] : !fir.heap<!fir.array<1x2xi32>>288// CHECK: return289// CHECK: }290 291// test that we support a hlfir.associate operation where the expr is also used in a hlfir.get_length op292func.func @test_get_length(%arg0: !fir.ref<!fir.char<1,2>>) {293 %0 = hlfir.as_expr %arg0 : (!fir.ref<!fir.char<1,2>>) -> !hlfir.expr<!fir.char<1,2>>294 %1 = hlfir.get_length %0 : (!hlfir.expr<!fir.char<1,2>>) -> index295 %c2 = arith.constant 2 : index296 %2:3 = hlfir.associate %0 typeparams %c2 {adapt.valuebyref} : (!hlfir.expr<!fir.char<1,2>>, index) -> (!fir.ref<!fir.char<1,2>>, !fir.ref<!fir.char<1,2>>, i1)297 // ...298 hlfir.end_associate %2#1, %2#2 : !fir.ref<!fir.char<1,2>>, i1299 return300}301// CHECK-LABEL: func.func @test_get_length(302// CHECK-SAME: %[[VAL_0:.*]]: !fir.ref<!fir.char<1,2>>) {303// CHECK: %[[VAL_1:.*]] = fir.alloca !fir.char<1,2> {bindc_name = ".tmp"}304// CHECK: %[[VAL_2:.*]] = arith.constant 2 : index305// CHECK: %[[VAL_4:.*]]:2 = hlfir.declare %[[VAL_1]] typeparams %[[VAL_2]] {uniq_name = ".tmp"} : (!fir.ref<!fir.char<1,2>>, index) -> (!fir.ref<!fir.char<1,2>>, !fir.ref<!fir.char<1,2>>)306// CHECK: hlfir.assign %[[VAL_0]] to %[[VAL_4]]#0 temporary_lhs : !fir.ref<!fir.char<1,2>>, !fir.ref<!fir.char<1,2>>307// CHECK: %[[VAL_3:.*]] = arith.constant false308// CHECK: %[[VAL_5:.*]] = fir.undefined tuple<!fir.ref<!fir.char<1,2>>, i1>309// CHECK: %[[VAL_6:.*]] = fir.insert_value %[[VAL_5]], %[[VAL_3]], [1 : index] : (tuple<!fir.ref<!fir.char<1,2>>, i1>, i1) -> tuple<!fir.ref<!fir.char<1,2>>, i1>310// CHECK: %[[VAL_7:.*]] = fir.insert_value %[[VAL_6]], %[[VAL_4]]#0, [0 : index] : (tuple<!fir.ref<!fir.char<1,2>>, i1>, !fir.ref<!fir.char<1,2>>) -> tuple<!fir.ref<!fir.char<1,2>>, i1>311// CHECK: %[[VAL_8:.*]] = arith.constant 2 : index312// CHECK: return313// CHECK: }314 315func.func private @take_i4(!fir.ref<i32>)316func.func private @take_r4(!fir.ref<f32>)317func.func private @take_l4(!fir.ref<!fir.logical<4>>)318func.func private @take_c(!fir.boxchar<1>)319 320// Test the hlfir.associate/hlfir.end_associate does not take ownership over321// a single-use hlfir.expr if hlfir.end_associate might be executed more times322// than the producer of hlfir.expr. This might cause double-free effects.323func.func @_QPtest_multiple_expr_uses_inside_elemental() {324 %true = arith.constant true325 %18 = fir.undefined !fir.heap<!fir.char<1,?>>326 %17 = fir.undefined index327 %19:2 = hlfir.declare %18 typeparams %17 {uniq_name = ".tmp.intrinsic_result"} : (!fir.heap<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.heap<!fir.char<1,?>>)328 %20 = hlfir.as_expr %19#0 move %true : (!fir.boxchar<1>, i1) -> !hlfir.expr<!fir.char<1,?>>329 %21 = fir.undefined index330 %22 = fir.shape %21 : (index) -> !fir.shape<1>331 %23 = hlfir.elemental %22 unordered : (!fir.shape<1>) -> !hlfir.expr<?x!fir.logical<4>> {332 ^bb0(%arg2: index):333 %35:3 = hlfir.associate %20 typeparams %17 {adapt.valuebyref} : (!hlfir.expr<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>, i1)334 hlfir.end_associate %35#1, %35#2 : !fir.ref<!fir.char<1,?>>, i1335 %ci1 = arith.constant 1 : i1336 %42 = fir.convert %ci1 : (i1) -> !fir.logical<4>337 hlfir.yield_element %42 : !fir.logical<4>338 }339 return340}341// CHECK-LABEL: func.func @_QPtest_multiple_expr_uses_inside_elemental() {342// CHECK: %[[VAL_2:.*]] = arith.constant true343// CHECK: %[[VAL_3:.*]] = fir.undefined !fir.heap<!fir.char<1,?>>344// CHECK: %[[VAL_4:.*]] = fir.undefined index345// CHECK: %[[VAL_5:.*]]:2 = hlfir.declare %[[VAL_3]] typeparams %[[VAL_4]] {uniq_name = ".tmp.intrinsic_result"} : (!fir.heap<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.heap<!fir.char<1,?>>)346// CHECK: %[[VAL_6:.*]] = fir.undefined tuple<!fir.boxchar<1>, i1>347// CHECK: %[[VAL_7:.*]] = fir.insert_value %[[VAL_6]], %[[VAL_2]], [1 : index] : (tuple<!fir.boxchar<1>, i1>, i1) -> tuple<!fir.boxchar<1>, i1>348// CHECK: %[[VAL_8:.*]] = fir.insert_value %[[VAL_7]], %[[VAL_5]]#0, [0 : index] : (tuple<!fir.boxchar<1>, i1>, !fir.boxchar<1>) -> tuple<!fir.boxchar<1>, i1>349// CHECK: %[[VAL_9:.*]] = fir.undefined index350// CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_9]] : (index) -> !fir.shape<1>351// CHECK: %[[VAL_11:.*]] = fir.allocmem !fir.array<?x!fir.logical<4>>, %[[VAL_9]] {bindc_name = ".tmp.array", uniq_name = ""}352// CHECK: %[[VAL_12:.*]]:2 = hlfir.declare %[[VAL_11]](%[[VAL_10]]) {uniq_name = ".tmp.array"} : (!fir.heap<!fir.array<?x!fir.logical<4>>>, !fir.shape<1>) -> (!fir.box<!fir.array<?x!fir.logical<4>>>, !fir.heap<!fir.array<?x!fir.logical<4>>>)353// CHECK: %[[VAL_13:.*]] = arith.constant true354// CHECK: %[[VAL_14:.*]] = arith.constant 1 : index355// CHECK: fir.do_loop %[[VAL_15:.*]] = %[[VAL_14]] to %[[VAL_9]] step %[[VAL_14]] unordered {356// CHECK: %[[VAL_16:.*]] = fir.alloca !fir.char<1,?>(%[[VAL_4]] : index) {bindc_name = ".tmp"}357// CHECK: %[[VAL_18:.*]]:2 = hlfir.declare %[[VAL_16]] typeparams %[[VAL_4]] {uniq_name = ".tmp"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)358// CHECK: hlfir.assign %[[VAL_5]]#0 to %[[VAL_18]]#0 temporary_lhs : !fir.boxchar<1>, !fir.boxchar<1>359// CHECK: %[[VAL_17:.*]] = arith.constant false360// CHECK: %[[VAL_19:.*]] = fir.undefined tuple<!fir.boxchar<1>, i1>361// CHECK: %[[VAL_20:.*]] = fir.insert_value %[[VAL_19]], %[[VAL_17]], [1 : index] : (tuple<!fir.boxchar<1>, i1>, i1) -> tuple<!fir.boxchar<1>, i1>362// CHECK: %[[VAL_21:.*]] = fir.insert_value %[[VAL_20]], %[[VAL_18]]#0, [0 : index] : (tuple<!fir.boxchar<1>, i1>, !fir.boxchar<1>) -> tuple<!fir.boxchar<1>, i1>363// CHECK: %[[VAL_22:.*]] = arith.constant true364// CHECK: %[[VAL_23:.*]] = fir.convert %[[VAL_22]] : (i1) -> !fir.logical<4>365// CHECK: %[[VAL_24:.*]] = hlfir.designate %[[VAL_12]]#0 (%[[VAL_15]]) : (!fir.box<!fir.array<?x!fir.logical<4>>>, index) -> !fir.ref<!fir.logical<4>>366// CHECK: hlfir.assign %[[VAL_23]] to %[[VAL_24]] temporary_lhs : !fir.logical<4>, !fir.ref<!fir.logical<4>>367// CHECK: }368// CHECK: %[[VAL_25:.*]] = fir.undefined tuple<!fir.box<!fir.array<?x!fir.logical<4>>>, i1>369// CHECK: %[[VAL_26:.*]] = fir.insert_value %[[VAL_25]], %[[VAL_13]], [1 : index] : (tuple<!fir.box<!fir.array<?x!fir.logical<4>>>, i1>, i1) -> tuple<!fir.box<!fir.array<?x!fir.logical<4>>>, i1>370// CHECK: %[[VAL_27:.*]] = fir.insert_value %[[VAL_26]], %[[VAL_12]]#0, [0 : index] : (tuple<!fir.box<!fir.array<?x!fir.logical<4>>>, i1>, !fir.box<!fir.array<?x!fir.logical<4>>>) -> tuple<!fir.box<!fir.array<?x!fir.logical<4>>>, i1>371// CHECK: return372// CHECK: }373 374// Verify that we properly recognize multiple consequent hlfir.associate using375// the same result of hlfir.elemental.376func.func @_QPtest_multitple_associates_for_same_expr() {377 %c1 = arith.constant 1 : index378 %c10 = arith.constant 10 : index379 %4 = fir.shape %c10 : (index) -> !fir.shape<1>380 %11 = hlfir.elemental %4 typeparams %c1 unordered : (!fir.shape<1>, index) -> !hlfir.expr<10x!fir.char<1>> {381 ^bb0(%arg1: index):382 %44 = fir.undefined !fir.ref<!fir.char<1>>383 hlfir.yield_element %44 : !fir.ref<!fir.char<1>>384 }385 %12:3 = hlfir.associate %11(%4) typeparams %c1 {adapt.valuebyref} : (!hlfir.expr<10x!fir.char<1>>, !fir.shape<1>, index) -> (!fir.ref<!fir.array<10x!fir.char<1>>>, !fir.ref<!fir.array<10x!fir.char<1>>>, i1)386 hlfir.end_associate %12#1, %12#2 : !fir.ref<!fir.array<10x!fir.char<1>>>, i1387 %31:3 = hlfir.associate %11(%4) typeparams %c1 {adapt.valuebyref} : (!hlfir.expr<10x!fir.char<1>>, !fir.shape<1>, index) -> (!fir.ref<!fir.array<10x!fir.char<1>>>, !fir.ref<!fir.array<10x!fir.char<1>>>, i1)388 hlfir.end_associate %31#1, %31#2 : !fir.ref<!fir.array<10x!fir.char<1>>>, i1389 hlfir.destroy %11 : !hlfir.expr<10x!fir.char<1>>390 return391}392// CHECK-LABEL: func.func @_QPtest_multitple_associates_for_same_expr() {393// CHECK: %[[VAL_0:.*]] = arith.constant 1 : index394// CHECK: %[[VAL_1:.*]] = arith.constant 10 : index395// CHECK: %[[VAL_2:.*]] = fir.shape %[[VAL_1]] : (index) -> !fir.shape<1>396// CHECK: %[[VAL_3:.*]] = fir.allocmem !fir.array<10x!fir.char<1>> {bindc_name = ".tmp.array", uniq_name = ""}397// CHECK: %[[VAL_4:.*]]:2 = hlfir.declare %[[VAL_3]](%[[VAL_2]]) typeparams %[[VAL_0]] {uniq_name = ".tmp.array"} : (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.shape<1>, index) -> (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.heap<!fir.array<10x!fir.char<1>>>)398// CHECK: %[[VAL_5:.*]] = arith.constant true399// CHECK: %[[VAL_6:.*]] = arith.constant 1 : index400// CHECK: fir.do_loop %[[VAL_7:.*]] = %[[VAL_6]] to %[[VAL_1]] step %[[VAL_6]] unordered {401// CHECK: %[[VAL_8:.*]] = fir.undefined !fir.ref<!fir.char<1>>402// CHECK: %[[VAL_9:.*]] = hlfir.designate %[[VAL_4]]#0 (%[[VAL_7]]) typeparams %[[VAL_0]] : (!fir.heap<!fir.array<10x!fir.char<1>>>, index, index) -> !fir.ref<!fir.char<1>>403// CHECK: hlfir.assign %[[VAL_8]] to %[[VAL_9]] temporary_lhs : !fir.ref<!fir.char<1>>, !fir.ref<!fir.char<1>>404// CHECK: }405// CHECK: %[[VAL_10:.*]] = fir.undefined tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>406// CHECK: %[[VAL_11:.*]] = fir.insert_value %[[VAL_10]], %[[VAL_5]], [1 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, i1) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>407// CHECK: %[[VAL_12:.*]] = fir.insert_value %[[VAL_11]], %[[VAL_4]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, !fir.heap<!fir.array<10x!fir.char<1>>>) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>408// CHECK: %[[VAL_13:.*]] = fir.allocmem !fir.array<10x!fir.char<1>> {bindc_name = ".tmp", uniq_name = ""}409// CHECK: %[[VAL_15:.*]]:2 = hlfir.declare %[[VAL_13]](%[[VAL_2]]) typeparams %[[VAL_0]] {uniq_name = ".tmp"} : (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.shape<1>, index) -> (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.heap<!fir.array<10x!fir.char<1>>>)410// CHECK: hlfir.assign %[[VAL_4]]#0 to %[[VAL_15]]#0 temporary_lhs : !fir.heap<!fir.array<10x!fir.char<1>>>, !fir.heap<!fir.array<10x!fir.char<1>>>411// CHECK: %[[VAL_14:.*]] = arith.constant true412// CHECK: %[[VAL_16:.*]] = fir.undefined tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>413// CHECK: %[[VAL_17:.*]] = fir.insert_value %[[VAL_16]], %[[VAL_14]], [1 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, i1) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>414// CHECK: %[[VAL_18:.*]] = fir.insert_value %[[VAL_17]], %[[VAL_15]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, !fir.heap<!fir.array<10x!fir.char<1>>>) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>415// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]]#0 : (!fir.heap<!fir.array<10x!fir.char<1>>>) -> !fir.ref<!fir.array<10x!fir.char<1>>>416// CHECK: %[[VAL_20:.*]] = fir.convert %[[VAL_15]]#0 : (!fir.heap<!fir.array<10x!fir.char<1>>>) -> !fir.ref<!fir.array<10x!fir.char<1>>>417// CHECK: %[[VAL_21:.*]] = fir.convert %[[VAL_20]] : (!fir.ref<!fir.array<10x!fir.char<1>>>) -> !fir.heap<!fir.array<10x!fir.char<1>>>418// CHECK: fir.freemem %[[VAL_21]] : !fir.heap<!fir.array<10x!fir.char<1>>>419// CHECK: %[[VAL_22:.*]] = fir.allocmem !fir.array<10x!fir.char<1>> {bindc_name = ".tmp", uniq_name = ""}420// CHECK: %[[VAL_24:.*]]:2 = hlfir.declare %[[VAL_22]](%[[VAL_2]]) typeparams %[[VAL_0]] {uniq_name = ".tmp"} : (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.shape<1>, index) -> (!fir.heap<!fir.array<10x!fir.char<1>>>, !fir.heap<!fir.array<10x!fir.char<1>>>)421// CHECK: hlfir.assign %[[VAL_4]]#0 to %[[VAL_24]]#0 temporary_lhs : !fir.heap<!fir.array<10x!fir.char<1>>>, !fir.heap<!fir.array<10x!fir.char<1>>>422// CHECK: %[[VAL_23:.*]] = arith.constant true423// CHECK: %[[VAL_25:.*]] = fir.undefined tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>424// CHECK: %[[VAL_26:.*]] = fir.insert_value %[[VAL_25]], %[[VAL_23]], [1 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, i1) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>425// CHECK: %[[VAL_27:.*]] = fir.insert_value %[[VAL_26]], %[[VAL_24]]#0, [0 : index] : (tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>, !fir.heap<!fir.array<10x!fir.char<1>>>) -> tuple<!fir.heap<!fir.array<10x!fir.char<1>>>, i1>426// CHECK: %[[VAL_28:.*]] = fir.convert %[[VAL_24]]#0 : (!fir.heap<!fir.array<10x!fir.char<1>>>) -> !fir.ref<!fir.array<10x!fir.char<1>>>427// CHECK: %[[VAL_29:.*]] = fir.convert %[[VAL_24]]#0 : (!fir.heap<!fir.array<10x!fir.char<1>>>) -> !fir.ref<!fir.array<10x!fir.char<1>>>428// CHECK: %[[VAL_30:.*]] = fir.convert %[[VAL_29]] : (!fir.ref<!fir.array<10x!fir.char<1>>>) -> !fir.heap<!fir.array<10x!fir.char<1>>>429// CHECK: fir.freemem %[[VAL_30]] : !fir.heap<!fir.array<10x!fir.char<1>>>430// CHECK: fir.freemem %[[VAL_4]]#0 : !fir.heap<!fir.array<10x!fir.char<1>>>431// CHECK: return432// CHECK: }433 434// Test hlfir.associate codegen, when its operand is used435// by hlfir.shape located in a block different from the block436// of the hlfir.end_associate.437func.func @_QPtest(%arg0: index, %arg1: index, %arg2 : i32) {438 %c1_i32 = arith.constant 1 : i32439 %3 = fir.shape %arg0, %arg1 : (index, index) -> !fir.shape<2>440 %4 = hlfir.elemental %3 unordered : (!fir.shape<2>) -> !hlfir.expr<?x?xi32> {441 ^bb0(%arg3: index, %arg4: index):442 %16 = fir.undefined i32443 hlfir.yield_element %16 : i32444 }445 %5 = hlfir.shape_of %4 : (!hlfir.expr<?x?xi32>) -> !fir.shape<2>446 %6:3 = hlfir.associate %4(%5) {adapt.valuebyref} : (!hlfir.expr<?x?xi32>, !fir.shape<2>) -> (!fir.box<!fir.array<?x?xi32>>, !fir.ref<!fir.array<?x?xi32>>, i1)447 %13 = arith.cmpi ne, %arg2, %c1_i32 : i32448 cf.cond_br %13, ^bb1, ^bb2449^bb1: // pred: ^bb0450 fir.unreachable451^bb2: // pred: ^bb0452 hlfir.end_associate %6#1, %6#2 : !fir.ref<!fir.array<?x?xi32>>, i1453 hlfir.destroy %4 : !hlfir.expr<?x?xi32>454 return455}456