xref: /llvm-project/flang/test/HLFIR/maxval.fir (revision 2318bc878a51f14c0935efebce152560e9c3071e) 
1// Test hlfir.maxval 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 @maxval0(%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.maxval %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
14  return
15}
16// CHECK:      func.func @maxval0(%[[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.maxval %[[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 @maxval1(%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.maxval %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
36  return
37}
38// CHECK:      func.func @maxval1(%[[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.maxval %[[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 @maxval2(%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.maxval %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 @maxval2(%[[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.maxval %[[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 @maxval3(%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.maxval %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 @maxval3(%[[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.maxval %[[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 @maxval4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<42x!fir.logical<4>>) {
95  %c_1 = arith.constant 1 : index
96  %0 = hlfir.maxval %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 @maxval4(%[[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.maxval %[[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 @maxval5(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<?x!fir.logical<4>>) {
107  %c_1 = arith.constant 1 : index
108  %0 = hlfir.maxval %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 @maxval5(%[[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.maxval %[[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 @maxval6(%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.maxval %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 @maxval6(%[[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.maxval %[[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 @maxval7(%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.maxval %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 @maxval7(%[[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.maxval %[[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 @maxval8(%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.maxval %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 @maxval8(%[[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.maxval %[[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 @maxval9(%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.maxval %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 @maxval9(%[[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.maxval %[[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.maxval with only an array argument
182func.func @maxval10(%arg0: !fir.box<!fir.array<?x?xi32>>) {
183  %maxval = hlfir.maxval %arg0 : (!fir.box<!fir.array<?x?xi32>>) -> i32
184  return
185}
186// CHECK:      func.func @maxval10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>
187// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] : (!fir.box<!fir.array<?x?xi32>>) -> i32
188// CHECK-NEXT:   return
189// CHECK-NEXT: }
190
191// hlfir.maxval with only a character array argument
192func.func @maxval11(%arg0: !fir.box<!fir.array<?x?x!fir.char<1,?>>>) {
193  %maxval = hlfir.maxval %arg0 : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<!fir.char<1,?>>
194  return
195}
196// CHECK:      func.func @maxval11(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.char<1,?>>>
197// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<!fir.char<1,?>>
198// CHECK-NEXT:   return
199// CHECK-NEXT: }
200
201// hlfir.maxval with array and dim argument
202func.func @maxval12(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
203  %maxval = hlfir.maxval %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
204  return
205}
206// CHECK:      func.func @maxval12(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32
207// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
208// CHECK-NEXT:   return
209// CHECK-NEXT: }
210
211// hlfir.maxval with array and mask argument
212func.func @maxval13(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.logical<4>) {
213  %maxval = hlfir.maxval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
214  return
215}
216// CHECK:      func.func @maxval13(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.logical<4>
217// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
218// CHECK-NEXT:   return
219// CHECK-NEXT: }
220
221// hlfir.maxval with dim argument with an unusual type
222func.func @maxval14(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: index) {
223  %maxval = hlfir.maxval %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
224  return
225}
226// CHECK:      func.func @maxval14(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: index
227// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
228// CHECK-NEXT:   return
229// CHECK-NEXT: }
230
231// hlfir.maxval with mask argument of unusual type
232func.func @maxval15(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
233  %maxval = hlfir.maxval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> i32
234  return
235}
236// CHECK:      func.func @maxval15(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
237// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> i32
238// CHECK-NEXT:   return
239// CHECK-NEXT: }
240
241// hlfir.maxval with mask argument of ref<array<>> type
242func.func @maxval16(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
243  %maxval = hlfir.maxval %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
244  return
245}
246// CHECK:      func.func @maxval16(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>
247// CHECK-NEXT:   %[[maxval:.*]] = hlfir.maxval %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
248// CHECK-NEXT:   return
249// CHECK-NEXT: }
250