xref: /llvm-project/flang/test/Lower/OpenMP/parallel-reduction.f90 (revision d84252e064b3f35aa879c10e207f77e931f351d9)
19ecf4d20SDavid Truby! RUN: bbc -emit-hlfir -fopenmp -o - %s 2>&1 | FileCheck %s
29ecf4d20SDavid Truby! RUN: %flang_fc1 -emit-hlfir -fopenmp -o - %s 2>&1 | FileCheck %s
39ecf4d20SDavid Truby
4*d84252e0SSergio Afonso!CHECK:  omp.declare_reduction @[[REDUCTION_DECLARE:[_a-z0-9]+]] : i32 init {
59ecf4d20SDavid Truby!CHECK:  ^bb0(%{{.*}}: i32):
69ecf4d20SDavid Truby!CHECK:    %[[I0:[_a-z0-9]+]] = arith.constant 0 : i32
79ecf4d20SDavid Truby!CHECK:    omp.yield(%[[I0]] : i32)
89ecf4d20SDavid Truby!CHECK:  } combiner {
99ecf4d20SDavid Truby!CHECK:  ^bb0(%[[C0:[_a-z0-9]+]]: i32, %[[C1:[_a-z0-9]+]]: i32):
109ecf4d20SDavid Truby!CHECK:    %[[CR:[_a-z0-9]+]] = arith.addi %[[C0]], %[[C1]] : i32
119ecf4d20SDavid Truby!CHECK:    omp.yield(%[[CR]] : i32)
129ecf4d20SDavid Truby!CHECK:  }
139ecf4d20SDavid Truby!CHECK:  func.func @_QQmain() attributes {fir.bindc_name = "mn"} {
149ecf4d20SDavid Truby!CHECK:    %[[RED_ACCUM_REF:[_a-z0-9]+]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
159ecf4d20SDavid Truby!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>)
169ecf4d20SDavid Truby!CHECK:    %[[C0:[_a-z0-9]+]] = arith.constant 0 : i32
179ecf4d20SDavid Truby!CHECK:    hlfir.assign %[[C0]] to %[[RED_ACCUM_DECL]]#0 : i32, !fir.ref<i32>
189ecf4d20SDavid Truby!CHECK:    omp.parallel reduction(@[[REDUCTION_DECLARE]] %[[RED_ACCUM_DECL]]#0 -> %[[PRIVATE_RED:[a-z0-9]+]] : !fir.ref<i32>) {
199ecf4d20SDavid Truby!CHECK:      %[[PRIVATE_DECL:[_a-z0-9]+]]:2 = hlfir.declare %[[PRIVATE_RED]] {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
209ecf4d20SDavid Truby!CHECK:      %[[C1:[_a-z0-9]+]] = arith.constant 1 : i32
219ecf4d20SDavid Truby!CHECK:      hlfir.assign %[[C1]] to %[[PRIVATE_DECL]]#0 : i32, !fir.ref<i32>
229ecf4d20SDavid Truby!CHECK:      omp.terminator
239ecf4d20SDavid Truby!CHECK:    }
249ecf4d20SDavid Truby!CHECK:    %[[RED_ACCUM_VAL:[_a-z0-9]+]] = fir.load %[[RED_ACCUM_DECL]]#0 : !fir.ref<i32>
259ecf4d20SDavid Truby!CHECK:    {{.*}} = fir.call @_FortranAioOutputInteger32(%{{.*}}, %[[RED_ACCUM_VAL]]) fastmath<contract> : (!fir.ref<i8>, i32) -> i1
269ecf4d20SDavid Truby!CHECK:    return
279ecf4d20SDavid Truby!CHECK:  }
289ecf4d20SDavid Truby
299ecf4d20SDavid Trubyprogram mn
309ecf4d20SDavid Truby    integer :: i
319ecf4d20SDavid Truby    i = 0
329ecf4d20SDavid Truby
339ecf4d20SDavid Truby    !$omp parallel reduction(+:i)
349ecf4d20SDavid Truby      i = 1
359ecf4d20SDavid Truby    !$omp end parallel
369ecf4d20SDavid Truby
379ecf4d20SDavid Truby    print *, i
389ecf4d20SDavid Trubyend program
39