53 lines · plain
1! Test delayed privatization for allocatables: `private`.2 3! RUN: %flang_fc1 -emit-hlfir -fopenmp -mmlir --enable-delayed-privatization \4! RUN: -o - %s 2>&1 | FileCheck %s5! RUN: bbc -emit-hlfir -fopenmp --enable-delayed-privatization -o - %s 2>&1 |\6! RUN: FileCheck %s7 8subroutine delayed_privatization_allocatable9 implicit none10 integer, allocatable :: var111 12!$omp parallel private(var1)13 var1 = 1014!$omp end parallel15end subroutine16 17! CHECK-LABEL: omp.private {type = private}18! CHECK-SAME: @[[PRIVATIZER_SYM:.*]] : [[BOX_TYPE:!fir.box<!fir.heap<i32>>]] init {19 20! CHECK-NEXT: ^bb0(%[[PRIV_ARG:.*]]: [[TYPE:!fir.ref<!fir.box<!fir.heap<i32>>>]], %[[PRIV_ALLOC:.*]]: [[TYPE]]):21 22! CHECK-NEXT: %[[PRIV_ARG_VAL:.*]] = fir.load %[[PRIV_ARG]] : !fir.ref<!fir.box<!fir.heap<i32>>>23! CHECK-NEXT: %[[PRIV_ARG_BOX:.*]] = fir.box_addr %[[PRIV_ARG_VAL]] : (!fir.box<!fir.heap<i32>>) -> !fir.heap<i32>24! CHECK-NEXT: %[[PRIV_ARG_ADDR:.*]] = fir.convert %[[PRIV_ARG_BOX]] : (!fir.heap<i32>) -> i6425! CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : i6426! CHECK-NEXT: %[[ALLOC_COND:.*]] = arith.cmpi eq, %[[PRIV_ARG_ADDR]], %[[C0]] : i6427 28! CHECK-NEXT: fir.if %[[ALLOC_COND]] {29! CHECK-NEXT: %[[ZERO_BOX:.*]] = fir.embox %[[PRIV_ARG_BOX]] : (!fir.heap<i32>) -> !fir.box<!fir.heap<i32>>30! CHECK-NEXT: fir.store %[[ZERO_BOX]] to %[[PRIV_ALLOC]] : !fir.ref<!fir.box<!fir.heap<i32>>>31! CHECK-NEXT: } else {32! CHECK-NEXT: %[[PRIV_ALLOCMEM:.*]] = fir.allocmem i3233! CHECK-NEXT: %[[PRIV_ALLOCMEM_BOX:.*]] = fir.embox %[[PRIV_ALLOCMEM]] : (!fir.heap<i32>) -> !fir.box<!fir.heap<i32>>34! CHECK-NEXT: fir.store %[[PRIV_ALLOCMEM_BOX]] to %[[PRIV_ALLOC]] : !fir.ref<!fir.box<!fir.heap<i32>>>35! CHECK-NEXT: }36 37! CHECK-NEXT: omp.yield(%[[PRIV_ALLOC]] : [[TYPE]])38 39! CHECK-NEXT: } dealloc {40! CHECK-NEXT: ^bb0(%[[PRIV_ARG:.*]]: [[TYPE]]):41 42! CHECK-NEXT: %[[PRIV_VAL:.*]] = fir.load %[[PRIV_ARG]]43! CHECK-NEXT: %[[PRIV_ADDR:.*]] = fir.box_addr %[[PRIV_VAL]]44! CHECK-NEXT: %[[PRIV_ADDR_I64:.*]] = fir.convert %[[PRIV_ADDR]]45! CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : i6446! CHECK-NEXT: %[[PRIV_NULL_COND:.*]] = arith.cmpi ne, %[[PRIV_ADDR_I64]], %[[C0]] : i6447 48! CHECK-NEXT: fir.if %[[PRIV_NULL_COND]] {49! CHECK-NEXT: fir.freemem %[[PRIV_ADDR]]50! CHECK-NEXT: }51 52! CHECK-NEXT: omp.yield53