xref: /llvm-project/flang/test/Lower/OpenMP/parallel-reduction-add.f90 (revision cdb3ebf1e62df060767863e1e683409d6077ca6e)
1! RUN: bbc -emit-hlfir -fopenmp -o - %s 2>&1 | FileCheck %s
2! RUN: %flang_fc1 -emit-hlfir -fopenmp -o - %s 2>&1 | FileCheck %s
3
4!CHECK-LABEL: omp.declare_reduction
5!CHECK-SAME: @[[RED_F32_NAME:.*]] : f32 init {
6!CHECK: ^bb0(%{{.*}}: f32):
7!CHECK:  %[[C0_1:.*]] = arith.constant 0.000000e+00 : f32
8!CHECK:  omp.yield(%[[C0_1]] : f32)
9!CHECK: } combiner {
10!CHECK: ^bb0(%[[ARG0:.*]]: f32, %[[ARG1:.*]]: f32):
11!CHECK:  %[[RES:.*]] = arith.addf %[[ARG0]], %[[ARG1]] {{.*}}: f32
12!CHECK:  omp.yield(%[[RES]] : f32)
13!CHECK: }
14
15!CHECK-LABEL: omp.declare_reduction
16!CHECK-SAME: @[[RED_I32_NAME:.*]] : i32 init {
17!CHECK: ^bb0(%{{.*}}: i32):
18!CHECK:  %[[C0_1:.*]] = arith.constant 0 : i32
19!CHECK:  omp.yield(%[[C0_1]] : i32)
20!CHECK: } combiner {
21!CHECK: ^bb0(%[[ARG0:.*]]: i32, %[[ARG1:.*]]: i32):
22!CHECK:  %[[RES:.*]] = arith.addi %[[ARG0]], %[[ARG1]] : i32
23!CHECK:  omp.yield(%[[RES]] : i32)
24!CHECK: }
25
26!CHECK-LABEL: func.func @_QPsimple_int_add
27!CHECK:  %[[IREF:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFsimple_int_addEi"}
28!CHECK:  %[[I_DECL:.*]]:2 = hlfir.declare %[[IREF]] {uniq_name = "_QFsimple_int_addEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
29!CHECK:  %[[I_START:.*]] = arith.constant 0 : i32
30!CHECK:  hlfir.assign %[[I_START]] to %[[I_DECL]]#0 : i32, !fir.ref<i32>
31!CHECK:  omp.parallel reduction(@[[RED_I32_NAME]] %[[I_DECL]]#0 -> %[[PRV:.+]] : !fir.ref<i32>) {
32!CHECK:    %[[P_DECL:.+]]:2 = hlfir.declare %[[PRV]] {{.*}} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
33!CHECK:    %[[LPRV:.+]] = fir.load %[[P_DECL]]#0 : !fir.ref<i32>
34!CHECK:    %[[I_INCR:.*]] = arith.constant 1 : i32
35!CHECK:    %[[RES:.+]] = arith.addi %[[LPRV]], %[[I_INCR]] : i32
36!CHECK:    hlfir.assign %[[RES]] to %[[P_DECL]]#0 : i32, !fir.ref<i32>
37!CHECK:    omp.terminator
38!CHECK:  }
39!CHECK: return
40subroutine simple_int_add
41    integer :: i
42    i = 0
43
44    !$omp parallel reduction(+:i)
45    i = i + 1
46    !$omp end parallel
47
48    print *, i
49end subroutine
50
51!CHECK-LABEL: func.func @_QPsimple_real_add
52!CHECK:  %[[RREF:.*]] = fir.alloca f32 {bindc_name = "r", uniq_name = "_QFsimple_real_addEr"}
53!CHECK:  %[[R_DECL:.*]]:2 = hlfir.declare %[[RREF]] {uniq_name = "_QFsimple_real_addEr"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
54!CHECK:  %[[R_START:.*]] = arith.constant 0.000000e+00 : f32
55!CHECK:  hlfir.assign %[[R_START]] to %[[R_DECL]]#0 : f32, !fir.ref<f32>
56!CHECK:  omp.parallel reduction(@[[RED_F32_NAME]] %[[R_DECL]]#0 -> %[[PRV:.+]] : !fir.ref<f32>) {
57!CHECK:    %[[P_DECL:.+]]:2 = hlfir.declare %[[PRV]] {{.*}} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
58!CHECK:    %[[LPRV:.+]] = fir.load %[[P_DECL]]#0 : !fir.ref<f32>
59!CHECK:    %[[R_INCR:.*]] = arith.constant 1.500000e+00 : f32
60!CHECK:    %[[RES:.+]] = arith.addf %[[LPRV]], %[[R_INCR]] {{.*}} : f32
61!CHECK:    hlfir.assign %[[RES]] to %[[P_DECL]]#0 : f32, !fir.ref<f32>
62!CHECK:    omp.terminator
63!CHECK:  }
64!CHECK: return
65subroutine simple_real_add
66    real :: r
67    r = 0.0
68
69    !$omp parallel reduction(+:r)
70    r = r + 1.5
71    !$omp end parallel
72
73    print *, r
74end subroutine
75
76!CHECK-LABEL: func.func @_QPint_real_add
77!CHECK:  %[[IREF:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFint_real_addEi"}
78!CHECK:  %[[I_DECL:.*]]:2 = hlfir.declare %[[IREF]] {uniq_name = "_QFint_real_addEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
79!CHECK:  %[[RREF:.*]] = fir.alloca f32 {bindc_name = "r", uniq_name = "_QFint_real_addEr"}
80!CHECK:  %[[R_DECL:.*]]:2 = hlfir.declare %[[RREF]] {uniq_name = "_QFint_real_addEr"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
81!CHECK:  %[[R_START:.*]] = arith.constant 0.000000e+00 : f32
82!CHECK:  hlfir.assign %[[R_START]] to %[[R_DECL]]#0 : f32, !fir.ref<f32>
83!CHECK:  %[[I_START:.*]] = arith.constant 0 : i32
84!CHECK:  hlfir.assign %[[I_START]] to %[[I_DECL]]#0 : i32, !fir.ref<i32>
85!CHECK:  omp.parallel reduction(@[[RED_I32_NAME]] %[[I_DECL]]#0 -> %[[IPRV:.+]], @[[RED_F32_NAME]] %[[R_DECL]]#0 -> %[[RPRV:.+]] : !fir.ref<i32>, !fir.ref<f32>) {
86!CHECK:    %[[IP_DECL:.+]]:2 = hlfir.declare %[[IPRV]] {{.*}} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
87!CHECK:    %[[RP_DECL:.+]]:2 = hlfir.declare %[[RPRV]] {{.*}} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
88!CHECK:    %[[R_INCR:.*]] = arith.constant 1.500000e+00 : f32
89!CHECK:    %[[R_LPRV:.+]] = fir.load %[[RP_DECL]]#0 : !fir.ref<f32>
90!CHECK:    %[[RES1:.+]] = arith.addf %[[R_INCR]], %[[R_LPRV]] {{.*}} : f32
91!CHECK:    hlfir.assign %[[RES1]] to %[[RP_DECL]]#0 : f32, !fir.ref<f32>
92!CHECK:    %[[I_LPRV:.+]] = fir.load %[[IP_DECL]]#0 : !fir.ref<i32>
93!CHECK:    %[[I_INCR:.*]] = arith.constant 3 : i32
94!CHECK:    %[[RES0:.+]] = arith.addi %[[I_LPRV]], %[[I_INCR]] : i32
95!CHECK:    hlfir.assign %[[RES0]] to %[[IP_DECL]]#0 : i32, !fir.ref<i32>
96!CHECK:    omp.terminator
97!CHECK:  }
98!CHECK: return
99subroutine int_real_add
100    real :: r
101    integer :: i
102
103    r = 0.0
104    i = 0
105
106    !$omp parallel reduction(+:i,r)
107    r = 1.5 + r
108    i = i + 3
109    !$omp end parallel
110
111    print *, r
112    print *, i
113end subroutine
114