250 lines · plain
1// Test hlfir.minval operation parse, verify (no errors), and unparse2 3// RUN: fir-opt %s | fir-opt | FileCheck %s4 5// array is an expression of known shape6func.func @minval0(%arg0: !hlfir.expr<42xi32>) {7 %mask = fir.alloca !fir.logical<4>8 %c_1 = arith.constant 1 : index9 %true = arith.constant true10 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>11 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>12 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>13 %0 = hlfir.minval %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i3214 return15}16// CHECK: func.func @minval0(%[[ARRAY:.*]]: !hlfir.expr<42xi32>) {17// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>18// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index19// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true20// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>21// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>22// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>23// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i3224// CHECK-NEXT: return25// CHECK-NEXT: }26 27// array is an expression of assumed shape28func.func @minval1(%arg0: !hlfir.expr<?xi32>) {29 %mask = fir.alloca !fir.logical<4>30 %c_1 = arith.constant 1 : index31 %true = arith.constant true32 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>33 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>34 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>35 %0 = hlfir.minval %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i3236 return37}38// CHECK: func.func @minval1(%[[ARRAY:.*]]: !hlfir.expr<?xi32>) {39// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>40// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index41// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true42// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>43// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>44// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>45// CHECK-NEXT: hlfir.minval %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i3246// CHECK-NEXT: return47// CHECK-NEXT: }48 49// boxed array50func.func @minval2(%arg0: !fir.box<!fir.array<42xi32>>) {51 %mask = fir.alloca !fir.logical<4>52 %c_1 = arith.constant 1 : index53 %true = arith.constant true54 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>55 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>56 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>57 %0 = hlfir.minval %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i3258 return59}60// CHECK: func.func @minval2(%[[ARRAY:.*]]: !fir.box<!fir.array<42xi32>>) {61// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>62// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index63// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true64// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>65// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>66// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>67// CHECK-NEXT: hlfir.minval %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i3268// CHECK-NEXT: return69// CHECK-NEXT: }70 71// assumed shape boxed array72func.func @minval3(%arg0: !fir.box<!fir.array<?xi32>>) {73 %mask = fir.alloca !fir.logical<4>74 %c_1 = arith.constant 1 : index75 %true = arith.constant true76 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>77 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>78 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>79 %0 = hlfir.minval %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i3280 return81}82// CHECK: func.func @minval3(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>) {83// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>84// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index85// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true86// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>87// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>88// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>89// CHECK-NEXT: hlfir.minval %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i3290// CHECK-NEXT: return91// CHECK-NEXT: }92 93// known shape expr mask94func.func @minval4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<42x!fir.logical<4>>) {95 %c_1 = arith.constant 1 : index96 %0 = hlfir.minval %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i3297 return98}99// CHECK: func.func @minval4(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<42x!fir.logical<4>>) {100// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index101// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32102// CHECK-NEXT: return103// CHECK-NEXT: }104 105// assumed shape expr mask106func.func @minval5(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<?x!fir.logical<4>>) {107 %c_1 = arith.constant 1 : index108 %0 = hlfir.minval %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32109 return110}111// CHECK: func.func @minval5(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<?x!fir.logical<4>>) {112// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index113// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32114// CHECK-NEXT: return115// CHECK-NEXT: }116 117// known shape array mask118func.func @minval6(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<42x!fir.logical<4>>>) {119 %c_1 = arith.constant 1 : index120 %0 = hlfir.minval %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32121 return122}123// CHECK: func.func @minval6(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<42x!fir.logical<4>>>) {124// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index125// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32126// CHECK-NEXT: return127// CHECK-NEXT: }128 129// assumed shape array mask130func.func @minval7(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<?x!fir.logical<4>>>) {131 %c_1 = arith.constant 1 : index132 %0 = hlfir.minval %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32133 return134}135// CHECK: func.func @minval7(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {136// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index137// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32138// CHECK-NEXT: return139// CHECK-NEXT: }140 141// known shape expr return142func.func @minval8(%arg0: !fir.box<!fir.array<2x2xi32>>, %arg1: i32) {143 %mask = fir.alloca !fir.logical<4>144 %true = arith.constant true145 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>146 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>147 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>148 %0 = hlfir.minval %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>149 return150}151// CHECK: func.func @minval8(%[[ARRAY:.*]]: !fir.box<!fir.array<2x2xi32>>, %[[DIM:.*]]: i32) {152// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>153// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true154// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>155// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>156// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>157// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>158// CHECK-NEXT: return159// CHECK-NEXT: }160 161// assumed shape expr return162func.func @minval9(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {163 %mask = fir.alloca !fir.logical<4>164 %true = arith.constant true165 %true_logical = fir.convert %true : (i1) -> !fir.logical<4>166 fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>167 %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>168 %0 = hlfir.minval %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>169 return170}171// CHECK: func.func @minval9(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32) {172// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>173// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true174// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>175// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>176// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>177// CHECK-NEXT: hlfir.minval %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>178// CHECK-NEXT: return179// CHECK-NEXT: }180 181// hlfir.minval with only an array argument182func.func @minval10(%arg0: !fir.box<!fir.array<?x?xi32>>) {183 %minval = hlfir.minval %arg0 : (!fir.box<!fir.array<?x?xi32>>) -> i32184 return185}186// CHECK: func.func @minval10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>187// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] : (!fir.box<!fir.array<?x?xi32>>) -> i32188// CHECK-NEXT: return189// CHECK-NEXT: }190 191// hlfir.minval with only a character array argument192func.func @minval11(%arg0: !fir.box<!fir.array<?x?x!fir.char<1,?>>>) {193 %minval = hlfir.minval %arg0 : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<!fir.char<1,?>>194 return195}196// CHECK: func.func @minval11(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.char<1,?>>>197// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<!fir.char<1,?>>198// CHECK-NEXT: return199// CHECK-NEXT: }200 201// hlfir.minval with array and dim argument202func.func @minval12(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {203 %minval = hlfir.minval %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>204 return205}206// CHECK: func.func @minval12(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32207// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>208// CHECK-NEXT: return209// CHECK-NEXT: }210 211// hlfir.minval with array and mask argument212func.func @minval13(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.logical<4>) {213 %minval = hlfir.minval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32214 return215}216// CHECK: func.func @minval13(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.logical<4>217// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32218// CHECK-NEXT: return219// CHECK-NEXT: }220 221// hlfir.minval with dim argument with an unusual type222func.func @minval14(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: index) {223 %minval = hlfir.minval %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>224 return225}226// CHECK: func.func @minval14(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: index227// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>228// CHECK-NEXT: return229// CHECK-NEXT: }230 231// hlfir.minval with mask argument of unusual type232func.func @minval15(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {233 %minval = hlfir.minval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> i32234 return235}236// CHECK: func.func @minval15(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1237// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> i32238// CHECK-NEXT: return239// CHECK-NEXT: }240 241// hlfir.minval with mask argument of ref<array<>> type242func.func @minval16(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.ref<!fir.array<?x!fir.logical<4>>>) {243 %minval = hlfir.minval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32244 return245}246// CHECK: func.func @minval16(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>247// CHECK-NEXT: %[[minval:.*]] = hlfir.minval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32248// CHECK-NEXT: return249// CHECK-NEXT: }250