xref: /llvm-project/flang/test/Lower/OpenMP/delayed-privatization-allocatable-private.f90 (revision 57d0d3b4d638d170035f55d79f0202f1042de345)
1! Test delayed privatization for allocatables: `private`.
2
3! RUN: %flang_fc1 -emit-hlfir -fopenmp -mmlir --openmp-enable-delayed-privatization \
4! RUN:   -o - %s 2>&1 | FileCheck %s
5! RUN: bbc -emit-hlfir -fopenmp --openmp-enable-delayed-privatization -o - %s 2>&1 |\
6! RUN:   FileCheck %s
7
8subroutine delayed_privatization_allocatable
9  implicit none
10  integer, allocatable :: var1
11
12!$omp parallel private(var1)
13  var1 = 10
14!$omp end parallel
15end subroutine
16
17! CHECK-LABEL: omp.private {type = private}
18! CHECK-SAME: @[[PRIVATIZER_SYM:.*]] : [[TYPE:!fir.ref<!fir.box<!fir.heap<i32>>>]] alloc {
19
20! CHECK-NEXT: ^bb0(%[[PRIV_ARG:.*]]: [[TYPE]]):
21
22! CHECK-NEXT:   %[[PRIV_ALLOC:.*]] = fir.alloca !fir.box<!fir.heap<i32>> {bindc_name = "var1", pinned, uniq_name = "_QFdelayed_privatization_allocatableEvar1"}
23
24! CHECK-NEXT:   %[[PRIV_ARG_VAL:.*]] = fir.load %[[PRIV_ARG]] : !fir.ref<!fir.box<!fir.heap<i32>>>
25! CHECK-NEXT:   %[[PRIV_ARG_BOX:.*]] = fir.box_addr %[[PRIV_ARG_VAL]] : (!fir.box<!fir.heap<i32>>) -> !fir.heap<i32>
26! CHECK-NEXT:   %[[PRIV_ARG_ADDR:.*]] = fir.convert %[[PRIV_ARG_BOX]] : (!fir.heap<i32>) -> i64
27! CHECK-NEXT:   %[[C0:.*]] = arith.constant 0 : i64
28! CHECK-NEXT:   %[[ALLOC_COND:.*]] = arith.cmpi ne, %[[PRIV_ARG_ADDR]], %[[C0]] : i64
29
30! CHECK-NEXT:   fir.if %[[ALLOC_COND]] {
31! CHECK:          %[[PRIV_ALLOCMEM:.*]] = fir.allocmem i32 {fir.must_be_heap = true, uniq_name = "_QFdelayed_privatization_allocatableEvar1.alloc"}
32! CHECK-NEXT:     %[[PRIV_ALLOCMEM_BOX:.*]] = fir.embox %[[PRIV_ALLOCMEM]] : (!fir.heap<i32>) -> !fir.box<!fir.heap<i32>>
33! CHECK-NEXT:     fir.store %[[PRIV_ALLOCMEM_BOX]] to %[[PRIV_ALLOC]] : !fir.ref<!fir.box<!fir.heap<i32>>>
34! CHECK-NEXT:   } else {
35! CHECK-NEXT:     %[[ZERO_BITS:.*]] = fir.zero_bits !fir.heap<i32>
36! CHECK-NEXT:     %[[ZERO_BOX:.*]] = fir.embox %[[ZERO_BITS]] : (!fir.heap<i32>) -> !fir.box<!fir.heap<i32>>
37! CHECK-NEXT:     fir.store %[[ZERO_BOX]] to %[[PRIV_ALLOC]] : !fir.ref<!fir.box<!fir.heap<i32>>>
38! CHECK-NEXT:   }
39
40! CHECK-NEXT:   %[[PRIV_DECL:.*]]:2 = hlfir.declare %[[PRIV_ALLOC]]
41! CHECK-NEXT:   omp.yield(%[[PRIV_DECL]]#0 : [[TYPE]])
42
43! CHECK-NEXT: } dealloc {
44! CHECK-NEXT: ^bb0(%[[PRIV_ARG:.*]]: [[TYPE]]):
45
46! CHECK-NEXT:   %[[PRIV_VAL:.*]] = fir.load %[[PRIV_ARG]]
47! CHECK-NEXT:   %[[PRIV_ADDR:.*]] = fir.box_addr %[[PRIV_VAL]]
48! CHECK-NEXT:   %[[PRIV_ADDR_I64:.*]] = fir.convert %[[PRIV_ADDR]]
49! CHECK-NEXT:   %[[C0:.*]] = arith.constant 0 : i64
50! CHECK-NEXT:   %[[PRIV_NULL_COND:.*]] = arith.cmpi ne, %[[PRIV_ADDR_I64]], %[[C0]] : i64
51
52! CHECK-NEXT:   fir.if %[[PRIV_NULL_COND]] {
53! CHECK:          %[[PRIV_VAL_2:.*]] = fir.load %[[PRIV_ARG]]
54! CHECK-NEXT:     %[[PRIV_ADDR_2:.*]] = fir.box_addr %[[PRIV_VAL_2]]
55! CHECK-NEXT:     fir.freemem %[[PRIV_ADDR_2]]
56! CHECK-NEXT:     %[[ZEROS:.*]] = fir.zero_bits
57! CHECK-NEXT:     %[[ZEROS_BOX:.*]]  = fir.embox %[[ZEROS]]
58! CHECK-NEXT:     fir.store %[[ZEROS_BOX]] to %[[PRIV_ARG]]
59! CHECK-NEXT:   }
60
61! CHECK-NEXT:   omp.yield
62