xref: /llvm-project/flang/test/Lower/OpenMP/parallel-reduction-byref.f90 (revision f2027a9388728094d84837fc0fdd2e0325362e51)
1! RUN: bbc -emit-hlfir -fopenmp --force-byref-reduction -o - %s 2>&1 | FileCheck %s
2! RUN: %flang_fc1 -emit-hlfir -fopenmp -mmlir --force-byref-reduction -o - %s 2>&1 | FileCheck %s
3
4!CHECK:  omp.declare_reduction @[[REDUCTION_DECLARE:[_a-z0-9]+]] : !fir.ref<i32>
5!CHECK-SAME: alloc {
6!CHECK:    %[[REF:.*]] = fir.alloca i32
7!CHECK:    omp.yield(%[[REF]] : !fir.ref<i32>)
8!CHECK-LABEL: } init {
9!CHECK:  ^bb0(%{{.*}}: !fir.ref<i32>, %[[ALLOC:.*]]: !fir.ref<i32>):
10!CHECK:    %[[I0:[_a-z0-9]+]] = arith.constant 0 : i32
11!CHECKL    fir.store [[%I0]] to %[[ALLOC]] : !fir.ref<i32>
12!CHECK:    omp.yield(%[[ALLOC]] : !fir.ref<i32>)
13!CHECK:  } combiner {
14!CHECK:  ^bb0(%[[C0:[_a-z0-9]+]]: !fir.ref<i32>, %[[C1:[_a-z0-9]+]]: !fir.ref<i32>):
15!CHECK:    %[[LD0:.*]] = fir.load %[[C0]] : !fir.ref<i32>
16!CHECK:    %[[LD1:.*]] = fir.load %[[C1]] : !fir.ref<i32>
17!CHECK:    %[[CR:[_a-z0-9]+]] = arith.addi %[[LD0]], %[[LD1]] : i32
18!CHECK:    fir.store %[[CR]] to %[[C0]] : !fir.ref<i32>
19!CHECK:    omp.yield(%[[C0]] : !fir.ref<i32>)
20!CHECK:  }
21!CHECK:  func.func @_QQmain() attributes {fir.bindc_name = "mn"} {
22!CHECK:    %[[RED_ACCUM_REF:[_a-z0-9]+]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
23!CHECK:    %[[RED_ACCUM_DECL:[_a-z0-9]+]]:2 = hlfir.declare %[[RED_ACCUM_REF]] {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
24!CHECK:    %[[C0:[_a-z0-9]+]] = arith.constant 0 : i32
25!CHECK:    hlfir.assign %[[C0]] to %[[RED_ACCUM_DECL]]#0 : i32, !fir.ref<i32>
26!CHECK:    omp.parallel reduction(byref @[[REDUCTION_DECLARE]] %[[RED_ACCUM_DECL]]#0 -> %[[PRIVATE_RED:[a-z0-9]+]] : !fir.ref<i32>) {
27!CHECK:      %[[PRIVATE_DECL:[_a-z0-9]+]]:2 = hlfir.declare %[[PRIVATE_RED]] {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
28!CHECK:      %[[C1:[_a-z0-9]+]] = arith.constant 1 : i32
29!CHECK:      hlfir.assign %[[C1]] to %[[PRIVATE_DECL]]#0 : i32, !fir.ref<i32>
30!CHECK:      omp.terminator
31!CHECK:    }
32!CHECK:    %[[RED_ACCUM_VAL:[_a-z0-9]+]] = fir.load %[[RED_ACCUM_DECL]]#0 : !fir.ref<i32>
33!CHECK:    {{.*}} = fir.call @_FortranAioOutputInteger32(%{{.*}}, %[[RED_ACCUM_VAL]]) fastmath<contract> : (!fir.ref<i8>, i32) -> i1
34!CHECK:    return
35!CHECK:  }
36
37program mn
38    integer :: i
39    i = 0
40
41    !$omp parallel reduction(+:i)
42      i = 1
43    !$omp end parallel
44
45    print *, i
46end program
47