xref: /llvm-project/flang/test/HLFIR/elemental.fir (revision 93fea7dd11477983a6903187fa6fec65be1ffe1b) 
1// Test hlfir.elemental and hlfir.yield_element operation parse, verify
2// (no errors), and unparse.
3
4// RUN: fir-opt %s | fir-opt | FileCheck %s
5
6func.func @numeric_type(%x: !fir.ref<!fir.array<10x20xi32>>, %y: !fir.ref<!fir.array<10x20xi32>>) {
7  %c10 = arith.constant 10 : index
8  %c20 = arith.constant 20 : index
9  %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 : i32
17    hlfir.yield_element %8 : i32
18  }
19  return
20}
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.shape
25// 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]] : i32
32// CHECK:    hlfir.yield_element %[[VAL_12]] : i32
33// 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  return
44}
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  return
68}
69// CHECK-LABEL:   func.func @parametrized_derived_transpose(
70// CHECK-SAME: %[[VAL_0:.*]]: !fir.box
71// CHECK-SAME: %[[VAL_3:[^:]*]]: i32
72// CHECK:  %[[VAL_4:.*]] = fir.shape
73// 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 : index
82  %c20 = arith.constant 20 : index
83  %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 : i32
87    hlfir.yield_element %c0 : i32
88  }
89  return
90}
91// CHECK-LABEL:   func.func @unordered() {
92// CHECK:           %[[VAL_0:.*]] = arith.constant 10 : index
93// CHECK:           %[[VAL_1:.*]] = arith.constant 20 : index
94// 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 : i32
98// CHECK:             hlfir.yield_element %[[VAL_6]] : i32
99// CHECK:           }
100// CHECK:           return
101// 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  return
111}
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:           return
121// 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  return
132}
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:           return
143// CHECK:         }
144