1// Test hlfir.forall_mask operation parse, verify (no errors), and unparse. 2// RUN: fir-opt %s | fir-opt | FileCheck %s 3 4func.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: } 50