xref: /llvm-project/flang/test/HLFIR/convert-assign-inside-openacc-recipe.fir (revision 86b44f3760b05df7cd13110a1bca3d9170552185) 
1// Check that hlfir.assign codegen is able to insert fir.alloca's inside
2// the regions of the OpenACC recipe.
3// RUN: fir-opt %s -convert-hlfir-to-fir | FileCheck %s
4
5acc.reduction.recipe @reduction_add_box_heap_Uxi32 : !fir.box<!fir.heap<!fir.array<?xi32>>> reduction_operator <add> init {
6^bb0(%arg0: !fir.box<!fir.heap<!fir.array<?xi32>>>):
7  %c0_i32 = arith.constant 0 : i32
8  %c0 = arith.constant 0 : index
9  %0:3 = fir.box_dims %arg0, %c0 : (!fir.box<!fir.heap<!fir.array<?xi32>>>, index) -> (index, index, index)
10  %1 = fir.shape %0#1 : (index) -> !fir.shape<1>
11  %2 = fir.allocmem !fir.array<?xi32>, %0#1 {bindc_name = ".tmp", uniq_name = ""}
12  %3:2 = hlfir.declare %2(%1) {uniq_name = ".tmp"} : (!fir.heap<!fir.array<?xi32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xi32>>, !fir.heap<!fir.array<?xi32>>)
13  hlfir.assign %c0_i32 to %3#0 : i32, !fir.box<!fir.array<?xi32>>
14  acc.yield %3#0 : !fir.box<!fir.array<?xi32>>
15} combiner {
16^bb0(%arg0: !fir.box<!fir.heap<!fir.array<?xi32>>>, %arg1: !fir.box<!fir.heap<!fir.array<?xi32>>>, %arg2: index, %arg3: index, %arg4: index):
17  %c1 = arith.constant 1 : index
18  %c0 = arith.constant 0 : index
19  %0 = arith.subi %arg3, %arg2 : index
20  %1 = arith.addi %0, %c1 : index
21  %2 = arith.divsi %1, %arg4 : index
22  %3 = arith.cmpi sgt, %2, %c0 : index
23  %4 = arith.select %3, %2, %c0 : index
24  %5 = fir.shape %4 : (index) -> !fir.shape<1>
25  %6 = hlfir.designate %arg0 (%arg2:%arg3:%arg4)  shape %5 : (!fir.box<!fir.heap<!fir.array<?xi32>>>, index, index, index, !fir.shape<1>) -> !fir.box<!fir.heap<!fir.array<?xi32>>>
26  %7 = hlfir.designate %arg1 (%arg2:%arg3:%arg4)  shape %5 : (!fir.box<!fir.heap<!fir.array<?xi32>>>, index, index, index, !fir.shape<1>) -> !fir.box<!fir.heap<!fir.array<?xi32>>>
27  %8 = fir.allocmem !fir.array<?xi32>, %4 {bindc_name = ".tmp.array", uniq_name = ""}
28  %9:2 = hlfir.declare %8(%5) {uniq_name = ".tmp.array"} : (!fir.heap<!fir.array<?xi32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xi32>>, !fir.heap<!fir.array<?xi32>>)
29  %true = arith.constant true
30  %c1_0 = arith.constant 1 : index
31  fir.do_loop %arg5 = %c1_0 to %4 step %c1_0 unordered {
32    %13 = hlfir.designate %6 (%arg5)  : (!fir.box<!fir.heap<!fir.array<?xi32>>>, index) -> !fir.ref<i32>
33    %14 = hlfir.designate %7 (%arg5)  : (!fir.box<!fir.heap<!fir.array<?xi32>>>, index) -> !fir.ref<i32>
34    %15 = fir.load %13 : !fir.ref<i32>
35    %16 = fir.load %14 : !fir.ref<i32>
36    %17 = arith.addi %15, %16 : i32
37    %18 = hlfir.designate %9#0 (%arg5)  : (!fir.box<!fir.array<?xi32>>, index) -> !fir.ref<i32>
38    hlfir.assign %17 to %18 temporary_lhs : i32, !fir.ref<i32>
39  }
40  %10 = fir.undefined tuple<!fir.box<!fir.array<?xi32>>, i1>
41  %11 = fir.insert_value %10, %true, [1 : index] : (tuple<!fir.box<!fir.array<?xi32>>, i1>, i1) -> tuple<!fir.box<!fir.array<?xi32>>, i1>
42  %12 = fir.insert_value %11, %9#0, [0 : index] : (tuple<!fir.box<!fir.array<?xi32>>, i1>, !fir.box<!fir.array<?xi32>>) -> tuple<!fir.box<!fir.array<?xi32>>, i1>
43  hlfir.assign %9#0 to %arg0 : !fir.box<!fir.array<?xi32>>, !fir.box<!fir.heap<!fir.array<?xi32>>>
44  acc.yield %arg0 : !fir.box<!fir.heap<!fir.array<?xi32>>>
45}
46// CHECK-LABEL:   acc.reduction.recipe @reduction_add_box_heap_Uxi32 : !fir.box<!fir.heap<!fir.array<?xi32>>> reduction_operator <add> init {
47// CHECK:         ^bb0(%[[VAL_0:.*]]: !fir.box<!fir.heap<!fir.array<?xi32>>>):
48// CHECK:           %[[VAL_1:.*]] = fir.alloca !fir.box<!fir.array<?xi32>>
49// CHECK-LABEL:   } combiner {
50// CHECK:         ^bb0(%[[VAL_0:.*]]: !fir.box<!fir.heap<!fir.array<?xi32>>>, %[[VAL_1:.*]]: !fir.box<!fir.heap<!fir.array<?xi32>>>, %[[VAL_2:.*]]: index, %[[VAL_3:.*]]: index, %[[VAL_4:.*]]: index):
51// CHECK:           %[[VAL_5:.*]] = fir.alloca !fir.box<!fir.heap<!fir.array<?xi32>>>
52