Skip to content

Commit c84f525

Browse files
[Flang][OpenMP] Add new test to demonstrate reduction
1 parent 9e6aa2f commit c84f525

File tree

1 file changed

+38
-0
lines changed

1 file changed

+38
-0
lines changed
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
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: omp.reduction.declare @[[REDUCTION_DECLARE:[_a-z0-9]+]] : i32 init {
5+
!CHECK: ^bb0(%{{.*}}: i32):
6+
!CHECK: %[[I0:[_a-z0-9]+]] = arith.constant 0 : i32
7+
!CHECK: omp.yield(%[[I0]] : i32)
8+
!CHECK: } combiner {
9+
!CHECK: ^bb0(%[[C0:[_a-z0-9]+]]: i32, %[[C1:[_a-z0-9]+]]: i32):
10+
!CHECK: %[[CR:[_a-z0-9]+]] = arith.addi %[[C0]], %[[C1]] : i32
11+
!CHECK: omp.yield(%[[CR]] : i32)
12+
!CHECK: }
13+
!CHECK: func.func @_QQmain() attributes {fir.bindc_name = "mn"} {
14+
!CHECK: %[[RED_ACCUM_REF:[_a-z0-9]+]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
15+
!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>)
16+
!CHECK: %[[C0:[_a-z0-9]+]] = arith.constant 0 : i32
17+
!CHECK: hlfir.assign %[[C0]] to %[[RED_ACCUM_DECL]]#0 : i32, !fir.ref<i32>
18+
!CHECK: omp.parallel reduction(@[[REDUCTION_DECLARE]] %[[RED_ACCUM_DECL]]#0 -> %[[PRIVATE_RED:[a-z0-9]+]] : !fir.ref<i32>) {
19+
!CHECK: %[[PRIVATE_DECL:[_a-z0-9]+]]:2 = hlfir.declare %[[PRIVATE_RED]] {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
20+
!CHECK: %[[C1:[_a-z0-9]+]] = arith.constant 1 : i32
21+
!CHECK: hlfir.assign %[[C1]] to %[[PRIVATE_DECL]]#0 : i32, !fir.ref<i32>
22+
!CHECK: omp.terminator
23+
!CHECK: }
24+
!CHECK: %[[RED_ACCUM_VAL:[_a-z0-9]+]] = fir.load %[[RED_ACCUM_DECL]]#0 : !fir.ref<i32>
25+
!CHECK: {{.*}} = fir.call @_FortranAioOutputInteger32(%{{.*}}, %[[RED_ACCUM_VAL]]) fastmath<contract> : (!fir.ref<i8>, i32) -> i1
26+
!CHECK: return
27+
!CHECK: }
28+
29+
program mn
30+
integer :: i
31+
i = 0
32+
33+
!$omp parallel reduction(+:i)
34+
i = 1
35+
!$omp end parallel
36+
37+
print *, i
38+
end program

0 commit comments

Comments
 (0)