flang/test/HLFIR/forall_mask.fir

   1 // Test hlfir.forall_mask operation parse, verify (no errors), and unparse.
   2 // RUN: fir-opt %s | fir-opt | FileCheck %s
   3
   4 func.func @forall_mask_test(%x: !fir.box<!fir.array<?xf32>>) {
   5   %c1 = arith.constant 1 : index
   6   %c10 = arith.constant 10 : index
   7   hlfir.forall lb {
   8     hlfir.yield %c1 : index
   9   } ub {
  10     hlfir.yield %c10 : index
  11   } (%i : index) {
  12     hlfir.forall_mask {
  13       %mask = fir.call @some_condition(%i) : (index) -> i1
  14       hlfir.yield %mask : i1
  15     } do {
  16       hlfir.region_assign {
  17         %res = fir.call @foo(%i) : (index) -> f32
  18         hlfir.yield %res : f32
  19       } to {
  20         %xi = hlfir.designate %x(%i) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
  21         hlfir.yield %xi : !fir.ref<f32>
  22       }
  23     }
  24   }
  25   return
  26 }
  27
  28 // CHECK-LABEL:   func.func @forall_mask_test(
  29 // CHECK-SAME:                                %[[VAL_0:.*]]: !fir.box<!fir.array<?xf32>>) {
  30 // CHECK:           %[[VAL_1:.*]] = arith.constant 1 : index
  31 // CHECK:           %[[VAL_2:.*]] = arith.constant 10 : index
  32 // CHECK:           hlfir.forall lb {
  33 // CHECK:             hlfir.yield %[[VAL_1]] : index
  34 // CHECK:           } ub {
  35 // CHECK:             hlfir.yield %[[VAL_2]] : index
  36 // CHECK:           }  (%[[VAL_3:.*]]: index) {
  37 // CHECK:             hlfir.forall_mask {
  38 // CHECK:               %[[VAL_4:.*]] = fir.call @some_condition(%[[VAL_3]]) : (index) -> i1
  39 // CHECK:               hlfir.yield %[[VAL_4]] : i1
  40 // CHECK:             } do {
  41 // CHECK:               hlfir.region_assign {
  42 // CHECK:                 %[[VAL_5:.*]] = fir.call @foo(%[[VAL_3]]) : (index) -> f32
  43 // CHECK:                 hlfir.yield %[[VAL_5]] : f32
  44 // CHECK:               } to {
  45 // CHECK:                 %[[VAL_6:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_3]])  : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
  46 // CHECK:                 hlfir.yield %[[VAL_6]] : !fir.ref<f32>
  47 // CHECK:               }
  48 // CHECK:             }
  49 // CHECK:           }