Skip to content

Commit 0d4c96c

Browse files
vzakharivzakhari
committed
Added LIT test.
1 parent 41f8ec7 commit 0d4c96c

File tree

1 file changed

+254
-0
lines changed

1 file changed

+254
-0
lines changed

‎flang/test/HLFIR/order_assignments/where-after-cse.fir‎

Lines changed: 254 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,254 @@
1+
// Test canonicalization of hlfir.exactly_once operations
2+
// after CSE. The live-in values that are not dominating
3+
// the top-level hlfir.where must be cloned inside hlfir.exactly_once,
4+
// otherwise, the cloning of the hlfir.exactly_once before hlfir.where
5+
// would cause def-use issues:
6+
// RUN: fir-opt %s --lower-hlfir-ordered-assignments | FileCheck %s
7+
8+
// Simple case, where CSE makes only hlfir.designate live-in:
9+
// CHECK-LABEL: func.func @_QPtest1(
10+
func.func @_QPtest1(%arg0: !fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>> {fir.bindc_name = "x"}) {
11+
%true = arith.constant true
12+
%cst = arith.constant 0.000000e+00 : f32
13+
%c1 = arith.constant 1 : index
14+
%c0 = arith.constant 0 : index
15+
%0 = fir.dummy_scope : !fir.dscope
16+
%1:2 = hlfir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<intent_inout>, uniq_name = "_QFtest1Ex"} : (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>, !fir.dscope) -> (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>, !fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>)
17+
hlfir.where {
18+
%2 = hlfir.designate %1#0{"p2"} {fortran_attrs = #fir.var_attrs<pointer>} : (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
19+
%3 = fir.load %2 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
20+
%4:3 = fir.box_dims %3, %c0 : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, index) -> (index, index, index)
21+
%5 = arith.addi %4#0, %4#1 : index
22+
%6 = arith.subi %5, %c1 : index
23+
%7 = arith.subi %6, %4#0 : index
24+
%8 = arith.addi %7, %c1 : index
25+
%9 = arith.cmpi sgt, %8, %c0 : index
26+
%10 = arith.select %9, %8, %c0 : index
27+
%11 = fir.shape %10 : (index) -> !fir.shape<1>
28+
%12 = hlfir.designate %3 (%4#0:%6:%c1) shape %11 : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
29+
%13 = hlfir.elemental %11 unordered : (!fir.shape<1>) -> !hlfir.expr<?x!fir.logical<4>> {
30+
^bb0(%arg1: index):
31+
%14 = hlfir.designate %12 (%arg1) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
32+
%15 = fir.load %14 : !fir.ref<f32>
33+
%16 = arith.cmpf ogt, %15, %cst fastmath<contract> : f32
34+
%17 = fir.convert %16 : (i1) -> !fir.logical<4>
35+
hlfir.yield_element %17 : !fir.logical<4>
36+
}
37+
hlfir.yield %13 : !hlfir.expr<?x!fir.logical<4>> cleanup {
38+
hlfir.destroy %13 : !hlfir.expr<?x!fir.logical<4>>
39+
}
40+
} do {
41+
hlfir.region_assign {
42+
%2 = hlfir.designate %1#0{"p1"} {fortran_attrs = #fir.var_attrs<pointer>} : (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>
43+
%3 = fir.load %2 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>
44+
%4:3 = fir.box_dims %3, %c0 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index) -> (index, index, index)
45+
%5 = arith.addi %4#0, %4#1 : index
46+
%6 = arith.subi %5, %c1 : index
47+
%7 = arith.subi %6, %4#0 : index
48+
%8 = arith.addi %7, %c1 : index
49+
%9 = arith.cmpi sgt, %8, %c0 : index
50+
%10 = arith.select %9, %8, %c0 : index
51+
%11 = fir.shape %10 : (index) -> !fir.shape<1>
52+
%12 = hlfir.designate %3 (%4#0:%6:%c1, %c1) shape %11 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
53+
%13 = hlfir.exactly_once : !hlfir.expr<?xf32> {
54+
// CHECK: %[[VAL_26:.*]] = hlfir.designate %{{.*}}#0{"p1"} {fortran_attrs = #fir.var_attrs<pointer>} : (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>
55+
// CHECK: fir.load %[[VAL_26]] : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>
56+
// CHECK: %[[VAL_47:.*]] = fir.call @_QPcallee(%{{.*}}) fastmath<contract> : (!fir.box<!fir.array<?xf32>>) -> !fir.array<?xf32>
57+
// CHECK: fir.do_loop
58+
%15 = fir.load %2 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>
59+
%16:3 = fir.box_dims %15, %c0 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index) -> (index, index, index)
60+
%17 = arith.addi %16#0, %16#1 : index
61+
%18 = arith.subi %17, %c1 : index
62+
%19 = arith.subi %18, %16#0 : index
63+
%20 = arith.addi %19, %c1 : index
64+
%21 = arith.cmpi sgt, %20, %c0 : index
65+
%22 = arith.select %21, %20, %c0 : index
66+
%23 = fir.shape %22 : (index) -> !fir.shape<1>
67+
%24 = hlfir.designate %15 (%16#0:%18:%c1, %c1) shape %23 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
68+
%25:2 = hlfir.declare %24 {fortran_attrs = #fir.var_attrs<intent_in>, uniq_name = "_QMmy_moduleFcalleeEx"} : (!fir.box<!fir.array<?xf32>>) -> (!fir.box<!fir.array<?xf32>>, !fir.box<!fir.array<?xf32>>)
69+
%26:3 = fir.box_dims %25#0, %c0 : (!fir.box<!fir.array<?xf32>>, index) -> (index, index, index)
70+
%27 = fir.convert %26#1 : (index) -> i64
71+
%28 = fir.convert %27 : (i64) -> index
72+
%29 = arith.cmpi sgt, %28, %c0 : index
73+
%30 = arith.select %29, %28, %c0 : index
74+
%31 = fir.shape %30 : (index) -> !fir.shape<1>
75+
%32 = fir.allocmem !fir.array<?xf32>, %30 {bindc_name = ".tmp.expr_result", uniq_name = ""}
76+
%33 = fir.convert %32 : (!fir.heap<!fir.array<?xf32>>) -> !fir.ref<!fir.array<?xf32>>
77+
%34:2 = hlfir.declare %33(%31) {uniq_name = ".tmp.expr_result"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>)
78+
%35 = fir.call @_QPcallee(%24) fastmath<contract> : (!fir.box<!fir.array<?xf32>>) -> !fir.array<?xf32>
79+
fir.save_result %35 to %34#1(%31) : !fir.array<?xf32>, !fir.ref<!fir.array<?xf32>>, !fir.shape<1>
80+
%36 = hlfir.as_expr %34#0 move %true : (!fir.box<!fir.array<?xf32>>, i1) -> !hlfir.expr<?xf32>
81+
hlfir.yield %36 : !hlfir.expr<?xf32> cleanup {
82+
hlfir.destroy %36 : !hlfir.expr<?xf32>
83+
}
84+
}
85+
%14 = hlfir.elemental %11 unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {
86+
^bb0(%arg1: index):
87+
%15 = hlfir.designate %12 (%arg1) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
88+
%16 = hlfir.apply %13, %arg1 : (!hlfir.expr<?xf32>, index) -> f32
89+
%17 = fir.load %15 : !fir.ref<f32>
90+
%18 = arith.divf %17, %16 fastmath<contract> : f32
91+
hlfir.yield_element %18 : f32
92+
}
93+
hlfir.yield %14 : !hlfir.expr<?xf32> cleanup {
94+
hlfir.destroy %14 : !hlfir.expr<?xf32>
95+
}
96+
} to {
97+
%2 = hlfir.designate %1#0{"p2"} {fortran_attrs = #fir.var_attrs<pointer>} : (!fir.ref<!fir.type<_QMmy_moduleTtt{p1:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>,p2:!fir.box<!fir.ptr<!fir.array<?xf32>>>}>>) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
98+
%3 = fir.load %2 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
99+
%4:3 = fir.box_dims %3, %c0 : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, index) -> (index, index, index)
100+
%5 = arith.addi %4#0, %4#1 : index
101+
%6 = arith.subi %5, %c1 : index
102+
%7 = arith.subi %6, %4#0 : index
103+
%8 = arith.addi %7, %c1 : index
104+
%9 = arith.cmpi sgt, %8, %c0 : index
105+
%10 = arith.select %9, %8, %c0 : index
106+
%11 = fir.shape %10 : (index) -> !fir.shape<1>
107+
%12 = hlfir.designate %3 (%4#0:%6:%c1) shape %11 : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
108+
hlfir.yield %12 : !fir.box<!fir.array<?xf32>>
109+
}
110+
}
111+
return
112+
}
113+
114+
// CSE makes a chain of operations live-in:
115+
// CHECK-LABEL: func.func @_QPtest_where_in_forall(
116+
func.func @_QPtest_where_in_forall(%arg0: !fir.box<!fir.array<?x?xf32>> {fir.bindc_name = "a"}, %arg1: !fir.box<!fir.array<?x?xf32>> {fir.bindc_name = "b"}, %arg2: !fir.box<!fir.array<?x?xf32>> {fir.bindc_name = "c"}) {
117+
%false = arith.constant false
118+
%c1_i32 = arith.constant 1 : i32
119+
%c10_i32 = arith.constant 10 : i32
120+
%c0 = arith.constant 0 : index
121+
%c1 = arith.constant 1 : index
122+
%c2_i32 = arith.constant 2 : i32
123+
%c100 = arith.constant 100 : index
124+
%0 = fir.alloca !fir.array<100x!fir.logical<4>> {bindc_name = ".tmp.expr_result"}
125+
%1 = fir.alloca !fir.array<100x!fir.logical<4>> {bindc_name = ".tmp.expr_result"}
126+
%2 = fir.dummy_scope : !fir.dscope
127+
// CHECK: %[[VAL_21:.*]]:2 = hlfir.declare %{{.*}} dummy_scope %{{.*}} {uniq_name = "_QFtest_where_in_forallEb"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> (!fir.box<!fir.array<?x?xf32>>, !fir.box<!fir.array<?x?xf32>>)
128+
%3:2 = hlfir.declare %arg0 dummy_scope %2 {uniq_name = "_QFtest_where_in_forallEa"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> (!fir.box<!fir.array<?x?xf32>>, !fir.box<!fir.array<?x?xf32>>)
129+
%4:2 = hlfir.declare %arg1 dummy_scope %2 {uniq_name = "_QFtest_where_in_forallEb"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> (!fir.box<!fir.array<?x?xf32>>, !fir.box<!fir.array<?x?xf32>>)
130+
%5:2 = hlfir.declare %arg2 dummy_scope %2 {uniq_name = "_QFtest_where_in_forallEc"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> (!fir.box<!fir.array<?x?xf32>>, !fir.box<!fir.array<?x?xf32>>)
131+
hlfir.forall lb {
132+
hlfir.yield %c1_i32 : i32
133+
} ub {
134+
hlfir.yield %c10_i32 : i32
135+
} (%arg3: i32) {
136+
hlfir.where {
137+
%6 = fir.shape %c100 : (index) -> !fir.shape<1>
138+
%7:2 = hlfir.declare %0(%6) {uniq_name = ".tmp.expr_result"} : (!fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.shape<1>) -> (!fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.ref<!fir.array<100x!fir.logical<4>>>)
139+
%8 = fir.call @_QPpure_logical_func1() proc_attrs<pure> fastmath<contract> : () -> !fir.array<100x!fir.logical<4>>
140+
fir.save_result %8 to %7#1(%6) : !fir.array<100x!fir.logical<4>>, !fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.shape<1>
141+
%9 = hlfir.as_expr %7#0 move %false : (!fir.ref<!fir.array<100x!fir.logical<4>>>, i1) -> !hlfir.expr<100x!fir.logical<4>>
142+
hlfir.yield %9 : !hlfir.expr<100x!fir.logical<4>> cleanup {
143+
hlfir.destroy %9 : !hlfir.expr<100x!fir.logical<4>>
144+
}
145+
} do {
146+
hlfir.region_assign {
147+
%6 = fir.convert %arg3 : (i32) -> i64
148+
// CHECK: %[[VAL_58:.*]]:3 = fir.box_dims %[[VAL_21]]#1, %{{.*}} : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
149+
// CHECK: %[[VAL_59:.*]] = arith.cmpi sgt, %[[VAL_58]]#1, %{{.*}} : index
150+
// CHECK: %[[VAL_60:.*]] = arith.select %[[VAL_59]], %[[VAL_58]]#1, %{{.*}} : index
151+
// CHECK: %[[VAL_61:.*]] = fir.shape %[[VAL_60]] : (index) -> !fir.shape<1>
152+
// CHECK: %[[VAL_62:.*]] = hlfir.designate %[[VAL_21]]#0 (%{{.*}}, %{{.*}}:%[[VAL_58]]#1:%{{.*}}) shape %[[VAL_61]] : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
153+
%7:3 = fir.box_dims %4#1, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
154+
%8 = arith.cmpi sgt, %7#1, %c0 : index
155+
%9 = arith.select %8, %7#1, %c0 : index
156+
%10 = fir.shape %9 : (index) -> !fir.shape<1>
157+
%11 = hlfir.designate %4#0 (%6, %c1:%7#1:%c1) shape %10 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
158+
%12 = hlfir.exactly_once : f32 {
159+
%19:3 = fir.box_dims %3#1, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
160+
%20 = arith.cmpi sgt, %19#1, %c0 : index
161+
%21 = arith.select %20, %19#1, %c0 : index
162+
%22 = fir.shape %21 : (index) -> !fir.shape<1>
163+
%23 = hlfir.designate %3#0 (%6, %c1:%19#1:%c1) shape %22 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
164+
// CHECK: %[[VAL_68:.*]] = fir.call @_QPpure_real_func2() fastmath<contract> : () -> f32
165+
// CHECK: %[[VAL_69:.*]] = hlfir.elemental %{{.*}} unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {
166+
// CHECK: ^bb0(%[[VAL_70:.*]]: index):
167+
// CHECK: %[[VAL_72:.*]] = hlfir.designate %[[VAL_62]] (%[[VAL_70]]) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
168+
%24 = fir.call @_QPpure_real_func2() fastmath<contract> : () -> f32
169+
%25 = hlfir.elemental %22 unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {
170+
^bb0(%arg4: index):
171+
%28 = hlfir.designate %23 (%arg4) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
172+
%29 = hlfir.designate %11 (%arg4) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
173+
%30 = fir.load %28 : !fir.ref<f32>
174+
%31 = fir.load %29 : !fir.ref<f32>
175+
%32 = arith.addf %30, %31 fastmath<contract> : f32
176+
%33 = arith.addf %32, %24 fastmath<contract> : f32
177+
hlfir.yield_element %33 : f32
178+
}
179+
%26:3 = hlfir.associate %25(%22) {adapt.valuebyref} : (!hlfir.expr<?xf32>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>, i1)
180+
%27 = fir.call @_QPpure_real_func(%26#1) fastmath<contract> : (!fir.ref<!fir.array<?xf32>>) -> f32
181+
hlfir.yield %27 : f32 cleanup {
182+
hlfir.end_associate %26#1, %26#2 : !fir.ref<!fir.array<?xf32>>, i1
183+
hlfir.destroy %25 : !hlfir.expr<?xf32>
184+
}
185+
}
186+
%13:3 = fir.box_dims %3#1, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
187+
%14 = arith.cmpi sgt, %13#1, %c0 : index
188+
%15 = arith.select %14, %13#1, %c0 : index
189+
%16 = fir.shape %15 : (index) -> !fir.shape<1>
190+
%17 = hlfir.designate %3#0 (%6, %c1:%13#1:%c1) shape %16 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
191+
%18 = hlfir.elemental %10 unordered : (!fir.shape<1>) -> !hlfir.expr<?xf32> {
192+
^bb0(%arg4: index):
193+
%19 = hlfir.designate %11 (%arg4) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
194+
%20 = fir.load %19 : !fir.ref<f32>
195+
%21 = arith.addf %20, %12 fastmath<contract> : f32
196+
%22 = hlfir.designate %17 (%arg4) : (!fir.box<!fir.array<?xf32>>, index) -> !fir.ref<f32>
197+
%23 = fir.call @_QPpure_elem_func(%22) proc_attrs<elemental, pure> fastmath<contract> : (!fir.ref<f32>) -> f32
198+
%24 = arith.addf %21, %23 fastmath<contract> : f32
199+
hlfir.yield_element %24 : f32
200+
}
201+
hlfir.yield %18 : !hlfir.expr<?xf32> cleanup {
202+
hlfir.destroy %18 : !hlfir.expr<?xf32>
203+
}
204+
} to {
205+
%6 = arith.muli %arg3, %c2_i32 overflow<nsw> : i32
206+
%7 = fir.convert %6 : (i32) -> i64
207+
%8:3 = fir.box_dims %3#1, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
208+
%9 = arith.cmpi sgt, %8#1, %c0 : index
209+
%10 = arith.select %9, %8#1, %c0 : index
210+
%11 = fir.shape %10 : (index) -> !fir.shape<1>
211+
%12 = hlfir.designate %3#0 (%7, %c1:%8#1:%c1) shape %11 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
212+
hlfir.yield %12 : !fir.box<!fir.array<?xf32>>
213+
}
214+
hlfir.elsewhere mask {
215+
%6 = hlfir.exactly_once : !hlfir.expr<100x!fir.logical<4>> {
216+
%7 = fir.shape %c100 : (index) -> !fir.shape<1>
217+
%8:2 = hlfir.declare %1(%7) {uniq_name = ".tmp.expr_result"} : (!fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.shape<1>) -> (!fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.ref<!fir.array<100x!fir.logical<4>>>)
218+
%9 = fir.call @_QPpure_logical_func2() proc_attrs<pure> fastmath<contract> : () -> !fir.array<100x!fir.logical<4>>
219+
fir.save_result %9 to %8#1(%7) : !fir.array<100x!fir.logical<4>>, !fir.ref<!fir.array<100x!fir.logical<4>>>, !fir.shape<1>
220+
%10 = hlfir.as_expr %8#0 move %false : (!fir.ref<!fir.array<100x!fir.logical<4>>>, i1) -> !hlfir.expr<100x!fir.logical<4>>
221+
hlfir.yield %10 : !hlfir.expr<100x!fir.logical<4>> cleanup {
222+
hlfir.destroy %10 : !hlfir.expr<100x!fir.logical<4>>
223+
}
224+
}
225+
hlfir.yield %6 : !hlfir.expr<100x!fir.logical<4>>
226+
} do {
227+
hlfir.region_assign {
228+
%6 = fir.convert %arg3 : (i32) -> i64
229+
%7:3 = fir.box_dims %5#1, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
230+
%8 = arith.cmpi sgt, %7#1, %c0 : index
231+
%9 = arith.select %8, %7#1, %c0 : index
232+
%10 = fir.shape %9 : (index) -> !fir.shape<1>
233+
%11 = hlfir.designate %5#0 (%6, %c1:%7#1:%c1) shape %10 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
234+
hlfir.yield %11 : !fir.box<!fir.array<?xf32>>
235+
} to {
236+
%6 = arith.muli %arg3, %c2_i32 overflow<nsw> : i32
237+
%7 = fir.convert %6 : (i32) -> i64
238+
%8 = hlfir.exactly_once : i32 {
239+
%14 = fir.call @_QPpure_ifoo() proc_attrs<pure> fastmath<contract> : () -> i32
240+
hlfir.yield %14 : i32 cleanup {
241+
}
242+
}
243+
%9 = fir.convert %8 : (i32) -> index
244+
%10 = arith.cmpi sgt, %9, %c0 : index
245+
%11 = arith.select %10, %9, %c0 : index
246+
%12 = fir.shape %11 : (index) -> !fir.shape<1>
247+
%13 = hlfir.designate %3#0 (%7, %c1:%9:%c1) shape %12 : (!fir.box<!fir.array<?x?xf32>>, i64, index, index, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>
248+
hlfir.yield %13 : !fir.box<!fir.array<?xf32>>
249+
}
250+
}
251+
}
252+
}
253+
return
254+
}

0 commit comments

Comments
 (0)