144 lines · plain
1// Test hlfir.elemental and hlfir.yield_element operation parse, verify2// (no errors), and unparse.3 4// RUN: fir-opt %s | fir-opt | FileCheck %s5 6func.func @numeric_type(%x: !fir.ref<!fir.array<10x20xi32>>, %y: !fir.ref<!fir.array<10x20xi32>>) {7 %c10 = arith.constant 10 : index8 %c20 = arith.constant 20 : index9 %0 = fir.shape %c10, %c20 : (index, index) -> !fir.shape<2>10 %3 = hlfir.elemental %0 : (!fir.shape<2>) -> !hlfir.expr<10x20xi32> {11 ^bb0(%i: index, %j: index):12 %4 = hlfir.designate %x (%i, %j) : (!fir.ref<!fir.array<10x20xi32>>, index, index) -> !fir.ref<i32>13 %5 = hlfir.designate %y (%i, %j) : (!fir.ref<!fir.array<10x20xi32>>, index, index) -> !fir.ref<i32>14 %6 = fir.load %4 : !fir.ref<i32>15 %7 = fir.load %5 : !fir.ref<i32>16 %8 = arith.addi %6, %7 : i3217 hlfir.yield_element %8 : i3218 }19 return20}21// CHECK-LABEL: func.func @numeric_type(22// CHECK-SAME: %[[VAL_0:[^:]*]]: !fir.ref<!fir.array<10x20xi32>>,23// CHECK-SAME: %[[VAL_1:[^:]*]]: !fir.ref<!fir.array<10x20xi32>>24// CHECK: %[[VAL_4:.*]] = fir.shape25// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] : (!fir.shape<2>) -> !hlfir.expr<10x20xi32> {26// CHECK: ^bb0(%[[VAL_6:.*]]: index, %[[VAL_7:.*]]: index):27// CHECK: %[[VAL_8:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_6]], %[[VAL_7]]) : (!fir.ref<!fir.array<10x20xi32>>, index, index) -> !fir.ref<i32>28// CHECK: %[[VAL_9:.*]] = hlfir.designate %[[VAL_1]] (%[[VAL_6]], %[[VAL_7]]) : (!fir.ref<!fir.array<10x20xi32>>, index, index) -> !fir.ref<i32>29// CHECK: %[[VAL_10:.*]] = fir.load %[[VAL_8]] : !fir.ref<i32>30// CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_9]] : !fir.ref<i32>31// CHECK: %[[VAL_12:.*]] = arith.addi %[[VAL_10]], %[[VAL_11]] : i3232// CHECK: hlfir.yield_element %[[VAL_12]] : i3233// CHECK: }34 35func.func @char_type(%x: !fir.box<!fir.array<?x!fir.char<1,?>>>, %n: index, %l : index, %l0 :index) {36 %0 = fir.shape %n : (index) -> !fir.shape<1>37 %3 = hlfir.elemental %0 typeparams %l : (!fir.shape<1>, index) -> !hlfir.expr<?x!fir.char<1,?>> {38 ^bb0(%i: index):39 %4 = hlfir.designate %x (%i) typeparams %l0 : (!fir.box<!fir.array<?x!fir.char<1,?>>>, index, index) -> !fir.boxchar<1>40 %6 = hlfir.concat %4, %4 len %l : (!fir.boxchar<1>, !fir.boxchar<1>, index) -> !hlfir.expr<!fir.char<1,?>>41 hlfir.yield_element %6 : !hlfir.expr<!fir.char<1,?>>42 }43 return44}45// CHECK-LABEL: func.func @char_type(46// CHECK-SAME: %[[VAL_0:[^:]*]]: !fir.box<!fir.array<?x!fir.char<1,?>>>,47// CHECK-SAME: %[[VAL_1:[^:]*]]: index,48// CHECK-SAME: %[[VAL_2:[^:]*]]: index,49// CHECK-SAME: %[[VAL_3:[^:]*]]: index) {50// CHECK: %[[VAL_4:.*]] = fir.shape %[[VAL_1]] : (index) -> !fir.shape<1>51// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] typeparams %[[VAL_2]] : (!fir.shape<1>, index) -> !hlfir.expr<?x!fir.char<1,?>> {52// CHECK: ^bb0(%[[VAL_6:.*]]: index):53// CHECK: %[[VAL_7:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_6]]) typeparams %[[VAL_3]] : (!fir.box<!fir.array<?x!fir.char<1,?>>>, index, index) -> !fir.boxchar<1>54// CHECK: %[[VAL_8:.*]] = hlfir.concat %[[VAL_7]], %[[VAL_7]] len %[[VAL_2]] : (!fir.boxchar<1>, !fir.boxchar<1>, index) -> !hlfir.expr<!fir.char<1,?>>55// CHECK: hlfir.yield_element %[[VAL_8]] : !hlfir.expr<!fir.char<1,?>>56// CHECK: }57 58!pdt = !fir.type<pdt(param:i32){field:f32}>59func.func @parametrized_derived_transpose(%x: !fir.box<!fir.array<?x?x!pdt>>, %n: index, %m: index, %l: i32) {60 %0 = fir.shape %m, %n : (index, index) -> !fir.shape<2>61 %3 = hlfir.elemental %0 typeparams %l : (!fir.shape<2>, i32) -> !hlfir.expr<?x?x!pdt> {62 ^bb0(%j: index, %i: index):63 %4 = hlfir.designate %x (%j, %i) typeparams %l : (!fir.box<!fir.array<?x?x!pdt>>, index, index, i32) -> !fir.box<!pdt>64 %5 = hlfir.as_expr %4 : (!fir.box<!pdt>) -> !hlfir.expr<!pdt>65 hlfir.yield_element %5 : !hlfir.expr<!pdt>66 }67 return68}69// CHECK-LABEL: func.func @parametrized_derived_transpose(70// CHECK-SAME: %[[VAL_0:.*]]: !fir.box71// CHECK-SAME: %[[VAL_3:[^:]*]]: i3272// CHECK: %[[VAL_4:.*]] = fir.shape73// CHECK: %[[VAL_5:.*]] = hlfir.elemental %[[VAL_4]] typeparams %[[VAL_3]] : (!fir.shape<2>, i32) -> !hlfir.expr<?x?x!fir.type<pdt(param:i32){field:f32}>> {74// CHECK: ^bb0(%[[VAL_6:.*]]: index, %[[VAL_7:.*]]: index):75// CHECK: %[[VAL_8:.*]] = hlfir.designate %[[VAL_0]] (%[[VAL_6]], %[[VAL_7]]) typeparams %[[VAL_3]] : (!fir.box<!fir.array<?x?x!fir.type<pdt(param:i32){field:f32}>>>, index, index, i32) -> !fir.box<!fir.type<pdt(param:i32){field:f32}>>76// CHECK: %[[VAL_9:.*]] = hlfir.as_expr %[[VAL_8]] : (!fir.box<!fir.type<pdt(param:i32){field:f32}>>) -> !hlfir.expr<!fir.type<pdt(param:i32){field:f32}>>77// CHECK: hlfir.yield_element %[[VAL_9]] : !hlfir.expr<!fir.type<pdt(param:i32){field:f32}>>78// CHECK: }79 80func.func @unordered() {81 %c10 = arith.constant 10 : index82 %c20 = arith.constant 20 : index83 %0 = fir.shape %c10, %c20 : (index, index) -> !fir.shape<2>84 %3 = hlfir.elemental %0 unordered : (!fir.shape<2>) -> !hlfir.expr<10x20xi32> {85 ^bb0(%i: index, %j: index):86 %c0 = arith.constant 0 : i3287 hlfir.yield_element %c0 : i3288 }89 return90}91// CHECK-LABEL: func.func @unordered() {92// CHECK: %[[VAL_0:.*]] = arith.constant 10 : index93// CHECK: %[[VAL_1:.*]] = arith.constant 20 : index94// CHECK: %[[VAL_2:.*]] = fir.shape %[[VAL_0]], %[[VAL_1]] : (index, index) -> !fir.shape<2>95// CHECK: %[[VAL_3:.*]] = hlfir.elemental %[[VAL_2]] unordered : (!fir.shape<2>) -> !hlfir.expr<10x20xi32> {96// CHECK: ^bb0(%[[VAL_4:.*]]: index, %[[VAL_5:.*]]: index):97// CHECK: %[[VAL_6:.*]] = arith.constant 0 : i3298// CHECK: hlfir.yield_element %[[VAL_6]] : i3299// CHECK: }100// CHECK: return101// CHECK: }102 103func.func @polymorphic_mold_var(%arg0: !fir.class<!fir.array<?x!fir.type<_QMtypesTt>>>, %shape : index) {104 %3 = fir.shape %shape : (index) -> !fir.shape<1>105 %4 = hlfir.elemental %3 mold %arg0 unordered : (!fir.shape<1>, !fir.class<!fir.array<?x!fir.type<_QMtypesTt>>>) -> !hlfir.expr<?x!fir.type<_QMtypesTt>?> {106 ^bb0(%arg2: index):107 %6 = fir.undefined !hlfir.expr<!fir.type<_QMtypesTt>?>108 hlfir.yield_element %6 : !hlfir.expr<!fir.type<_QMtypesTt>?>109 }110 return111}112// CHECK-LABEL: func.func @polymorphic_mold_var(113// CHECK-SAME: %[[VAL_0:.*]]: !fir.class<!fir.array<?x!fir.type<_QMtypesTt>>>, %[[VAL_1:.*]]: index) {114// CHECK: %[[VAL_2:.*]] = fir.shape %[[VAL_1]] : (index) -> !fir.shape<1>115// CHECK: %[[VAL_3:.*]] = hlfir.elemental %[[VAL_2]] mold %[[VAL_0]] unordered : (!fir.shape<1>, !fir.class<!fir.array<?x!fir.type<_QMtypesTt>>>) -> !hlfir.expr<?x!fir.type<_QMtypesTt>?> {116// CHECK: ^bb0(%[[VAL_4:.*]]: index):117// CHECK: %[[VAL_5:.*]] = fir.undefined !hlfir.expr<!fir.type<_QMtypesTt>?>118// CHECK: hlfir.yield_element %[[VAL_5]] : !hlfir.expr<!fir.type<_QMtypesTt>?>119// CHECK: }120// CHECK: return121// CHECK: }122 123func.func @polymorphic_mold_expr(%shape : index) {124 %3 = fir.shape %shape : (index) -> !fir.shape<1>125 %mold = fir.undefined !hlfir.expr<?x!fir.type<_QMtypesTt>?>126 %4 = hlfir.elemental %3 mold %mold unordered : (!fir.shape<1>, !hlfir.expr<?x!fir.type<_QMtypesTt>?>) -> !hlfir.expr<?x!fir.type<_QMtypesTt>?> {127 ^bb0(%arg2: index):128 %6 = fir.undefined !hlfir.expr<!fir.type<_QMtypesTt>?>129 hlfir.yield_element %6 : !hlfir.expr<!fir.type<_QMtypesTt>?>130 }131 return132}133// CHECK-LABEL: func.func @polymorphic_mold_expr(134// CHECK-SAME: %[[VAL_0:.*]]: index) {135// CHECK: %[[VAL_1:.*]] = fir.shape %[[VAL_0]] : (index) -> !fir.shape<1>136// CHECK: %[[VAL_2:.*]] = fir.undefined !hlfir.expr<?x!fir.type<_QMtypesTt>?>137// CHECK: %[[VAL_3:.*]] = hlfir.elemental %[[VAL_1]] mold %[[VAL_2]] unordered : (!fir.shape<1>, !hlfir.expr<?x!fir.type<_QMtypesTt>?>) -> !hlfir.expr<?x!fir.type<_QMtypesTt>?> {138// CHECK: ^bb0(%[[VAL_4:.*]]: index):139// CHECK: %[[VAL_5:.*]] = fir.undefined !hlfir.expr<!fir.type<_QMtypesTt>?>140// CHECK: hlfir.yield_element %[[VAL_5]] : !hlfir.expr<!fir.type<_QMtypesTt>?>141// CHECK: }142// CHECK: return143// CHECK: }144