50 lines · plain
1// Test hlfir.forall_mask operation parse, verify (no errors), and unparse.2// RUN: fir-opt %s | fir-opt | FileCheck %s3 4func.func @forall_mask_test(%x: !fir.box<!fir.array<?xf32>>) {5 %c1 = arith.constant 1 : index6 %c10 = arith.constant 10 : index7 hlfir.forall lb {8 hlfir.yield %c1 : index9 } ub {10 hlfir.yield %c10 : index11 } (%i : index) {12 hlfir.forall_mask {13 %mask = fir.call @some_condition(%i) : (index) -> i114 hlfir.yield %mask : i115 } do {16 hlfir.region_assign {17 %res = fir.call @foo(%i) : (index) -> f3218 hlfir.yield %res : f3219 } 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 return26}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 : index31// CHECK: %[[VAL_2:.*]] = arith.constant 10 : index32// CHECK: hlfir.forall lb {33// CHECK: hlfir.yield %[[VAL_1]] : index34// CHECK: } ub {35// CHECK: hlfir.yield %[[VAL_2]] : index36// CHECK: } (%[[VAL_3:.*]]: index) {37// CHECK: hlfir.forall_mask {38// CHECK: %[[VAL_4:.*]] = fir.call @some_condition(%[[VAL_3]]) : (index) -> i139// CHECK: hlfir.yield %[[VAL_4]] : i140// CHECK: } do {41// CHECK: hlfir.region_assign {42// CHECK: %[[VAL_5:.*]] = fir.call @foo(%[[VAL_3]]) : (index) -> f3243// CHECK: hlfir.yield %[[VAL_5]] : f3244// 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: }50