5109cb28fd
The buffer deallocation pipeline now works on modules and functions. Also add extra test cases that run the buffer deallocation pipeline on modules and functions. (Test cases that insert a helper function.)
21 lines
900 B
MLIR
21 lines
900 B
MLIR
// RUN: mlir-opt %s -buffer-deallocation-pipeline | FileCheck %s
|
|
// RUN: mlir-opt %s --pass-pipeline="builtin.module(buffer-deallocation-pipeline)" | FileCheck %s
|
|
|
|
// Test case requires a helper function.
|
|
|
|
// CHECK-LABEL: func.func @test(
|
|
// CHECK: func.call @[[helper:.*]]({{.*}}) : ({{.*}}) -> ()
|
|
func.func @test(%lb : index, %ub: index) -> (memref<5xf32>, memref<5xf32>) {
|
|
%0 = memref.alloc() : memref<5xf32>
|
|
%1 = memref.alloc() : memref<5xf32>
|
|
%c1 = arith.constant 1 : index
|
|
%a, %b = scf.for %iv = %lb to %ub step %c1 iter_args(%c = %0, %d = %1)
|
|
-> (memref<5xf32>, memref<5xf32>) {
|
|
scf.yield %d, %c : memref<5xf32>, memref<5xf32>
|
|
}
|
|
return %a, %b : memref<5xf32>, memref<5xf32>
|
|
}
|
|
|
|
// CHECK: func.func private @[[helper]](
|
|
// CHECK-SAME: %{{.*}}: memref<?xindex>, %{{.*}}: memref<?xindex>, %{{.*}}: memref<?xi1>, %{{.*}}: memref<?xi1>, %{{.*}}: memref<?xi1>)
|