1// Test when the saved mask is used in the same run to save 2// another value (like the RHS). 3// RUN: fir-opt %s --lower-hlfir-ordered-assignments | FileCheck %s 4 5func.func @saving_mask_and_rhs(%arg0: !fir.ref<!fir.array<10xi32>>) { 6 %c-1 = arith.constant -1 : index 7 %c1 = arith.constant 1 : index 8 %c0_i32 = arith.constant 0 : i32 9 %c10 = arith.constant 10 : index 10 %0 = fir.shape %c10 : (index) -> !fir.shape<1> 11 %1:2 = hlfir.declare %arg0(%0) {uniq_name = "x"} : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<10xi32>>, !fir.ref<!fir.array<10xi32>>) 12 hlfir.where { 13 %2 = hlfir.elemental %0 : (!fir.shape<1>) -> !hlfir.expr<10x!fir.logical<4>> { 14 ^bb0(%arg1: index): 15 %3 = hlfir.designate %1#0 (%arg1) : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32> 16 %4 = fir.load %3 : !fir.ref<i32> 17 %5 = arith.cmpi sgt, %4, %c0_i32 : i32 18 %6 = fir.convert %5 : (i1) -> !fir.logical<4> 19 hlfir.yield_element %6 : !fir.logical<4> 20 } 21 hlfir.yield %2 : !hlfir.expr<10x!fir.logical<4>> cleanup { 22 hlfir.destroy %2 : !hlfir.expr<10x!fir.logical<4>> 23 } 24 } do { 25 hlfir.region_assign { 26 hlfir.yield %1#0 : !fir.ref<!fir.array<10xi32>> 27 } to { 28 %2 = hlfir.designate %1#0 (%c10:%c1:%c-1) shape %0 : (!fir.ref<!fir.array<10xi32>>, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<10xi32>> 29 hlfir.yield %2 : !fir.box<!fir.array<10xi32>> 30 } 31 } 32 return 33} 34 35 36// Creating mask temporary. 37 38// CHECK-LABEL: func.func @saving_mask_and_rhs( 39// CHECK: %[[VAL_8:.*]] = hlfir.elemental {{.*}} !hlfir.expr<10x!fir.logical<4>> 40// CHECK: %[[VAL_14:.*]]:3 = hlfir.associate %[[VAL_8]]({{.*}}) {uniq_name = ".tmp.where"} 41 42// Creating RHS temporary using the mask temporary (and not the hlfir.elemental) 43 44// CHECK: %[[VAL_25:.*]] = fir.allocmem !fir.array<?xi32>, %{{.*}} {bindc_name = ".tmp.where", uniq_name = ""} 45// CHECK: %[[VAL_27:.*]]:2 = hlfir.declare %[[VAL_25]]({{.*}}) {uniq_name = ".tmp.where"} 46// CHECK: fir.do_loop 47// CHECK: %[[VAL_29:.*]] = hlfir.designate %[[VAL_14]]#0 ({{.*}}) 48// CHECK: %[[VAL_30:.*]] = fir.load %[[VAL_29]] : !fir.ref<!fir.logical<4>> 49// CHECK: %[[VAL_31:.*]] = fir.convert %[[VAL_30]] : (!fir.logical<4>) -> i1 50// CHECK: fir.if %[[VAL_31]] { 51// CHECK: %[[VAL_36:.*]] = hlfir.designate %[[VAL_27]]#0 ({{.*}}) 52// CHECK: hlfir.assign %{{.*}} to %[[VAL_36]] : i32, !fir.ref<i32> 53// CHECK: } 54// CHECK: } 55 56func.func @forall_mask_and_rhs(%arg0: !fir.ref<!fir.array<10xi32>>) { 57 %c0_i32 = arith.constant 0 : i32 58 %c1 = arith.constant 1 : index 59 %c10 = arith.constant 10 : index 60 %c11 = arith.constant 11 : index 61 %0 = fir.shape %c10 : (index) -> !fir.shape<1> 62 %1:2 = hlfir.declare %arg0(%0) {uniq_name = "x"} : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<10xi32>>, !fir.ref<!fir.array<10xi32>>) 63 hlfir.forall lb { 64 hlfir.yield %c1 : index 65 } ub { 66 hlfir.yield %c10 : index 67 } (%arg1: index) { 68 hlfir.forall_mask { 69 %2 = arith.subi %c11, %arg1 : index 70 %3 = hlfir.designate %1#0 (%2) : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32> 71 %4 = fir.load %3 : !fir.ref<i32> 72 %5 = arith.cmpi sgt, %4, %c0_i32 : i32 73 hlfir.yield %5 : i1 74 } do { 75 hlfir.region_assign { 76 %2 = arith.subi %c11, %arg1 : index 77 %3 = hlfir.designate %1#0 (%2) : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32> 78 %4 = fir.load %3 : !fir.ref<i32> 79 hlfir.yield %4 : i32 80 } to { 81 %2 = hlfir.designate %1#0 (%arg1) : (!fir.ref<!fir.array<10xi32>>, index) -> !fir.ref<i32> 82 hlfir.yield %2 : !fir.ref<i32> 83 } 84 } 85 } 86 return 87} 88 89// The mask and rhs are saved in the same loop, the mask value is a scalar i1 90// and it can be used directly to mask the rhs (instead of loading the temp). 91 92// CHECK-LABEL: func.func @forall_mask_and_rhs( 93// CHECK: %[[VAL_18:.*]] = fir.allocmem !fir.array<?xi1>, %{{.*}} {bindc_name = ".tmp.forall", uniq_name = ""} 94// CHECK: %[[VAL_20:.*]]:2 = hlfir.declare %[[VAL_18]](%{{.*}}) {uniq_name = ".tmp.forall"} 95// CHECK: %[[VAL_29:.*]] = fir.allocmem !fir.array<?xi32>, %{{.*}} {bindc_name = ".tmp.forall", uniq_name = ""} 96// CHECK: %[[VAL_31:.*]]:2 = hlfir.declare %[[VAL_29]](%{{.*}}) {uniq_name = ".tmp.forall"} 97// CHECK: %[[VAL_36:.*]] = arith.cmpi sgt, %{{.*}}, %{{.*}} : i32 98// CHECK: %[[VAL_39:.*]] = hlfir.designate %[[VAL_20]]#0 (%{{.*}}) 99// CHECK: hlfir.assign %[[VAL_36]] to %[[VAL_39]] : i1, !fir.ref<i1> 100// CHECK: fir.if %[[VAL_36]] { 101// CHECK: %[[VAL_45:.*]] = hlfir.designate %[[VAL_31]]#0 (%{{.*}}) 102// CHECK: hlfir.assign %{{.*}} to %[[VAL_45]] : i32, !fir.ref<i32> 103// CHECK: } 104// CHECK: } 105