flang/test/HLFIR/order_assignments/saving-mask-and-rhs.fir

   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
   5 func.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
  56 func.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:           }