xref: /llvm-project/flang/test/HLFIR/sum.fir (revision aa9679304011170905a37758c6ce5c0c6e7a6265) 
1// Test hlfir.sum operation parse, verify (no errors), and unparse
2
3// RUN: fir-opt %s | fir-opt | FileCheck %s
4
5// array is an expression of known shape
6func.func @sum0(%arg0: !hlfir.expr<42xi32>) {
7  %mask = fir.alloca !fir.logical<4>
8  %c_1 = arith.constant 1 : index
9  %true = arith.constant true
10  %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.sum %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
14  return
15}
16// CHECK:      func.func @sum0(%[[ARRAY:.*]]: !hlfir.expr<42xi32>) {
17// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
18// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
19// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
20// 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.sum %[[ARRAY]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
24// CHECK-NEXT:   return
25// CHECK-NEXT: }
26
27// array is an expression of assumed shape
28func.func @sum1(%arg0: !hlfir.expr<?xi32>) {
29  %mask = fir.alloca !fir.logical<4>
30  %c_1 = arith.constant 1 : index
31  %true = arith.constant true
32  %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.sum %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
36  return
37}
38// CHECK:      func.func @sum1(%[[ARRAY:.*]]: !hlfir.expr<?xi32>) {
39// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
40// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
41// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
42// 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.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
46// CHECK-NEXT:   return
47// CHECK-NEXT: }
48
49// boxed array
50func.func @sum2(%arg0: !fir.box<!fir.array<42xi32>>) {
51  %mask = fir.alloca !fir.logical<4>
52  %c_1 = arith.constant 1 : index
53  %true = arith.constant true
54  %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.sum %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
58  return
59}
60// CHECK:      func.func @sum2(%[[ARRAY:.*]]: !fir.box<!fir.array<42xi32>>) {
61// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
62// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
63// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
64// 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.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
68// CHECK-NEXT:   return
69// CHECK-NEXT: }
70
71// assumed shape boxed array
72func.func @sum3(%arg0: !fir.box<!fir.array<?xi32>>) {
73  %mask = fir.alloca !fir.logical<4>
74  %c_1 = arith.constant 1 : index
75  %true = arith.constant true
76  %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.sum %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
80  return
81}
82// CHECK:      func.func @sum3(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>) {
83// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
84// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
85// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
86// 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.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
90// CHECK-NEXT:   return
91// CHECK-NEXT: }
92
93// known shape expr mask
94func.func @sum4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<42x!fir.logical<4>>) {
95  %c_1 = arith.constant 1 : index
96  %0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
97  return
98}
99// CHECK:      func.func @sum4(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<42x!fir.logical<4>>) {
100// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
101// CHECK-NEXT:   hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
102// CHECK-NEXT:   return
103// CHECK-NEXT: }
104
105// assumed shape expr mask
106func.func @sum5(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<?x!fir.logical<4>>) {
107  %c_1 = arith.constant 1 : index
108  %0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
109  return
110}
111// CHECK:      func.func @sum5(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<?x!fir.logical<4>>) {
112// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
113// CHECK-NEXT:   hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
114// CHECK-NEXT:   return
115// CHECK-NEXT: }
116
117// known shape array mask
118func.func @sum6(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<42x!fir.logical<4>>>) {
119  %c_1 = arith.constant 1 : index
120  %0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
121  return
122}
123// CHECK:      func.func @sum6(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<42x!fir.logical<4>>>) {
124// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
125// CHECK-NEXT:   hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
126// CHECK-NEXT:   return
127// CHECK-NEXT: }
128
129// assumed shape array mask
130func.func @sum7(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<?x!fir.logical<4>>>) {
131  %c_1 = arith.constant 1 : index
132  %0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
133  return
134}
135// CHECK:      func.func @sum7(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {
136// CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
137// CHECK-NEXT:   hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
138// CHECK-NEXT:   return
139// CHECK-NEXT: }
140
141// known shape expr return
142func.func @sum8(%arg0: !fir.box<!fir.array<2x2xi32>>, %arg1: i32) {
143  %mask = fir.alloca !fir.logical<4>
144  %true = arith.constant true
145  %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.sum %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
149  return
150}
151// CHECK:      func.func @sum8(%[[ARRAY:.*]]: !fir.box<!fir.array<2x2xi32>>, %[[DIM:.*]]: i32) {
152// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
153// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
154// 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.sum %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
158// CHECK-NEXT:   return
159// CHECK-NEXT: }
160
161// assumed shape expr return
162func.func @sum9(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
163  %mask = fir.alloca !fir.logical<4>
164  %true = arith.constant true
165  %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.sum %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
169  return
170}
171// CHECK:      func.func @sum9(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32) {
172// CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
173// CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
174// 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.sum %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
178// CHECK-NEXT:   return
179// CHECK-NEXT: }
180
181// hlfir.sum with only an array argument
182func.func @sum10(%arg0: !fir.box<!fir.array<?x?xi32>>) {
183  %sum = hlfir.sum %arg0 : (!fir.box<!fir.array<?x?xi32>>) -> i32
184  return
185}
186// CHECK:      func.func @sum10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>
187// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] : (!fir.box<!fir.array<?x?xi32>>) -> i32
188// CHECK-NEXT:   return
189// CHECK-NEXT: }
190
191// hlfir.sum with array and dim argument
192func.func @sum11(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
193  %sum = hlfir.sum %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
194  return
195}
196// CHECK:      func.func @sum11(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32
197// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
198// CHECK-NEXT:   return
199// CHECK-NEXT: }
200
201// hlfir.sum with array and mask argument
202func.func @sum12(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.logical<4>) {
203  %sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
204  return
205}
206// CHECK:      func.func @sum12(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.logical<4>
207// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
208// CHECK-NEXT:   return
209// CHECK-NEXT: }
210
211// hlfir.sum with dim argument with an unusual type
212func.func @sum13(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: index) {
213  %sum = hlfir.sum %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
214  return
215}
216// CHECK:      func.func @sum13(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: index
217// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
218// CHECK-NEXT:   return
219// CHECK-NEXT: }
220
221// hlfir.sum with mask argument of unusual type
222func.func @sum14(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
223  %sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> i32
224  return
225}
226// CHECK:      func.func @sum14(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
227// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> i32
228// CHECK-NEXT:   return
229// CHECK-NEXT: }
230
231// hlfir.sum with mask argument of ref<array<>> type
232func.func @sum15(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
233  %sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
234  return
235}
236// CHECK:      func.func @sum15(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>
237// CHECK-NEXT:   %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
238// CHECK-NEXT:   return
239// CHECK-NEXT: }
240