1// RUN: mlir-opt -verify-diagnostics -expand-realloc=emit-deallocs=false -ownership-based-buffer-deallocation \
2// RUN:  --buffer-deallocation-simplification -split-input-file %s | FileCheck %s
3
4// RUN: mlir-opt %s -buffer-deallocation-pipeline --split-input-file > /dev/null
5
6func.func @auto_dealloc() {
7  %c10 = arith.constant 10 : index
8  %c100 = arith.constant 100 : index
9  %alloc = memref.alloc(%c10) : memref<?xi32>
10  %realloc = memref.realloc %alloc(%c100) : memref<?xi32> to memref<?xi32>
11  "test.read_buffer"(%realloc) : (memref<?xi32>) -> ()
12  return
13}
14
15// CHECK-LABEL: func @auto_dealloc
16//       CHECK:  [[ALLOC:%.*]] = memref.alloc(
17//   CHECK-NOT:  bufferization.dealloc
18//       CHECK:  [[V0:%.+]]:2 = scf.if
19//   CHECK-NOT:  bufferization.dealloc
20//       CHECK:  test.read_buffer
21//  CHECK-NEXT:  [[BASE:%[a-zA-Z0-9_]+]]{{.*}} = memref.extract_strided_metadata [[V0]]#0
22//  CHECK-NEXT:  bufferization.dealloc ([[ALLOC]], [[BASE]] :{{.*}}) if (%true{{[0-9_]*}}, [[V0]]#1)
23//  CHECK-NEXT:  return
24
25// -----
26
27func.func @auto_dealloc_inside_nested_region(%arg0: memref<?xi32>, %arg1: i1) {
28  %c100 = arith.constant 100 : index
29  %0 = scf.if %arg1 -> memref<?xi32> {
30    %realloc = memref.realloc %arg0(%c100) : memref<?xi32> to memref<?xi32>
31    scf.yield %realloc : memref<?xi32>
32  } else {
33    scf.yield %arg0 : memref<?xi32>
34  }
35  "test.read_buffer"(%0) : (memref<?xi32>) -> ()
36  return
37}
38
39// CHECK-LABEL: func @auto_dealloc_inside_nested_region
40//  CHECK-SAME: (%arg0: memref<?xi32>, %arg1: i1)
41//   CHECK-NOT: dealloc
42//       CHECK: "test.read_buffer"([[V0:%.+]]#0)
43//  CHECK-NEXT: [[BASE:%[a-zA-Z0-9_]+]],{{.*}} = memref.extract_strided_metadata [[V0]]#0
44//  CHECK-NEXT: bufferization.dealloc ([[BASE]] : memref<i32>) if ([[V0]]#1)
45//  CHECK-NEXT: return
46