64 lines · plain
1! Test delayed privatization for allocatable arrays.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_private(var1, l1)9 implicit none10 integer(8):: l111 integer, allocatable, dimension(:) :: var112 13!$omp parallel firstprivate(var1)14 var1(l1 + 1) = 1015!$omp end parallel16end subroutine17 18! CHECK-LABEL: omp.private {type = firstprivate}19! CHECK-SAME: @[[PRIVATIZER_SYM:.*]] : [[BOX_TYPE:!fir.box<!fir.heap<!fir.array<\?xi32>>>]] init {20 21! CHECK-NEXT: ^bb0(%[[PRIV_ARG:.*]]: [[TYPE:!fir.ref<!fir.box<!fir.heap<!fir.array<\?xi32>>>>]], %[[PRIV_ALLOC:.*]]: [[TYPE]]):22 23! CHECK-NEXT: %[[PRIV_ARG_VAL:.*]] = fir.load %[[PRIV_ARG]]24! CHECK-NEXT: %[[PRIV_ARG_BOX:.*]] = fir.box_addr %[[PRIV_ARG_VAL]]25! CHECK-NEXT: %[[PRIV_ARG_ADDR:.*]] = fir.convert %[[PRIV_ARG_BOX]]26! CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : i6427! CHECK-NEXT: %[[ALLOC_COND:.*]] = arith.cmpi eq, %[[PRIV_ARG_ADDR]], %[[C0]] : i6428 29! CHECK-NEXT: fir.if %[[ALLOC_COND]] {30! CHECK-NEXT: %[[C0_2:.*]] = arith.constant 0 : index31! CHECK-NEXT: %[[SHAPE:.*]] = fir.shape %[[C0_2]]32! CHECK-NEXT: %[[EMBOX_2:.*]] = fir.embox %[[PRIV_ARG_BOX]](%[[SHAPE]])33! CHECK-NEXT: fir.store %[[EMBOX_2]] to %[[PRIV_ALLOC]]34! CHECK-NEXT: } else {35! CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : index36! CHECK-NEXT: %[[DIMS:.*]]:3 = fir.box_dims %[[PRIV_ARG_VAL]], %[[C0]]37! CHECK-NEXT: %[[SHAPE:.*]] = fir.shape %[[DIMS]]#138! CHECK-NEXT: %[[MEM:.*]] = fir.allocmem !fir.array<?xi32>, %[[DIMS]]#139! CHECK-NEXT: %[[DECL:.*]]:2 = hlfir.declare %[[MEM]](%[[SHAPE]])40! CHECK-NEXT: %[[C0_2:.*]] = arith.constant 0 : index41! CHECK-NEXT: %[[DIMS_2:.*]]:3 = fir.box_dims %[[PRIV_ARG_VAL]], %[[C0_2]]42! CHECK-NEXT: %[[SHAPE_SHIFT:.*]] = fir.shape_shift %[[DIMS_2]]#0, %[[DIMS_2]]#143! CHECK-NEXT: %[[EMBOX:.*]] = fir.rebox %[[DECL]]#0(%[[SHAPE_SHIFT]])44! CHECK-NEXT: fir.store %[[EMBOX]] to %[[PRIV_ALLOC]]45! CHECK-NEXT: }46 47! CHECK-NEXT: omp.yield48 49! CHECK-NEXT: } copy {50! CHECK-NEXT: ^bb0(%[[PRIV_ORIG_ARG:.*]]: [[TYPE]], %[[PRIV_PRIV_ARG:.*]]: [[TYPE]]):51! CHECK-NEXT: %[[PRIV_BASE_VAL:.*]] = fir.load %[[PRIV_PRIV_ARG]]52! CHECK-NEXT: %[[PRIV_BASE_BOX:.*]] = fir.box_addr %[[PRIV_BASE_VAL]]53! CHECK-NEXT: %[[PRIV_BASE_ADDR:.*]] = fir.convert %[[PRIV_BASE_BOX]]54! CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : i6455! CHECK-NEXT: %[[COPY_COND:.*]] = arith.cmpi ne, %[[PRIV_BASE_ADDR]], %[[C0]] : i6456 57 58! CHECK-NEXT: fir.if %[[COPY_COND]] {59! CHECK-NEXT: %[[PRIV_ORIG_ARG_VAL:.*]] = fir.load %[[PRIV_ORIG_ARG]]60! CHECK-NEXT: hlfir.assign %[[PRIV_ORIG_ARG_VAL]] to %[[PRIV_PRIV_ARG]] realloc61! CHECK-NEXT: }62! CHECK-NEXT: omp.yield63! CHECK-NEXT: }64