|
| 1 | +! Simple test for lowering of OpenMP Threadprivate Directive with HLFIR. |
| 2 | + |
| 3 | +!RUN: %flang_fc1 -emit-hlfir -fopenmp %s -o - | FileCheck %s |
| 4 | + |
| 5 | +!CHECK-LABEL: func.func @_QPsub1() { |
| 6 | +!CHECK: %[[A:.*]] = fir.address_of(@_QFsub1Ea) : !fir.ref<i32> |
| 7 | +!CHECK: %[[A_DECL:.*]]:2 = hlfir.declare %[[A]] {uniq_name = "_QFsub1Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 8 | +!CHECK: %[[A_TP0:.*]] = omp.threadprivate %[[A_DECL]]#1 : !fir.ref<i32> -> !fir.ref<i32> |
| 9 | +!CHECK: %[[A_CVT:.*]]:2 = hlfir.declare %[[A_TP0]] {uniq_name = "_QFsub1Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 10 | +!CHECK: omp.parallel { |
| 11 | +!CHECK: %[[A_TP:.*]] = omp.threadprivate %[[A_DECL]]#1 : !fir.ref<i32> -> !fir.ref<i32> |
| 12 | +!CHECK: %[[A_TP_DECL:.*]]:2 = hlfir.declare %[[A_TP]] {uniq_name = "_QFsub1Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 13 | +!CHECK: %[[TID:.*]] = fir.call @omp_get_thread_num() fastmath<contract> : () -> i32 |
| 14 | +!CHECK: hlfir.assign %[[TID]] to %[[A_TP_DECL]]#0 : i32, !fir.ref<i32> |
| 15 | +!CHECK: omp.terminator |
| 16 | +!CHECK: } |
| 17 | + |
| 18 | +subroutine sub1() |
| 19 | + use omp_lib |
| 20 | + integer, save :: a |
| 21 | + !$omp threadprivate(a) |
| 22 | + !$omp parallel default(none) |
| 23 | + a = omp_get_thread_num() |
| 24 | + !$omp end parallel |
| 25 | +end subroutine |
| 26 | + |
| 27 | +!CHECK-LABEL: func.func @_QPsub2() { |
| 28 | +!CHECK: %[[BLK:.*]] = fir.address_of(@blk_) : !fir.ref<!fir.array<4xi8>> |
| 29 | +!CHECK: %[[BLK_CVT:.*]] = fir.convert %[[BLK]] : (!fir.ref<!fir.array<4xi8>>) -> !fir.ref<!fir.array<?xi8>> |
| 30 | +!CHECK: %c0 = arith.constant 0 : index |
| 31 | +!CHECK: %[[A_ADDR:.*]] = fir.coordinate_of %[[BLK_CVT]], %c0 : (!fir.ref<!fir.array<?xi8>>, index) -> !fir.ref<i8> |
| 32 | +!CHECK: %[[A_CVT:.*]] = fir.convert %[[A_ADDR]] : (!fir.ref<i8>) -> !fir.ref<i32> |
| 33 | +!CHECK: %[[A_DECL:.*]]:2 = hlfir.declare %[[A_CVT]] {uniq_name = "_QFsub2Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 34 | +!CHECK: %[[A_TP0:.*]] = omp.threadprivate %[[BLK]] : !fir.ref<!fir.array<4xi8>> -> !fir.ref<!fir.array<4xi8>> |
| 35 | +!CHECK: %[[A_TP0_CVT:.*]] = fir.convert %[[A_TP0]] : (!fir.ref<!fir.array<4xi8>>) -> !fir.ref<!fir.array<?xi8>> |
| 36 | +!CHECK: %c0_0 = arith.constant 0 : index |
| 37 | +!CHECK: %[[A_TP0_ADDR:.*]] = fir.coordinate_of %[[A_TP0_CVT]], %c0_0 : (!fir.ref<!fir.array<?xi8>>, index) -> !fir.ref<i8> |
| 38 | +!CHECK: %[[A_TP0_ADDR_CVT:.*]] = fir.convert %[[A_TP0_ADDR]] : (!fir.ref<i8>) -> !fir.ref<i32> |
| 39 | +!CHECK: %[[A_TP0_DECL:.*]]:2 = hlfir.declare %[[A_TP0_ADDR_CVT]] {uniq_name = "_QFsub2Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 40 | +!CHECK: omp.parallel { |
| 41 | +!CHECK: %[[BLK_TP:.*]] = omp.threadprivate %[[BLK]] : !fir.ref<!fir.array<4xi8>> -> !fir.ref<!fir.array<4xi8>> |
| 42 | +!CHECK: %[[BLK_TP_CVT:.*]] = fir.convert %[[BLK_TP]] : (!fir.ref<!fir.array<4xi8>>) -> !fir.ref<!fir.array<?xi8>> |
| 43 | +!CHECK: %c0_1 = arith.constant 0 : index |
| 44 | +!CHECK: %[[A_TP_ADDR:.*]] = fir.coordinate_of %[[BLK_TP_CVT]], %c0_1 : (!fir.ref<!fir.array<?xi8>>, index) -> !fir.ref<i8> |
| 45 | +!CHECK: %[[A_TP_ADDR_CVT:.*]] = fir.convert %[[A_TP_ADDR]] : (!fir.ref<i8>) -> !fir.ref<i32> |
| 46 | +!CHECK: %[[A_TP_DECL:.*]]:2 = hlfir.declare %[[A_TP_ADDR_CVT]] {uniq_name = "_QFsub2Ea"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>) |
| 47 | +!CHECK: %[[TID:.*]] = fir.call @omp_get_thread_num() fastmath<contract> : () -> i32 |
| 48 | +!CHECK: hlfir.assign %[[TID]] to %[[A_TP_DECL]]#0 : i32, !fir.ref<i32> |
| 49 | +!CHECK: omp.terminator |
| 50 | +!CHECK: } |
| 51 | + |
| 52 | +subroutine sub2() |
| 53 | + use omp_lib |
| 54 | + integer :: a |
| 55 | + common/blk/a |
| 56 | + !$omp threadprivate(/blk/) |
| 57 | + !$omp parallel default(none) |
| 58 | + a = omp_get_thread_num() |
| 59 | + !$omp end parallel |
| 60 | +end subroutine |
0 commit comments