Skip to content

Commit 3bfc5eb

Browse files
kiranchandramohankiranchandramohan
committed
[Flang][OpenMP] NFC: Port a few parallel tests to HLFIR flow
1 parent d88d983 commit 3bfc5eb

File tree

4 files changed

+320
-0
lines changed

4 files changed

+320
-0
lines changed

‎flang/test/Lower/OpenMP/parallel-private-clause-fixes.f90

Lines changed: 92 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,92 @@
1+
! This test checks a few bug fixes in the PRIVATE clause lowering
2+
3+
! RUN: bbc -fopenmp -emit-hlfir %s -o - | FileCheck %s
4+
5+
! CHECK-LABEL: multiple_private_fix
6+
! CHECK-SAME: %[[GAMA:.*]]: !fir.ref<i32> {fir.bindc_name = "gama"}
7+
! CHECK: %[[GAMA_DECL:.*]]:2 = hlfir.declare %[[GAMA]] {uniq_name = "_QFmultiple_private_fixEgama"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
8+
! CHECK: %[[VAL_0:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFmultiple_private_fixEi"}
9+
! CHECK: %[[I_DECL:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "_QFmultiple_private_fixEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
10+
! CHECK: %[[VAL_1:.*]] = fir.alloca i32 {bindc_name = "j", uniq_name = "_QFmultiple_private_fixEj"}
11+
! CHECK: %[[J_DECL:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "_QFmultiple_private_fixEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
12+
! CHECK: %[[VAL_2:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFmultiple_private_fixEx"}
13+
! CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[VAL_2]] {uniq_name = "_QFmultiple_private_fixEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
14+
! CHECK: omp.parallel {
15+
! CHECK: %[[PRIV_J:.*]] = fir.alloca i32 {bindc_name = "j", pinned
16+
! CHECK: %[[PRIV_J_DECL:.*]]:2 = hlfir.declare %[[PRIV_J]] {uniq_name = "_QFmultiple_private_fixEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
17+
! CHECK: %[[PRIV_I:.*]] = fir.alloca i32 {adapt.valuebyref, pinned
18+
! CHECK: %[[PRIV_I_DECL:.*]]:2 = hlfir.declare %[[PRIV_I]] {uniq_name = "_QFmultiple_private_fixEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
19+
! CHECK: %[[PRIV_X:.*]] = fir.alloca i32 {bindc_name = "x", pinned
20+
! CHECK: %[[PRIV_X_DECL:.*]]:2 = hlfir.declare %[[PRIV_X]] {uniq_name = "_QFmultiple_private_fixEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
21+
! CHECK: %[[ONE:.*]] = arith.constant 1 : i32
22+
! CHECK: %[[VAL_3:.*]] = fir.load %[[GAMA_DECL]]#0 : !fir.ref<i32>
23+
! CHECK: %[[VAL_5:.*]] = arith.constant 1 : i32
24+
! CHECK: omp.wsloop for (%[[VAL_6:.*]]) : i32 = (%[[ONE]]) to (%[[VAL_3]]) inclusive step (%[[VAL_5]]) {
25+
! CHECK: fir.store %[[VAL_6]] to %[[PRIV_I_DECL]]#1 : !fir.ref<i32>
26+
! CHECK: %[[VAL_7:.*]] = arith.constant 1 : i32
27+
! CHECK: %[[VAL_8:.*]] = fir.convert %[[VAL_7]] : (i32) -> index
28+
! CHECK: %[[VAL_9:.*]] = fir.load %[[GAMA_DECL]]#0 : !fir.ref<i32>
29+
! CHECK: %[[VAL_10:.*]] = fir.convert %[[VAL_9]] : (i32) -> index
30+
! CHECK: %[[VAL_11:.*]] = arith.constant 1 : index
31+
! CHECK: %[[LB:.*]] = fir.convert %[[VAL_8]] : (index) -> i32
32+
! CHECK: %[[VAL_12:.*]]:2 = fir.do_loop %[[VAL_13:[^ ]*]] =
33+
! CHECK-SAME: %[[VAL_8]] to %[[VAL_10]] step %[[VAL_11]]
34+
! CHECK-SAME: iter_args(%[[IV:.*]] = %[[LB]]) -> (index, i32) {
35+
! CHECK: fir.store %[[IV]] to %[[PRIV_J_DECL]]#1 : !fir.ref<i32>
36+
! CHECK: %[[LOAD:.*]] = fir.load %[[PRIV_I_DECL]]#0 : !fir.ref<i32>
37+
! CHECK: %[[VAL_15:.*]] = fir.load %[[PRIV_J_DECL]]#0 : !fir.ref<i32>
38+
! CHECK: %[[VAL_16:.*]] = arith.addi %[[LOAD]], %[[VAL_15]] : i32
39+
! CHECK: hlfir.assign %[[VAL_16]] to %[[PRIV_X_DECL]]#0 : i32, !fir.ref<i32>
40+
! CHECK: %[[VAL_17:.*]] = arith.addi %[[VAL_13]], %[[VAL_11]] : index
41+
! CHECK: %[[STEPCAST:.*]] = fir.convert %[[VAL_11]] : (index) -> i32
42+
! CHECK: %[[IVLOAD:.*]] = fir.load %[[PRIV_J_DECL]]#1 : !fir.ref<i32>
43+
! CHECK: %[[IVINC:.*]] = arith.addi %[[IVLOAD]], %[[STEPCAST]]
44+
! CHECK: fir.result %[[VAL_17]], %[[IVINC]] : index, i32
45+
! CHECK: }
46+
! CHECK: fir.store %[[VAL_12]]#1 to %[[PRIV_J_DECL]]#1 : !fir.ref<i32>
47+
! CHECK: omp.yield
48+
! CHECK: }
49+
! CHECK: omp.terminator
50+
! CHECK: }
51+
! CHECK: return
52+
subroutine multiple_private_fix(gama)
53+
integer :: i, j, x, gama
54+
!$OMP PARALLEL DO PRIVATE(j,x)
55+
do i = 1, gama
56+
do j = 1, gama
57+
x = i + j
58+
end do
59+
end do
60+
!$OMP END PARALLEL DO
61+
end subroutine
62+
63+
! CHECK-LABEL: multiple_private_fix2
64+
! CHECK: %[[X1:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFmultiple_private_fix2Ex"}
65+
! CHECK: %[[X1_DECL:.*]]:2 = hlfir.declare %[[X1]] {uniq_name = "_QFmultiple_private_fix2Ex"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
66+
! CHECK: omp.parallel {
67+
! CHECK: %[[X2:.*]] = fir.alloca i32 {bindc_name = "x", pinned, uniq_name = "_QFmultiple_private_fix2Ex"}
68+
! CHECK: %[[X2_DECL:.*]]:2 = hlfir.declare %[[X2]] {uniq_name = "_QFmultiple_private_fix2Ex"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
69+
! CHECK: omp.parallel {
70+
! CHECK: %[[X3:.*]] = fir.alloca i32 {bindc_name = "x", pinned, uniq_name = "_QFmultiple_private_fix2Ex"}
71+
! CHECK: %[[X3_DECL:.*]]:2 = hlfir.declare %[[X3]] {uniq_name = "_QFmultiple_private_fix2Ex"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
72+
! CHECK: %[[C3:.*]] = arith.constant 1 : i32
73+
! CHECK: hlfir.assign %[[C3]] to %[[X3_DECL]]#0 : i32, !fir.ref<i32>
74+
! CHECK: omp.terminator
75+
! CHECK: }
76+
! CHECK: %[[C2:.*]] = arith.constant 1 : i32
77+
! CHECK: hlfir.assign %[[C2]] to %[[X2_DECL]]#0 : i32, !fir.ref<i32>
78+
! CHECK: omp.terminator
79+
! CHECK: }
80+
! CHECK: %[[C1:.*]] = arith.constant 1 : i32
81+
! CHECK: hlfir.assign %[[C1]] to %[[X1_DECL]]#0 : i32, !fir.ref<i32>
82+
! CHECK: return
83+
subroutine multiple_private_fix2()
84+
integer :: x
85+
!$omp parallel private(x)
86+
!$omp parallel private(x)
87+
x = 1
88+
!$omp end parallel
89+
x = 1
90+
!$omp end parallel
91+
x = 1
92+
end subroutine

‎flang/test/Lower/OpenMP/parallel-reduction-add.f90

Lines changed: 101 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,101 @@
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.reduction.declare
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.reduction.declare
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 : !fir.ref<i32>) {
32+
!CHECK: %[[I_INCR:.*]] = arith.constant 1 : i32
33+
!CHECK: omp.reduction %[[I_INCR]], %[[I_DECL]]#0 : i32, !fir.ref<i32>
34+
!CHECK: omp.terminator
35+
!CHECK: }
36+
!CHECK: return
37+
subroutine simple_int_add
38+
integer :: i
39+
i = 0
40+
41+
!$omp parallel reduction(+:i)
42+
i = i + 1
43+
!$omp end parallel
44+
45+
print *, i
46+
end subroutine
47+
48+
!CHECK-LABEL: func.func @_QPsimple_real_add
49+
!CHECK: %[[RREF:.*]] = fir.alloca f32 {bindc_name = "r", uniq_name = "_QFsimple_real_addEr"}
50+
!CHECK: %[[R_DECL:.*]]:2 = hlfir.declare %[[RREF]] {uniq_name = "_QFsimple_real_addEr"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
51+
!CHECK: %[[R_START:.*]] = arith.constant 0.000000e+00 : f32
52+
!CHECK: hlfir.assign %[[R_START]] to %[[R_DECL]]#0 : f32, !fir.ref<f32>
53+
!CHECK: omp.parallel reduction(@[[RED_F32_NAME]] -> %[[R_DECL]]#0 : !fir.ref<f32>) {
54+
!CHECK: %[[R_INCR:.*]] = arith.constant 1.500000e+00 : f32
55+
!CHECK: omp.reduction %[[R_INCR]], %[[R_DECL]]#0 : f32, !fir.ref<f32>
56+
!CHECK: omp.terminator
57+
!CHECK: }
58+
!CHECK: return
59+
subroutine simple_real_add
60+
real :: r
61+
r = 0.0
62+
63+
!$omp parallel reduction(+:r)
64+
r = r + 1.5
65+
!$omp end parallel
66+
67+
print *, r
68+
end subroutine
69+
70+
!CHECK-LABEL: func.func @_QPint_real_add
71+
!CHECK: %[[IREF:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFint_real_addEi"}
72+
!CHECK: %[[I_DECL:.*]]:2 = hlfir.declare %[[IREF]] {uniq_name = "_QFint_real_addEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
73+
!CHECK: %[[RREF:.*]] = fir.alloca f32 {bindc_name = "r", uniq_name = "_QFint_real_addEr"}
74+
!CHECK: %[[R_DECL:.*]]:2 = hlfir.declare %[[RREF]] {uniq_name = "_QFint_real_addEr"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
75+
!CHECK: %[[R_START:.*]] = arith.constant 0.000000e+00 : f32
76+
!CHECK: hlfir.assign %[[R_START]] to %[[R_DECL]]#0 : f32, !fir.ref<f32>
77+
!CHECK: %[[I_START:.*]] = arith.constant 0 : i32
78+
!CHECK: hlfir.assign %[[I_START]] to %[[I_DECL]]#0 : i32, !fir.ref<i32>
79+
!CHECK: omp.parallel reduction(@[[RED_I32_NAME]] -> %[[I_DECL]]#0 : !fir.ref<i32>, @[[RED_F32_NAME]] -> %[[R_DECL]]#0 : !fir.ref<f32>) {
80+
!CHECK: %[[R_INCR:.*]] = arith.constant 1.500000e+00 : f32
81+
!CHECK: omp.reduction %[[R_INCR]], %[[R_DECL]]#0 : f32, !fir.ref<f32>
82+
!CHECK: %[[I_INCR:.*]] = arith.constant 3 : i32
83+
!CHECK: omp.reduction %[[I_INCR]], %[[I_DECL]]#0 : i32, !fir.ref<i32>
84+
!CHECK: omp.terminator
85+
!CHECK: }
86+
!CHECK: return
87+
subroutine int_real_add
88+
real :: r
89+
integer :: i
90+
91+
r = 0.0
92+
i = 0
93+
94+
!$omp parallel reduction(+:i,r)
95+
r = 1.5 + r
96+
i = i + 3
97+
!$omp end parallel
98+
99+
print *, r
100+
print *, i
101+
end subroutine

‎flang/test/Lower/OpenMP/parallel-sections.f90

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
!RUN: %flang_fc1 -emit-hlfir -fopenmp %s -o - | FileCheck %s
2+
3+
!===============================================================================
4+
! Parallel sections construct
5+
!===============================================================================
6+
7+
!CHECK: func @_QPomp_parallel_sections
8+
subroutine omp_parallel_sections(x, y)
9+
integer, intent(inout) :: x, y
10+
!CHECK: omp.parallel {
11+
!CHECK: omp.sections {
12+
!$omp parallel sections
13+
!CHECK: omp.section {
14+
!$omp section
15+
!CHECK: fir.load
16+
!CHECK: arith.addi
17+
!CHECK: hlfir.assign
18+
x = x + 12
19+
!CHECK: omp.terminator
20+
!CHECK: omp.section {
21+
!$omp section
22+
!CHECK: fir.load
23+
!CHECK: arith.subi
24+
!CHECK: hlfir.assign
25+
y = y - 5
26+
!CHECK: omp.terminator
27+
!CHECK: omp.terminator
28+
!CHECK: omp.terminator
29+
!$omp end parallel sections
30+
end subroutine omp_parallel_sections
31+
32+
!===============================================================================
33+
! Parallel sections construct with allocate clause
34+
!===============================================================================
35+
36+
!CHECK: func @_QPomp_parallel_sections
37+
subroutine omp_parallel_sections_allocate(x, y)
38+
use omp_lib
39+
integer, intent(inout) :: x, y
40+
!CHECK: %[[allocator_1:.*]] = arith.constant 1 : i32
41+
!CHECK: %[[allocator_2:.*]] = arith.constant 1 : i32
42+
!CHECK: omp.parallel allocate(
43+
!CHECK: %[[allocator_2]] : i32 -> %{{.*}} : !fir.ref<i32>) {
44+
!CHECK: omp.sections allocate(
45+
!CHECK: %[[allocator_1]] : i32 -> %{{.*}} : !fir.ref<i32>) {
46+
!$omp parallel sections allocate(omp_high_bw_mem_alloc: x)
47+
!CHECK: omp.section {
48+
!$omp section
49+
x = x + 12
50+
!CHECK: omp.terminator
51+
!CHECK: omp.section {
52+
!$omp section
53+
y = y + 5
54+
!CHECK: omp.terminator
55+
!CHECK: omp.terminator
56+
!CHECK: omp.terminator
57+
!$omp end parallel sections
58+
end subroutine omp_parallel_sections_allocate

‎flang/test/Lower/OpenMP/parallel-wsloop-firstpriv.f90

Lines changed: 69 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
1+
! This test checks lowering of OpenMP parallel DO, with the loop bound being
2+
! a firstprivate variable
3+
4+
! RUN: bbc -fopenmp -emit-hlfir %s -o - | FileCheck %s
5+
6+
! CHECK: func @_QPomp_do_firstprivate(%[[ARG0:.*]]: !fir.ref<i32> {fir.bindc_name = "a"})
7+
subroutine omp_do_firstprivate(a)
8+
! CHECK: %[[ARG0_DECL:.*]]:2 = hlfir.declare %[[ARG0]] {uniq_name = "_QFomp_do_firstprivateEa"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
9+
integer::a
10+
integer::n
11+
n = a+1
12+
!$omp parallel do firstprivate(a)
13+
! CHECK: omp.parallel {
14+
! CHECK: %[[I_PVT_REF:.*]] = fir.alloca i32 {adapt.valuebyref, pinned}
15+
! CHECK: %[[I_PVT_DECL:.*]]:2 = hlfir.declare %[[I_PVT_REF]] {uniq_name = "_QFomp_do_firstprivateEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
16+
! CHECK: %[[A_PVT_REF:.*]] = fir.alloca i32 {bindc_name = "a", pinned, uniq_name = "_QFomp_do_firstprivateEa"}
17+
! CHECK: %[[A_PVT_DECL:.*]]:2 = hlfir.declare %[[A_PVT_REF]] {uniq_name = "_QFomp_do_firstprivateEa"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
18+
! CHECK-NEXT: %[[LD:.*]] = fir.load %[[ARG0_DECL]]#1 : !fir.ref<i32>
19+
! CHECK-NEXT: fir.store %[[LD]] to %[[A_PVT_DECL]]#1 : !fir.ref<i32>
20+
! CHECK: %[[LB:.*]] = arith.constant 1 : i32
21+
! CHECK-NEXT: %[[UB:.*]] = fir.load %[[A_PVT_DECL]]#0 : !fir.ref<i32>
22+
! CHECK-NEXT: %[[STEP:.*]] = arith.constant 1 : i32
23+
! CHECK-NEXT: omp.wsloop for (%[[ARG1:.*]]) : i32 = (%[[LB]]) to (%[[UB]]) inclusive step (%[[STEP]])
24+
! CHECK-NEXT: fir.store %[[ARG1]] to %[[I_PVT_DECL]]#1 : !fir.ref<i32>
25+
! CHECK-NEXT: fir.call @_QPfoo(%[[I_PVT_DECL]]#1, %[[A_PVT_DECL]]#1) {{.*}}: (!fir.ref<i32>, !fir.ref<i32>) -> ()
26+
! CHECK-NEXT: omp.yield
27+
do i=1, a
28+
call foo(i, a)
29+
end do
30+
!$omp end parallel do
31+
!CHECK: fir.call @_QPbar(%[[ARG0_DECL]]#1) {{.*}}: (!fir.ref<i32>) -> ()
32+
call bar(a)
33+
end subroutine omp_do_firstprivate
34+
35+
! CHECK: func @_QPomp_do_firstprivate2(%[[ARG0:.*]]: !fir.ref<i32> {fir.bindc_name = "a"}, %[[ARG1:.*]]: !fir.ref<i32> {fir.bindc_name = "n"})
36+
subroutine omp_do_firstprivate2(a, n)
37+
! CHECK: %[[ARG0_DECL:.*]]:2 = hlfir.declare %[[ARG0]] {uniq_name = "_QFomp_do_firstprivate2Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
38+
! CHECK: %[[ARG1_DECL:.*]]:2 = hlfir.declare %[[ARG1]] {uniq_name = "_QFomp_do_firstprivate2En"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
39+
integer::a
40+
integer::n
41+
n = a+1
42+
!$omp parallel do firstprivate(a, n)
43+
! CHECK: omp.parallel {
44+
! CHECK: %[[I_PVT_REF:.*]] = fir.alloca i32 {adapt.valuebyref, pinned}
45+
! CHECK: %[[I_PVT_DECL:.*]]:2 = hlfir.declare %[[I_PVT_REF]] {uniq_name = "_QFomp_do_firstprivate2Ei"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
46+
! CHECK: %[[A_PVT_REF:.*]] = fir.alloca i32 {bindc_name = "a", pinned
47+
! CHECK: %[[A_PVT_DECL:.*]]:2 = hlfir.declare %[[A_PVT_REF]] {uniq_name = "_QFomp_do_firstprivate2Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
48+
! CHECK: %[[LD:.*]] = fir.load %[[ARG0_DECL]]#1 : !fir.ref<i32>
49+
! CHECK: fir.store %[[LD]] to %[[A_PVT_DECL]]#1 : !fir.ref<i32>
50+
! CHECK: %[[N_PVT_REF:.*]] = fir.alloca i32 {bindc_name = "n", pinned, uniq_name = "_QFomp_do_firstprivate2En"}
51+
! CHECK: %[[N_PVT_DECL:.*]]:2 = hlfir.declare %[[N_PVT_REF]] {uniq_name = "_QFomp_do_firstprivate2En"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
52+
! CHECK: %[[LD1:.*]] = fir.load %[[ARG1_DECL]]#1 : !fir.ref<i32>
53+
! CHECK: fir.store %[[LD1]] to %[[N_PVT_DECL]]#1 : !fir.ref<i32>
54+
55+
56+
! CHECK: %[[LB:.*]] = fir.load %[[A_PVT_DECL]]#0 : !fir.ref<i32>
57+
! CHECK: %[[UB:.*]] = fir.load %[[N_PVT_DECL]]#0 : !fir.ref<i32>
58+
! CHECK: %[[STEP:.*]] = arith.constant 1 : i32
59+
! CHECK: omp.wsloop for (%[[ARG2:.*]]) : i32 = (%[[LB]]) to (%[[UB]]) inclusive step (%[[STEP]])
60+
! CHECK: fir.store %[[ARG2]] to %[[I_PVT_DECL]]#1 : !fir.ref<i32>
61+
! CHECK: fir.call @_QPfoo(%[[I_PVT_DECL]]#1, %[[A_PVT_DECL]]#1) {{.*}}: (!fir.ref<i32>, !fir.ref<i32>) -> ()
62+
! CHECK: omp.yield
63+
do i= a, n
64+
call foo(i, a)
65+
end do
66+
!$omp end parallel do
67+
!CHECK: fir.call @_QPbar(%[[ARG1_DECL]]#1) {{.*}}: (!fir.ref<i32>) -> ()
68+
call bar(n)
69+
end subroutine omp_do_firstprivate2

0 commit comments

Comments
 (0)