xref: /llvm-project/flang/test/Lower/OpenMP/wsloop-reduction-iand-byref.f90 (revision 937cbce14c9aa956342a9c818c26a8a557802843)
1! RUN: bbc -emit-hlfir -fopenmp --force-byref-reduction %s -o - | FileCheck %s
2! RUN: %flang_fc1 -emit-hlfir -fopenmp -mmlir --force-byref-reduction %s -o - | FileCheck %s
3
4! NOTE: Assertions have been autogenerated by utils/generate-test-checks.py
5
6! CHECK-LABEL:   omp.declare_reduction @iand_byref_i32 : !fir.ref<i32>
7! CHECK-SAME:    alloc {
8! CHECK:            %[[REF:.*]] = fir.alloca i32
9! CHECK:            omp.yield(%[[REF]] : !fir.ref<i32>)
10! CHECK-LABEL:   } init {
11! CHECK:         ^bb0(%[[VAL_0:.*]]: !fir.ref<i32>, %[[ALLOC:.*]]: !fir.ref<i32>):
12! CHECK:            %[[C0_1:.*]] = arith.constant -1 : i32
13! CHECK:            fir.store %[[C0_1]] to %[[ALLOC]] : !fir.ref<i32>
14! CHECK:            omp.yield(%[[ALLOC]] : !fir.ref<i32>)
15
16! CHECK-LABEL:   } combiner {
17! CHECK:         ^bb0(%[[ARG0:.*]]: !fir.ref<i32>, %[[ARG1:.*]]: !fir.ref<i32>):
18! CHECK:           %[[LD0:.*]] = fir.load %[[ARG0]] : !fir.ref<i32>
19! CHECK:           %[[LD1:.*]] = fir.load %[[ARG1]] : !fir.ref<i32>
20! CHECK:           %[[RES:.*]] = arith.andi %[[LD0]], %[[LD1]] : i32
21! CHECK:           fir.store %[[RES]] to %[[ARG0]] : !fir.ref<i32>
22! CHECK:           omp.yield(%[[ARG0]] : !fir.ref<i32>)
23! CHECK:         }
24
25! CHECK-LABEL:   func.func @_QPreduction_iand(
26! CHECK-SAME:                                 %[[VAL_0:.*]]: !fir.box<!fir.array<?xi32>> {fir.bindc_name = "y"}) {
27! CHECK:           %[[VAL_1:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFreduction_iandEi"}
28! CHECK:           %[[VAL_2:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "_QFreduction_iandEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
29! CHECK:           %[[VAL_3:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFreduction_iandEx"}
30! CHECK:           %[[VAL_4:.*]]:2 = hlfir.declare %[[VAL_3]] {uniq_name = "_QFreduction_iandEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
31! CHECK:           %[[VAL_5:.*]]:2 = hlfir.declare %[[VAL_0]] dummy_scope %{{[0-9]+}} {uniq_name = "_QFreduction_iandEy"} : (!fir.box<!fir.array<?xi32>>, !fir.dscope) -> (!fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<?xi32>>)
32! CHECK:           %[[VAL_6:.*]] = arith.constant 0 : i32
33! CHECK:           hlfir.assign %[[VAL_6]] to %[[VAL_4]]#0 : i32, !fir.ref<i32>
34! CHECK:           omp.parallel {
35! CHECK:             %[[VAL_7:.*]] = fir.alloca i32 {bindc_name = "i", pinned, {{.*}}}
36! CHECK:             %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_7]] {uniq_name = "_QFreduction_iandEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
37! CHECK:             %[[VAL_9:.*]] = arith.constant 1 : i32
38! CHECK:             %[[VAL_10:.*]] = arith.constant 100 : i32
39! CHECK:             %[[VAL_11:.*]] = arith.constant 1 : i32
40! CHECK:             omp.wsloop reduction(byref @iand_byref_i32 %[[VAL_4]]#0 -> %[[VAL_12:.*]] : !fir.ref<i32>) {
41! CHECK-NEXT:          omp.loop_nest (%[[VAL_13:.*]]) : i32 = (%[[VAL_9]]) to (%[[VAL_10]]) inclusive step (%[[VAL_11]]) {
42! CHECK:                 %[[VAL_14:.*]]:2 = hlfir.declare %[[VAL_12]] {uniq_name = "_QFreduction_iandEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
43! CHECK:                 fir.store %[[VAL_13]] to %[[VAL_8]]#1 : !fir.ref<i32>
44! CHECK:                 %[[VAL_15:.*]] = fir.load %[[VAL_8]]#0 : !fir.ref<i32>
45! CHECK:                 %[[VAL_16:.*]] = fir.convert %[[VAL_15]] : (i32) -> i64
46! CHECK:                 %[[VAL_17:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_16]])  : (!fir.box<!fir.array<?xi32>>, i64) -> !fir.ref<i32>
47! CHECK:                 %[[VAL_18:.*]] = fir.load %[[VAL_14]]#0 : !fir.ref<i32>
48! CHECK:                 %[[VAL_19:.*]] = fir.load %[[VAL_17]] : !fir.ref<i32>
49! CHECK:                 %[[VAL_20:.*]] = arith.andi %[[VAL_18]], %[[VAL_19]] : i32
50! CHECK:                 hlfir.assign %[[VAL_20]] to %[[VAL_14]]#0 : i32, !fir.ref<i32>
51! CHECK:                 omp.yield
52! CHECK:             omp.terminator
53
54
55
56subroutine reduction_iand(y)
57  integer :: x, y(:)
58  x = 0
59  !$omp parallel
60  !$omp do reduction(iand:x)
61  do i=1, 100
62    x = iand(x, y(i))
63  end do
64  !$omp end do
65  !$omp end parallel
66  print *, x
67end subroutine
68