440 lines · plain
1// RUN: fir-opt --simplify-hlfir-intrinsics %s | FileCheck %s2 3// box with known extents4func.func @sum_box_known_extents(%arg0: !fir.box<!fir.array<2x3xi32>>) -> !hlfir.expr<2xi32> {5 %cst = arith.constant 2 : i326 %res = hlfir.sum %arg0 dim %cst : (!fir.box<!fir.array<2x3xi32>>, i32) -> !hlfir.expr<2xi32>7 return %res : !hlfir.expr<2xi32>8}9// CHECK-LABEL: func.func @sum_box_known_extents(10// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<2x3xi32>>) -> !hlfir.expr<2xi32> {11// CHECK: %[[VAL_12:.*]] = arith.constant 0 : index12// CHECK: %[[VAL_8:.*]] = arith.constant 1 : index13// CHECK: %[[VAL_7:.*]] = arith.constant 0 : i3214// CHECK: %[[VAL_2:.*]] = arith.constant 2 : index15// CHECK: %[[VAL_3:.*]] = arith.constant 3 : index16// CHECK: %[[VAL_4:.*]] = fir.shape %[[VAL_2]] : (index) -> !fir.shape<1>17// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] unordered : (!fir.shape<1>) -> !hlfir.expr<2xi32> {18// CHECK: ^bb0(%[[VAL_6:.*]]: index):19// CHECK: %[[VAL_9:.*]] = fir.do_loop %[[VAL_10:.*]] = %[[VAL_8]] to %[[VAL_3]] step %[[VAL_8]] unordered iter_args(%[[VAL_11:.*]] = %[[VAL_7]]) -> (i32) {20// CHECK: %[[VAL_13:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_12]] : (!fir.box<!fir.array<2x3xi32>>, index) -> (index, index, index)21// CHECK: %[[VAL_15:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_8]] : (!fir.box<!fir.array<2x3xi32>>, index) -> (index, index, index)22// CHECK: %[[VAL_17:.*]] = arith.subi %[[VAL_13]]#0, %[[VAL_8]] : index23// CHECK: %[[VAL_18:.*]] = arith.addi %[[VAL_6]], %[[VAL_17]] : index24// CHECK: %[[VAL_19:.*]] = arith.subi %[[VAL_15]]#0, %[[VAL_8]] : index25// CHECK: %[[VAL_20:.*]] = arith.addi %[[VAL_10]], %[[VAL_19]] : index26// CHECK: %[[VAL_21:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_18]], %[[VAL_20]]) : (!fir.box<!fir.array<2x3xi32>>, index, index) -> !fir.ref<i32>27// CHECK: %[[VAL_22:.*]] = fir.load %[[VAL_21]] : !fir.ref<i32>28// CHECK: %[[VAL_23:.*]] = arith.addi %[[VAL_11]], %[[VAL_22]] : i3229// CHECK: fir.result %[[VAL_23]] : i3230// CHECK: }31// CHECK: hlfir.yield_element %[[VAL_9]] : i3232// CHECK: }33// CHECK: return34// CHECK: }35 36// expr with known extents37func.func @sum_expr_known_extents(%arg0: !hlfir.expr<2x3xi32>) -> !hlfir.expr<3xi32> {38 %cst = arith.constant 1 : i3239 %res = hlfir.sum %arg0 dim %cst : (!hlfir.expr<2x3xi32>, i32) -> !hlfir.expr<3xi32>40 return %res : !hlfir.expr<3xi32>41}42// CHECK-LABEL: func.func @sum_expr_known_extents(43// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<2x3xi32>) -> !hlfir.expr<3xi32> {44// CHECK: %[[VAL_8:.*]] = arith.constant 1 : index45// CHECK: %[[VAL_7:.*]] = arith.constant 0 : i3246// CHECK: %[[VAL_2:.*]] = arith.constant 2 : index47// CHECK: %[[VAL_3:.*]] = arith.constant 3 : index48// CHECK: %[[VAL_4:.*]] = fir.shape %[[VAL_3]] : (index) -> !fir.shape<1>49// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xi32> {50// CHECK: ^bb0(%[[VAL_6:.*]]: index):51// CHECK: %[[VAL_9:.*]] = fir.do_loop %[[VAL_10:.*]] = %[[VAL_8]] to %[[VAL_2]] step %[[VAL_8]] unordered iter_args(%[[VAL_11:.*]] = %[[VAL_7]]) -> (i32) {52// CHECK: %[[VAL_12:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_10]], %[[VAL_6]] : (!hlfir.expr<2x3xi32>, index, index) -> i3253// CHECK: %[[VAL_13:.*]] = arith.addi %[[VAL_11]], %[[VAL_12]] : i3254// CHECK: fir.result %[[VAL_13]] : i3255// CHECK: }56// CHECK: hlfir.yield_element %[[VAL_9]] : i3257// CHECK: }58// CHECK: return59// CHECK: }60 61// box with unknown extent62func.func @sum_box_unknown_extent1(%arg0: !fir.box<!fir.array<?x3xcomplex<f64>>>) -> !hlfir.expr<3xcomplex<f64>> {63 %cst = arith.constant 1 : i3264 %res = hlfir.sum %arg0 dim %cst : (!fir.box<!fir.array<?x3xcomplex<f64>>>, i32) -> !hlfir.expr<3xcomplex<f64>>65 return %res : !hlfir.expr<3xcomplex<f64>>66}67// CHECK-LABEL: func.func @sum_box_unknown_extent1(68// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?x3xcomplex<f64>>>) -> !hlfir.expr<3xcomplex<f64>> {69// CHECK: %[[VAL_12:.*]] = arith.constant 1 : index70// CHECK: %[[VAL_8:.*]] = arith.constant 0.000000e+00 : f6471// CHECK: %[[VAL_4:.*]] = arith.constant 3 : index72// CHECK: %[[VAL_2:.*]] = arith.constant 0 : index73// CHECK: %[[VAL_3:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_2]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)74// CHECK: %[[VAL_5:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1>75// CHECK: %[[VAL_6:.*]] = hlfir.elemental %[[VAL_5]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xcomplex<f64>> {76// CHECK: ^bb0(%[[VAL_7:.*]]: index):77// CHECK: %[[VAL_9:.*]] = fir.undefined complex<f64>78// CHECK: %[[VAL_10:.*]] = fir.insert_value %[[VAL_9]], %[[VAL_8]], [0 : index] : (complex<f64>, f64) -> complex<f64>79// CHECK: %[[VAL_11:.*]] = fir.insert_value %[[VAL_10]], %[[VAL_8]], [1 : index] : (complex<f64>, f64) -> complex<f64>80// CHECK: %[[VAL_13:.*]] = fir.do_loop %[[VAL_14:.*]] = %[[VAL_12]] to %[[VAL_3]]#1 step %[[VAL_12]] iter_args(%[[VAL_15:.*]] = %[[VAL_11]]) -> (complex<f64>) {81// CHECK: %[[VAL_17:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_2]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)82// CHECK: %[[VAL_19:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_12]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)83// CHECK: %[[VAL_21:.*]] = arith.subi %[[VAL_17]]#0, %[[VAL_12]] : index84// CHECK: %[[VAL_22:.*]] = arith.addi %[[VAL_14]], %[[VAL_21]] : index85// CHECK: %[[VAL_23:.*]] = arith.subi %[[VAL_19]]#0, %[[VAL_12]] : index86// CHECK: %[[VAL_24:.*]] = arith.addi %[[VAL_7]], %[[VAL_23]] : index87// CHECK: %[[VAL_25:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_22]], %[[VAL_24]]) : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index, index) -> !fir.ref<complex<f64>>88// CHECK: %[[VAL_26:.*]] = fir.load %[[VAL_25]] : !fir.ref<complex<f64>>89// CHECK: %[[VAL_27:.*]] = fir.addc %[[VAL_15]], %[[VAL_26]] : complex<f64>90// CHECK: fir.result %[[VAL_27]] : complex<f64>91// CHECK: }92// CHECK: hlfir.yield_element %[[VAL_13]] : complex<f64>93// CHECK: }94// CHECK: return95// CHECK: }96 97func.func @sum_box_unknown_extent2(%arg0: !fir.box<!fir.array<?x3xcomplex<f64>>>) -> !hlfir.expr<?xcomplex<f64>> {98 %cst = arith.constant 2 : i3299 %res = hlfir.sum %arg0 dim %cst : (!fir.box<!fir.array<?x3xcomplex<f64>>>, i32) -> !hlfir.expr<?xcomplex<f64>>100 return %res : !hlfir.expr<?xcomplex<f64>>101}102// CHECK-LABEL: func.func @sum_box_unknown_extent2(103// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?x3xcomplex<f64>>>) -> !hlfir.expr<?xcomplex<f64>> {104// CHECK: %[[VAL_12:.*]] = arith.constant 1 : index105// CHECK: %[[VAL_8:.*]] = arith.constant 0.000000e+00 : f64106// CHECK: %[[VAL_4:.*]] = arith.constant 3 : index107// CHECK: %[[VAL_2:.*]] = arith.constant 0 : index108// CHECK: %[[VAL_3:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_2]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)109// CHECK: %[[VAL_5:.*]] = fir.shape %[[VAL_3]]#1 : (index) -> !fir.shape<1>110// CHECK: %[[VAL_6:.*]] = hlfir.elemental %[[VAL_5]] unordered : (!fir.shape<1>) -> !hlfir.expr<?xcomplex<f64>> {111// CHECK: ^bb0(%[[VAL_7:.*]]: index):112// CHECK: %[[VAL_9:.*]] = fir.undefined complex<f64>113// CHECK: %[[VAL_10:.*]] = fir.insert_value %[[VAL_9]], %[[VAL_8]], [0 : index] : (complex<f64>, f64) -> complex<f64>114// CHECK: %[[VAL_11:.*]] = fir.insert_value %[[VAL_10]], %[[VAL_8]], [1 : index] : (complex<f64>, f64) -> complex<f64>115// CHECK: %[[VAL_13:.*]] = fir.do_loop %[[VAL_14:.*]] = %[[VAL_12]] to %[[VAL_4]] step %[[VAL_12]] iter_args(%[[VAL_15:.*]] = %[[VAL_11]]) -> (complex<f64>) {116// CHECK: %[[VAL_17:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_2]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)117// CHECK: %[[VAL_19:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_12]] : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index) -> (index, index, index)118// CHECK: %[[VAL_21:.*]] = arith.subi %[[VAL_17]]#0, %[[VAL_12]] : index119// CHECK: %[[VAL_22:.*]] = arith.addi %[[VAL_7]], %[[VAL_21]] : index120// CHECK: %[[VAL_23:.*]] = arith.subi %[[VAL_19]]#0, %[[VAL_12]] : index121// CHECK: %[[VAL_24:.*]] = arith.addi %[[VAL_14]], %[[VAL_23]] : index122// CHECK: %[[VAL_25:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_22]], %[[VAL_24]]) : (!fir.box<!fir.array<?x3xcomplex<f64>>>, index, index) -> !fir.ref<complex<f64>>123// CHECK: %[[VAL_26:.*]] = fir.load %[[VAL_25]] : !fir.ref<complex<f64>>124// CHECK: %[[VAL_27:.*]] = fir.addc %[[VAL_15]], %[[VAL_26]] : complex<f64>125// CHECK: fir.result %[[VAL_27]] : complex<f64>126// CHECK: }127// CHECK: hlfir.yield_element %[[VAL_13]] : complex<f64>128// CHECK: }129// CHECK: return130// CHECK: }131 132// expr with unknown extent133func.func @sum_expr_unknown_extent1(%arg0: !hlfir.expr<?x3xf32>) -> !hlfir.expr<3xf32> {134 %cst = arith.constant 1 : i32135 %res = hlfir.sum %arg0 dim %cst : (!hlfir.expr<?x3xf32>, i32) -> !hlfir.expr<3xf32>136 return %res : !hlfir.expr<3xf32>137}138// CHECK-LABEL: func.func @sum_expr_unknown_extent1(139// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>) -> !hlfir.expr<3xf32> {140// CHECK: %[[VAL_9:.*]] = arith.constant 1 : index141// CHECK: %[[VAL_8:.*]] = arith.constant 0.000000e+00 : f32142// CHECK: %[[VAL_4:.*]] = arith.constant 3 : index143// CHECK: %[[VAL_2:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>144// CHECK: %[[VAL_3:.*]] = hlfir.get_extent %[[VAL_2]] {dim = 0 : index} : (!fir.shape<2>) -> index145// CHECK: %[[VAL_5:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1>146// CHECK: %[[VAL_6:.*]] = hlfir.elemental %[[VAL_5]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xf32> {147// CHECK: ^bb0(%[[VAL_7:.*]]: index):148// CHECK: %[[VAL_10:.*]] = fir.do_loop %[[VAL_11:.*]] = %[[VAL_9]] to %[[VAL_3]] step %[[VAL_9]] iter_args(%[[VAL_12:.*]] = %[[VAL_8]]) -> (f32) {149// CHECK: %[[VAL_13:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_11]], %[[VAL_7]] : (!hlfir.expr<?x3xf32>, index, index) -> f32150// CHECK: %[[VAL_14:.*]] = arith.addf %[[VAL_12]], %[[VAL_13]] : f32151// CHECK: fir.result %[[VAL_14]] : f32152// CHECK: }153// CHECK: hlfir.yield_element %[[VAL_10]] : f32154// CHECK: }155// CHECK: return156// CHECK: }157 158func.func @sum_expr_unknown_extent2(%arg0: !hlfir.expr<?x3xf32>) -> !hlfir.expr<?xf32> {159 %cst = arith.constant 2 : i32160 %res = hlfir.sum %arg0 dim %cst : (!hlfir.expr<?x3xf32>, i32) -> !hlfir.expr<?xf32>161 return %res : !hlfir.expr<?xf32>162}163// CHECK-LABEL: func.func @sum_expr_unknown_extent2(164// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>) -> !hlfir.expr<?xf32> {165// CHECK: %[[VAL_9:.*]] = arith.constant 1 : index166// CHECK: %[[VAL_8:.*]] = arith.constant 0.000000e+00 : f32167// CHECK: %[[VAL_4:.*]] = arith.constant 3 : index168// CHECK: %[[VAL_2:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>169// CHECK: %[[VAL_3:.*]] = hlfir.get_extent %[[VAL_2]] {dim = 0 : index} : (!fir.shape<2>) -> index170// CHECK: %[[VAL_5:.*]] = fir.shape %[[VAL_3]] : (index) -> !fir.shape<1>171// CHECK: %[[VAL_6:.*]] = hlfir.elemental %[[VAL_5]] unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {172// CHECK: ^bb0(%[[VAL_7:.*]]: index):173// CHECK: %[[VAL_10:.*]] = fir.do_loop %[[VAL_11:.*]] = %[[VAL_9]] to %[[VAL_4]] step %[[VAL_9]] iter_args(%[[VAL_12:.*]] = %[[VAL_8]]) -> (f32) {174// CHECK: %[[VAL_13:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_7]], %[[VAL_11]] : (!hlfir.expr<?x3xf32>, index, index) -> f32175// CHECK: %[[VAL_14:.*]] = arith.addf %[[VAL_12]], %[[VAL_13]] : f32176// CHECK: fir.result %[[VAL_14]] : f32177// CHECK: }178// CHECK: hlfir.yield_element %[[VAL_10]] : f32179// CHECK: }180// CHECK: return181// CHECK: }182 183// scalar mask184func.func @sum_scalar_mask(%arg0: !hlfir.expr<?x3xf32>, %mask: !fir.ref<!fir.logical<1>>) -> !hlfir.expr<3xf32> {185 %cst = arith.constant 1 : i32186 %res = hlfir.sum %arg0 dim %cst mask %mask : (!hlfir.expr<?x3xf32>, i32, !fir.ref<!fir.logical<1>>) -> !hlfir.expr<3xf32>187 return %res : !hlfir.expr<3xf32>188}189// CHECK-LABEL: func.func @sum_scalar_mask(190// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>,191// CHECK-SAME: %[[VAL_1:.*]]: !fir.ref<!fir.logical<1>>) -> !hlfir.expr<3xf32> {192// CHECK: %[[VAL_11:.*]] = arith.constant 1 : index193// CHECK: %[[VAL_10:.*]] = arith.constant 0.000000e+00 : f32194// CHECK: %[[VAL_5:.*]] = arith.constant 3 : index195// CHECK: %[[VAL_3:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>196// CHECK: %[[VAL_4:.*]] = hlfir.get_extent %[[VAL_3]] {dim = 0 : index} : (!fir.shape<2>) -> index197// CHECK: %[[VAL_6:.*]] = fir.shape %[[VAL_5]] : (index) -> !fir.shape<1>198// CHECK: %[[VAL_7:.*]] = fir.load %[[VAL_1]] : !fir.ref<!fir.logical<1>>199// CHECK: %[[VAL_8:.*]] = hlfir.elemental %[[VAL_6]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xf32> {200// CHECK: ^bb0(%[[VAL_9:.*]]: index):201// CHECK: %[[VAL_12:.*]] = fir.do_loop %[[VAL_13:.*]] = %[[VAL_11]] to %[[VAL_4]] step %[[VAL_11]] iter_args(%[[VAL_14:.*]] = %[[VAL_10]]) -> (f32) {202// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_7]] : (!fir.logical<1>) -> i1203// CHECK: %[[VAL_16:.*]] = fir.if %[[VAL_15]] -> (f32) {204// CHECK: %[[VAL_17:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_13]], %[[VAL_9]] : (!hlfir.expr<?x3xf32>, index, index) -> f32205// CHECK: %[[VAL_18:.*]] = arith.addf %[[VAL_14]], %[[VAL_17]] : f32206// CHECK: fir.result %[[VAL_18]] : f32207// CHECK: } else {208// CHECK: fir.result %[[VAL_14]] : f32209// CHECK: }210// CHECK: fir.result %[[VAL_16]] : f32211// CHECK: }212// CHECK: hlfir.yield_element %[[VAL_12]] : f32213// CHECK: }214// CHECK: return215// CHECK: }216 217// scalar boxed mask218func.func @sum_scalar_boxed_mask(%arg0: !hlfir.expr<?x3xf32>, %mask: !fir.box<!fir.logical<1>>) -> !hlfir.expr<3xf32> {219 %cst = arith.constant 1 : i32220 %res = hlfir.sum %arg0 dim %cst mask %mask : (!hlfir.expr<?x3xf32>, i32, !fir.box<!fir.logical<1>>) -> !hlfir.expr<3xf32>221 return %res : !hlfir.expr<3xf32>222}223// CHECK-LABEL: func.func @sum_scalar_boxed_mask(224// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>,225// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.logical<1>>) -> !hlfir.expr<3xf32> {226// CHECK: %[[VAL_16:.*]] = arith.constant 1 : index227// CHECK: %[[VAL_15:.*]] = arith.constant 0.000000e+00 : f32228// CHECK: %[[VAL_11:.*]] = arith.constant true229// CHECK: %[[VAL_5:.*]] = arith.constant 3 : index230// CHECK: %[[VAL_3:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>231// CHECK: %[[VAL_4:.*]] = hlfir.get_extent %[[VAL_3]] {dim = 0 : index} : (!fir.shape<2>) -> index232// CHECK: %[[VAL_6:.*]] = fir.shape %[[VAL_5]] : (index) -> !fir.shape<1>233// CHECK: %[[VAL_7:.*]] = fir.is_present %[[VAL_1]] : (!fir.box<!fir.logical<1>>) -> i1234// CHECK: %[[VAL_8:.*]] = fir.if %[[VAL_7]] -> (!fir.logical<1>) {235// CHECK: %[[VAL_9:.*]] = fir.box_addr %[[VAL_1]] : (!fir.box<!fir.logical<1>>) -> !fir.ref<!fir.logical<1>>236// CHECK: %[[VAL_10:.*]] = fir.load %[[VAL_9]] : !fir.ref<!fir.logical<1>>237// CHECK: fir.result %[[VAL_10]] : !fir.logical<1>238// CHECK: } else {239// CHECK: %[[VAL_12:.*]] = fir.convert %[[VAL_11]] : (i1) -> !fir.logical<1>240// CHECK: fir.result %[[VAL_12]] : !fir.logical<1>241// CHECK: }242// CHECK: %[[VAL_13:.*]] = hlfir.elemental %[[VAL_6]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xf32> {243// CHECK: ^bb0(%[[VAL_14:.*]]: index):244// CHECK: %[[VAL_17:.*]] = fir.do_loop %[[VAL_18:.*]] = %[[VAL_16]] to %[[VAL_4]] step %[[VAL_16]] iter_args(%[[VAL_19:.*]] = %[[VAL_15]]) -> (f32) {245// CHECK: %[[VAL_20:.*]] = fir.convert %[[VAL_8]] : (!fir.logical<1>) -> i1246// CHECK: %[[VAL_21:.*]] = fir.if %[[VAL_20]] -> (f32) {247// CHECK: %[[VAL_22:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_18]], %[[VAL_14]] : (!hlfir.expr<?x3xf32>, index, index) -> f32248// CHECK: %[[VAL_23:.*]] = arith.addf %[[VAL_19]], %[[VAL_22]] : f32249// CHECK: fir.result %[[VAL_23]] : f32250// CHECK: } else {251// CHECK: fir.result %[[VAL_19]] : f32252// CHECK: }253// CHECK: fir.result %[[VAL_21]] : f32254// CHECK: }255// CHECK: hlfir.yield_element %[[VAL_17]] : f32256// CHECK: }257// CHECK: return258// CHECK: }259 260// array mask261func.func @sum_array_mask(%arg0: !hlfir.expr<?x3xf32>, %mask: !fir.box<!fir.array<?x3x!fir.logical<1>>>) -> !hlfir.expr<?xf32> {262 %cst = arith.constant 2 : i32263 %res = hlfir.sum %arg0 dim %cst mask %mask : (!hlfir.expr<?x3xf32>, i32, !fir.box<!fir.array<?x3x!fir.logical<1>>>) -> !hlfir.expr<?xf32>264 return %res : !hlfir.expr<?xf32>265}266// CHECK-LABEL: func.func @sum_array_mask(267// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>,268// CHECK-SAME: %[[VAL_1:.*]]: !fir.box<!fir.array<?x3x!fir.logical<1>>>) -> !hlfir.expr<?xf32> {269// CHECK: %[[VAL_27:.*]] = arith.constant true270// CHECK: %[[VAL_16:.*]] = arith.constant 0 : index271// CHECK: %[[VAL_11:.*]] = arith.constant 1 : index272// CHECK: %[[VAL_10:.*]] = arith.constant 0.000000e+00 : f32273// CHECK: %[[VAL_5:.*]] = arith.constant 3 : index274// CHECK: %[[VAL_3:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>275// CHECK: %[[VAL_4:.*]] = hlfir.get_extent %[[VAL_3]] {dim = 0 : index} : (!fir.shape<2>) -> index276// CHECK: %[[VAL_6:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1>277// CHECK: %[[VAL_7:.*]] = fir.is_present %[[VAL_1]] : (!fir.box<!fir.array<?x3x!fir.logical<1>>>) -> i1278// CHECK: %[[VAL_8:.*]] = hlfir.elemental %[[VAL_6]] unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {279// CHECK: ^bb0(%[[VAL_9:.*]]: index):280// CHECK: %[[VAL_12:.*]] = fir.do_loop %[[VAL_13:.*]] = %[[VAL_11]] to %[[VAL_5]] step %[[VAL_11]] iter_args(%[[VAL_14:.*]] = %[[VAL_10]]) -> (f32) {281// CHECK: %[[VAL_15:.*]] = fir.if %[[VAL_7]] -> (!fir.logical<1>) {282// CHECK: %[[VAL_17:.*]]:3 = fir.box_dims %[[VAL_1]], %[[VAL_16]] : (!fir.box<!fir.array<?x3x!fir.logical<1>>>, index) -> (index, index, index)283// CHECK: %[[VAL_19:.*]]:3 = fir.box_dims %[[VAL_1]], %[[VAL_11]] : (!fir.box<!fir.array<?x3x!fir.logical<1>>>, index) -> (index, index, index)284// CHECK: %[[VAL_21:.*]] = arith.subi %[[VAL_17]]#0, %[[VAL_11]] : index285// CHECK: %[[VAL_22:.*]] = arith.addi %[[VAL_9]], %[[VAL_21]] : index286// CHECK: %[[VAL_23:.*]] = arith.subi %[[VAL_19]]#0, %[[VAL_11]] : index287// CHECK: %[[VAL_24:.*]] = arith.addi %[[VAL_13]], %[[VAL_23]] : index288// CHECK: %[[VAL_25:.*]] = hlfir.designate %[[VAL_1]] (%[[VAL_22]], %[[VAL_24]]) : (!fir.box<!fir.array<?x3x!fir.logical<1>>>, index, index) -> !fir.ref<!fir.logical<1>>289// CHECK: %[[VAL_26:.*]] = fir.load %[[VAL_25]] : !fir.ref<!fir.logical<1>>290// CHECK: fir.result %[[VAL_26]] : !fir.logical<1>291// CHECK: } else {292// CHECK: %[[VAL_28:.*]] = fir.convert %[[VAL_27]] : (i1) -> !fir.logical<1>293// CHECK: fir.result %[[VAL_28]] : !fir.logical<1>294// CHECK: }295// CHECK: %[[VAL_29:.*]] = fir.convert %[[VAL_15]] : (!fir.logical<1>) -> i1296// CHECK: %[[VAL_30:.*]] = fir.if %[[VAL_29]] -> (f32) {297// CHECK: %[[VAL_31:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_9]], %[[VAL_13]] : (!hlfir.expr<?x3xf32>, index, index) -> f32298// CHECK: %[[VAL_32:.*]] = arith.addf %[[VAL_14]], %[[VAL_31]] : f32299// CHECK: fir.result %[[VAL_32]] : f32300// CHECK: } else {301// CHECK: fir.result %[[VAL_14]] : f32302// CHECK: }303// CHECK: fir.result %[[VAL_30]] : f32304// CHECK: }305// CHECK: hlfir.yield_element %[[VAL_12]] : f32306// CHECK: }307// CHECK: return308// CHECK: }309 310// array expr mask311func.func @sum_array_expr_mask(%arg0: !hlfir.expr<?x3xf32>, %mask: !hlfir.expr<?x3x!fir.logical<1>>) -> !hlfir.expr<?xf32> {312 %cst = arith.constant 2 : i32313 %res = hlfir.sum %arg0 dim %cst mask %mask : (!hlfir.expr<?x3xf32>, i32, !hlfir.expr<?x3x!fir.logical<1>>) -> !hlfir.expr<?xf32>314 return %res : !hlfir.expr<?xf32>315}316// CHECK-LABEL: func.func @sum_array_expr_mask(317// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<?x3xf32>,318// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr<?x3x!fir.logical<1>>) -> !hlfir.expr<?xf32> {319// CHECK: %[[VAL_10:.*]] = arith.constant 1 : index320// CHECK: %[[VAL_9:.*]] = arith.constant 0.000000e+00 : f32321// CHECK: %[[VAL_5:.*]] = arith.constant 3 : index322// CHECK: %[[VAL_3:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr<?x3xf32>) -> !fir.shape<2>323// CHECK: %[[VAL_4:.*]] = hlfir.get_extent %[[VAL_3]] {dim = 0 : index} : (!fir.shape<2>) -> index324// CHECK: %[[VAL_6:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1>325// CHECK: %[[VAL_7:.*]] = hlfir.elemental %[[VAL_6]] unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {326// CHECK: ^bb0(%[[VAL_8:.*]]: index):327// CHECK: %[[VAL_11:.*]] = fir.do_loop %[[VAL_12:.*]] = %[[VAL_10]] to %[[VAL_5]] step %[[VAL_10]] iter_args(%[[VAL_13:.*]] = %[[VAL_9]]) -> (f32) {328// CHECK: %[[VAL_14:.*]] = hlfir.apply %[[VAL_1]], %[[VAL_8]], %[[VAL_12]] : (!hlfir.expr<?x3x!fir.logical<1>>, index, index) -> !fir.logical<1>329// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_14]] : (!fir.logical<1>) -> i1330// CHECK: %[[VAL_16:.*]] = fir.if %[[VAL_15]] -> (f32) {331// CHECK: %[[VAL_17:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_8]], %[[VAL_12]] : (!hlfir.expr<?x3xf32>, index, index) -> f32332// CHECK: %[[VAL_18:.*]] = arith.addf %[[VAL_13]], %[[VAL_17]] : f32333// CHECK: fir.result %[[VAL_18]] : f32334// CHECK: } else {335// CHECK: fir.result %[[VAL_13]] : f32336// CHECK: }337// CHECK: fir.result %[[VAL_16]] : f32338// CHECK: }339// CHECK: hlfir.yield_element %[[VAL_11]] : f32340// CHECK: }341// CHECK: return342// CHECK: }343 344// unordered floating point reduction345func.func @sum_unordered_reduction(%arg0: !hlfir.expr<2x3xf32>) -> !hlfir.expr<3xf32> {346 %cst = arith.constant 1 : i32347 %res = hlfir.sum %arg0 dim %cst {fastmath = #arith.fastmath<reassoc>} : (!hlfir.expr<2x3xf32>, i32) -> !hlfir.expr<3xf32>348 return %res : !hlfir.expr<3xf32>349}350// CHECK-LABEL: func.func @sum_unordered_reduction(351// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr<2x3xf32>) -> !hlfir.expr<3xf32> {352// CHECK: %[[VAL_8:.*]] = arith.constant 1 : index353// CHECK: %[[VAL_7:.*]] = arith.constant 0.000000e+00 : f32354// CHECK: %[[VAL_2:.*]] = arith.constant 2 : index355// CHECK: %[[VAL_3:.*]] = arith.constant 3 : index356// CHECK: %[[VAL_4:.*]] = fir.shape %[[VAL_3]] : (index) -> !fir.shape<1>357// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] unordered : (!fir.shape<1>) -> !hlfir.expr<3xf32> {358// CHECK: ^bb0(%[[VAL_6:.*]]: index):359// CHECK: %[[VAL_9:.*]] = fir.do_loop %[[VAL_10:.*]] = %[[VAL_8]] to %[[VAL_2]] step %[[VAL_8]] unordered iter_args(%[[VAL_11:.*]] = %[[VAL_7]]) -> (f32) {360// CHECK: %[[VAL_12:.*]] = hlfir.apply %[[VAL_0]], %[[VAL_10]], %[[VAL_6]] : (!hlfir.expr<2x3xf32>, index, index) -> f32361// CHECK: %[[VAL_13:.*]] = arith.addf %[[VAL_11]], %[[VAL_12]] fastmath<reassoc> : f32362// CHECK: fir.result %[[VAL_13]] : f32363// CHECK: }364// CHECK: hlfir.yield_element %[[VAL_9]] : f32365// CHECK: }366// CHECK: return367// CHECK: }368 369// total 1d reduction370func.func @sum_total_1d_reduction(%arg0: !fir.box<!fir.array<3xi32>>) -> i32 {371 %cst = arith.constant 1 : i32372 %res = hlfir.sum %arg0 dim %cst : (!fir.box<!fir.array<3xi32>>, i32) -> i32373 return %res : i32374}375// CHECK-LABEL: func.func @sum_total_1d_reduction(376// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<3xi32>>) -> i32 {377// CHECK: %[[VAL_8:.*]] = arith.constant 0 : index378// CHECK: %[[VAL_2:.*]] = arith.constant 3 : index379// CHECK: %[[VAL_3:.*]] = arith.constant 0 : i32380// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index381// CHECK: %[[VAL_5:.*]] = fir.do_loop %[[VAL_6:.*]] = %[[VAL_4]] to %[[VAL_2]] step %[[VAL_4]] unordered iter_args(%[[VAL_7:.*]] = %[[VAL_3]]) -> (i32) {382// CHECK: %[[VAL_9:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_8]] : (!fir.box<!fir.array<3xi32>>, index) -> (index, index, index)383// CHECK: %[[VAL_11:.*]] = arith.subi %[[VAL_9]]#0, %[[VAL_4]] : index384// CHECK: %[[VAL_12:.*]] = arith.addi %[[VAL_6]], %[[VAL_11]] : index385// CHECK: %[[VAL_13:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_12]]) : (!fir.box<!fir.array<3xi32>>, index) -> !fir.ref<i32>386// CHECK: %[[VAL_14:.*]] = fir.load %[[VAL_13]] : !fir.ref<i32>387// CHECK: %[[VAL_15:.*]] = arith.addi %[[VAL_7]], %[[VAL_14]] : i32388// CHECK: fir.result %[[VAL_15]] : i32389// CHECK: }390// CHECK: return391// CHECK: }392 393// total 2d reduction394func.func @sum_total_2d_reduction(%arg0: !fir.box<!fir.array<?x3xi32>>) -> i32 {395 %res = hlfir.sum %arg0 : (!fir.box<!fir.array<?x3xi32>>) -> i32396 return %res : i32397}398// CHECK-LABEL: func.func @sum_total_2d_reduction(399// CHECK-SAME: %[[VAL_0:.*]]: !fir.box<!fir.array<?x3xi32>>) -> i32 {400// CHECK: %[[VAL_5:.*]] = arith.constant 1 : index401// CHECK: %[[VAL_4:.*]] = arith.constant 0 : i32402// CHECK: %[[VAL_3:.*]] = arith.constant 3 : index403// CHECK: %[[VAL_1:.*]] = arith.constant 0 : index404// CHECK: %[[VAL_2:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_1]] : (!fir.box<!fir.array<?x3xi32>>, index) -> (index, index, index)405// CHECK: %[[VAL_6:.*]] = fir.do_loop %[[VAL_7:.*]] = %[[VAL_5]] to %[[VAL_3]] step %[[VAL_5]] unordered iter_args(%[[VAL_8:.*]] = %[[VAL_4]]) -> (i32) {406// CHECK: %[[VAL_9:.*]] = fir.do_loop %[[VAL_10:.*]] = %[[VAL_5]] to %[[VAL_2]]#1 step %[[VAL_5]] unordered iter_args(%[[VAL_11:.*]] = %[[VAL_8]]) -> (i32) {407// CHECK: %[[VAL_13:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_1]] : (!fir.box<!fir.array<?x3xi32>>, index) -> (index, index, index)408// CHECK: %[[VAL_15:.*]]:3 = fir.box_dims %[[VAL_0]], %[[VAL_5]] : (!fir.box<!fir.array<?x3xi32>>, index) -> (index, index, index)409// CHECK: %[[VAL_17:.*]] = arith.subi %[[VAL_13]]#0, %[[VAL_5]] : index410// CHECK: %[[VAL_18:.*]] = arith.addi %[[VAL_10]], %[[VAL_17]] : index411// CHECK: %[[VAL_19:.*]] = arith.subi %[[VAL_15]]#0, %[[VAL_5]] : index412// CHECK: %[[VAL_20:.*]] = arith.addi %[[VAL_7]], %[[VAL_19]] : index413// CHECK: %[[VAL_21:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_18]], %[[VAL_20]]) : (!fir.box<!fir.array<?x3xi32>>, index, index) -> !fir.ref<i32>414// CHECK: %[[VAL_22:.*]] = fir.load %[[VAL_21]] : !fir.ref<i32>415// CHECK: %[[VAL_23:.*]] = arith.addi %[[VAL_11]], %[[VAL_22]] : i32416// CHECK: fir.result %[[VAL_23]] : i32417// CHECK: }418// CHECK: fir.result %[[VAL_9]] : i32419// CHECK: }420// CHECK: return421// CHECK: }422 423// negative: invalid dim==0424func.func @sum_invalid_dim0(%arg0: !hlfir.expr<2x3xi32>) -> !hlfir.expr<3xi32> {425 %cst = arith.constant 0 : i32426 %res = hlfir.sum %arg0 dim %cst : (!hlfir.expr<2x3xi32>, i32) -> !hlfir.expr<3xi32>427 return %res : !hlfir.expr<3xi32>428}429// CHECK-LABEL: func.func @sum_invalid_dim0(430// CHECK: hlfir.sum %{{.*}} dim %{{.*}} : (!hlfir.expr<2x3xi32>, i32) -> !hlfir.expr<3xi32>431 432// negative: invalid dim>rank433func.func @sum_invalid_dim_big(%arg0: !hlfir.expr<2x3xi32>) -> !hlfir.expr<3xi32> {434 %cst = arith.constant 3 : i32435 %res = hlfir.sum %arg0 dim %cst : (!hlfir.expr<2x3xi32>, i32) -> !hlfir.expr<3xi32>436 return %res : !hlfir.expr<3xi32>437}438// CHECK-LABEL: func.func @sum_invalid_dim_big(439// CHECK: hlfir.sum %{{.*}} dim %{{.*}} : (!hlfir.expr<2x3xi32>, i32) -> !hlfir.expr<3xi32>440