35 lines · plain
1// Tests that `fir.local` ops are able to provide an alloca block when required.2 3// RUN: fir-opt %s -convert-hlfir-to-fir | FileCheck %s4 5fir.local {type = local_init} @localizer : !fir.box<!fir.array<1xi32>> copy {6^bb0(%arg0: !fir.ref<!fir.box<!fir.array<1xi32>>>, %arg1: !fir.ref<!fir.box<!fir.array<1xi32>>>):7 %0 = fir.load %arg0 : !fir.ref<!fir.box<!fir.array<1xi32>>>8 hlfir.assign %0 to %arg1 : !fir.box<!fir.array<1xi32>>, !fir.ref<!fir.box<!fir.array<1xi32>>>9 fir.yield(%arg1 : !fir.ref<!fir.box<!fir.array<1xi32>>>)10}11 12func.func @foo() {13 %c1 = arith.constant 1 : index14 %0 = fir.alloca !fir.box<!fir.array<1xi32>>15 fir.do_concurrent {16 fir.do_concurrent.loop (%arg0) = (%c1) to (%c1) step (%c1) local(@localizer %0 -> %arg1 : !fir.ref<!fir.box<!fir.array<1xi32>>>) {17 }18 }19 return20}21 22// CHECK: fir.local {type = local_init} @localizer : ![[TYPE:fir.box<!fir.array<1xi32>>]] copy {23// CHECK: ^bb0(%[[VAL_0:.*]]: !fir.ref<![[TYPE]]>, %[[VAL_1:.*]]: !fir.ref<![[TYPE]]>):24// CHECK: %[[VAL_2:.*]] = fir.alloca ![[TYPE]]25// CHECK: %[[VAL_3:.*]] = fir.load %[[VAL_0]] : !fir.ref<![[TYPE]]>26// CHECK: %[[VAL_4:.*]] = arith.constant 0 : index27// CHECK: %[[VAL_5:.*]]:3 = fir.box_dims %[[VAL_3]], %[[VAL_4]] : (![[TYPE]], index) -> (index, index, index)28// CHECK: %[[VAL_6:.*]] = fir.load %[[VAL_1]] : !fir.ref<![[TYPE]]>29// CHECK: fir.store %[[VAL_6]] to %[[VAL_2]] : !fir.ref<![[TYPE]]>30// CHECK: %[[VAL_10:.*]] = fir.convert %[[VAL_2]] : (!fir.ref<![[TYPE]]>) -> !fir.ref<!fir.box<none>>31// CHECK: %[[VAL_11:.*]] = fir.convert %[[VAL_3]] : (![[TYPE]]) -> !fir.box<none>32// CHECK: fir.call @_FortranAAssign(%[[VAL_10]], %[[VAL_11]], %{{.*}}, %{{.*}})33// CHECK: fir.yield(%[[VAL_1]] : !fir.ref<![[TYPE]]>)34// CHECK: }35