43 lines · plain
1// RUN: fir-opt %s --opt-bufferization | FileCheck %s2 3// Test that alias analysis doesn't crash determining if the arguments to4// hlfir.assign alias.5// CHECK: omp.private {type = firstprivate} @_QFFbEl_firstprivate_box_Uxi326 7// TODO: we can't currently optimize this assign because alias analysis doesn't8// know that the block arguments of the copy region cannot alias.9 10omp.private {type = firstprivate} @_QFFbEl_firstprivate_box_Uxi32 : !fir.box<!fir.array<?xi32>> init {11^bb0(%arg0: !fir.ref<!fir.box<!fir.array<?xi32>>>, %arg1: !fir.ref<!fir.box<!fir.array<?xi32>>>):12 %0 = fir.load %arg0 : !fir.ref<!fir.box<!fir.array<?xi32>>>13 %c0 = arith.constant 0 : index14 %1:3 = fir.box_dims %0, %c0 : (!fir.box<!fir.array<?xi32>>, index) -> (index, index, index)15 %2 = fir.shape %1#1 : (index) -> !fir.shape<1>16 %3 = fir.allocmem !fir.array<?xi32>, %1#1 {bindc_name = ".tmp", uniq_name = ""}17 %true = arith.constant true18 %4:2 = hlfir.declare %3(%2) {uniq_name = ".tmp"} : (!fir.heap<!fir.array<?xi32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xi32>>, !fir.heap<!fir.array<?xi32>>)19 %c0_0 = arith.constant 0 : index20 %5:3 = fir.box_dims %0, %c0_0 : (!fir.box<!fir.array<?xi32>>, index) -> (index, index, index)21 %6 = fir.shape_shift %5#0, %5#1 : (index, index) -> !fir.shapeshift<1>22 %7 = fir.rebox %4#0(%6) : (!fir.box<!fir.array<?xi32>>, !fir.shapeshift<1>) -> !fir.box<!fir.array<?xi32>>23 fir.store %7 to %arg1 : !fir.ref<!fir.box<!fir.array<?xi32>>>24 omp.yield(%arg1 : !fir.ref<!fir.box<!fir.array<?xi32>>>)25} copy {26^bb0(%arg0: !fir.ref<!fir.box<!fir.array<?xi32>>>, %arg1 : !fir.ref<!fir.box<!fir.array<?xi32>>>):27 %0 = fir.load %arg0 {test.ptr = "load_from_block_arg"} : !fir.ref<!fir.box<!fir.array<?xi32>>>28 hlfir.assign %0 to %arg1 : !fir.box<!fir.array<?xi32>>, !fir.ref<!fir.box<!fir.array<?xi32>>>29 omp.yield(%arg1 : !fir.ref<!fir.box<!fir.array<?xi32>>>)30} dealloc {31^bb0(%arg0: !fir.ref<!fir.box<!fir.array<?xi32>>>):32 %0 = fir.load %arg0 : !fir.ref<!fir.box<!fir.array<?xi32>>>33 %1 = fir.box_addr %0 : (!fir.box<!fir.array<?xi32>>) -> !fir.ref<!fir.array<?xi32>>34 %2 = fir.convert %1 : (!fir.ref<!fir.array<?xi32>>) -> i6435 %c0_i64 = arith.constant 0 : i6436 %3 = arith.cmpi ne, %2, %c0_i64 : i6437 fir.if %3 {38 %4 = fir.convert %1 : (!fir.ref<!fir.array<?xi32>>) -> !fir.heap<!fir.array<?xi32>>39 fir.freemem %4 : !fir.heap<!fir.array<?xi32>>40 }41 omp.yield42}43