diff --git a/flang/test/Lower/OpenMP/threadprivate-use-association.f90 b/flang/test/Lower/OpenMP/threadprivate-use-association.f90 new file mode 100644 index 0000000000000..d0d461547db2f --- /dev/null +++ b/flang/test/Lower/OpenMP/threadprivate-use-association.f90 @@ -0,0 +1,82 @@ +! This test checks lowering of OpenMP Threadprivate Directive. +! Test for threadprivate variable in use association. + +!RUN: %flang_fc1 -emit-hlfir -fopenmp %s -o - | FileCheck %s + +!CHECK-DAG: fir.global common @blk_(dense<0> : vector<24xi8>) : !fir.array<24xi8> +!CHECK-DAG: fir.global @_QMtestEy : f32 { + +module test + integer :: x + real :: y, z(5) + common /blk/ x, z + + !$omp threadprivate(y, /blk/) + +contains + subroutine sub() +! CHECK-LABEL: @_QMtestPsub +!CHECK-DAG: [[ADDR0:%.*]] = fir.address_of(@blk_) : !fir.ref> +!CHECK-DAG: [[NEWADDR0:%.*]] = omp.threadprivate [[ADDR0]] : !fir.ref> -> !fir.ref> +!CHECK-DAG: [[ADDR1:%.*]] = fir.address_of(@_QMtestEy) : !fir.ref +!CHECK-DAG: %[[DECY:.*]]:2 = hlfir.declare [[ADDR1]] {uniq_name = "_QMtestEy"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[NEWADDR1:%.*]] = omp.threadprivate %[[DECY]]#1 : !fir.ref -> !fir.ref + + !$omp parallel +!CHECK-DAG: [[ADDR2:%.*]] = omp.threadprivate [[ADDR0]] : !fir.ref> -> !fir.ref> +!CHECK-DAG: [[ADDR3:%.*]] = fir.convert [[ADDR2]] : (!fir.ref>) -> !fir.ref> +!CHECK-DAG: [[ADDR4:%.*]] = fir.coordinate_of [[ADDR3]], %{{.*}} : (!fir.ref>, index) -> !fir.ref +!CHECK-DAG: [[ADDR5:%.*]] = fir.convert [[ADDR4]] : (!fir.ref) -> !fir.ref +!CHECK-DAG: %[[ADDR6:.*]]:2 = hlfir.declare [[ADDR5]] {uniq_name = "_QMtestEx"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[NEWADDR2:%.*]] = omp.threadprivate %[[DECY]]#1 : !fir.ref -> !fir.ref +!CHECK-DAG: %[[ADDR7:.*]]:2 = hlfir.declare [[NEWADDR2]] {uniq_name = "_QMtestEy"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[ADDR8:%.*]] = fir.convert [[ADDR2]] : (!fir.ref>) -> !fir.ref> +!CHECK-DAG: [[ADDR9:%.*]] = fir.coordinate_of [[ADDR8]], %{{.*}} : (!fir.ref>, index) -> !fir.ref +!CHECK-DAG: [[ADDR10:%.*]] = fir.convert [[ADDR9]] : (!fir.ref) -> !fir.ref> +!CHECK-DAG: %[[ADDR11:.*]]:2 = hlfir.declare [[ADDR10]](%{{.*}}) {uniq_name = "_QMtestEz"} : (!fir.ref>, !fir.shape<1>) -> (!fir.ref>, !fir.ref>) +!CHECK-DAG: %{{.*}} = fir.load %[[ADDR6]]#0 : !fir.ref +!CHECK-DAG: %{{.*}} = fir.load %[[ADDR7]]#0 : !fir.ref +!CHECK-DAG: %{{.*}} = fir.embox %[[ADDR11]]#1(%{{.*}}) : (!fir.ref>, !fir.shape<1>) -> !fir.box> + print *, x, y, z + !$omp end parallel + end +end + +program main + use test + integer :: x1 + real :: z1(5) + common /blk/ x1, z1 + + !$omp threadprivate(/blk/) + + call sub() + +! CHECK-LABEL: @_QQmain() +!CHECK-DAG: [[ADDR0:%.*]] = fir.address_of(@blk_) : !fir.ref> +!CHECK-DAG: [[NEWADDR0:%.*]] = omp.threadprivate [[ADDR0]] : !fir.ref> -> !fir.ref> +!CHECK-DAG: [[ADDR1:%.*]] = fir.address_of(@blk_) : !fir.ref> +!CHECK-DAG: [[NEWADDR1:%.*]] = omp.threadprivate [[ADDR1]] : !fir.ref> -> !fir.ref> +!CHECK-DAG: [[ADDR2:%.*]] = fir.address_of(@_QMtestEy) : !fir.ref +!CHECK-DAG: %[[ADDR3:.*]]:2 = hlfir.declare [[ADDR2]] {uniq_name = "_QMtestEy"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[NEWADDR2:%.*]] = omp.threadprivate %[[ADDR3]]#1 : !fir.ref -> !fir.ref + + !$omp parallel +!CHECK-DAG: [[ADDR4:%.*]] = omp.threadprivate [[ADDR1]] : !fir.ref> -> !fir.ref> +!CHECK-DAG: [[ADDR6:%.*]] = fir.convert [[ADDR4]] : (!fir.ref>) -> !fir.ref> +!CHECK-DAG: [[ADDR7:%.*]] = fir.coordinate_of [[ADDR6]], %{{.*}} : (!fir.ref>, index) -> !fir.ref +!CHECK-DAG: [[ADDR8:%.*]] = fir.convert [[ADDR7]] : (!fir.ref) -> !fir.ref +!CHECK-DAG: %[[DECX1:.*]]:2 = hlfir.declare [[ADDR8]] {uniq_name = "_QFEx1"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[ADDR5:%.*]] = omp.threadprivate %[[ADDR3]]#1 : !fir.ref -> !fir.ref +!CHECK-DAG: %[[DECY:.*]]:2 = hlfir.declare [[ADDR5]] {uniq_name = "_QMtestEy"} : (!fir.ref) -> (!fir.ref, !fir.ref) +!CHECK-DAG: [[ADDR9:%.*]] = fir.convert [[ADDR4]] : (!fir.ref>) -> !fir.ref> +!CHECK-DAG: [[ADDR10:%.*]] = fir.coordinate_of [[ADDR9]], %{{.*}} : (!fir.ref>, index) -> !fir.ref +!CHECK-DAG: [[ADDR11:%.*]] = fir.convert [[ADDR10]] : (!fir.ref) -> !fir.ref> +!CHECK-DAG: %[[DECZ1:.*]]:2 = hlfir.declare [[ADDR11]](%{{.*}}) {uniq_name = "_QFEz1"} : (!fir.ref>, !fir.shape<1>) -> (!fir.ref>, !fir.ref>) +!CHECK-DAG: %{{.*}} = fir.load %[[DECX1]]#0 : !fir.ref +!CHECK-DAG: %{{.*}} = fir.load %[[DECY]]#0 : !fir.ref +!CHECK-DAG: %{{.*}} = fir.embox %[[DECZ1]]#1(%{{.*}}) : (!fir.ref>, !fir.shape<1>) -> !fir.box> + print *, x1, y, z1 + !$omp end parallel + +end