BufferReuse

mlir/include/mlir/Transforms/Passes.h

@@ -46,6 +46,9 @@ std::unique_ptr<Pass>
 createPromoteBuffersToStackPass(unsigned maxAllocSizeInBytes = 1024,
                                 unsigned bitwidthOfIndexType = 64);
 
+/// Creates a pass that reuses buffers which are already allocated.
+std::unique_ptr<Pass> createBufferReusePass();
+
 /// Creates a pass that converts memref function results to out-params.
 std::unique_ptr<Pass> createBufferResultsToOutParamsPass();
 

mlir/include/mlir/Transforms/Passes.td

@@ -221,6 +221,18 @@ def PromoteBuffersToStack : FunctionPass<"promote-buffers-to-stack"> {
   ];
 }
 
+def BufferReuse : FunctionPass<"buffer-reuse"> {
+  let summary = "Reuses already allocated buffers to save allocation "
+                "operations if all criteria are met";
+  let description = [{
+    This pass tries to reuse already allocated buffers if they have the same
+    size, are in the right appearance order, have no interference among
+    themselves and have a short distance between those buffers (this prevents
+    unnecessary allocated memory for long distances).
+  }];
+  let constructor = "mlir::createBufferReusePass()";
+}
+
 def BufferResultsToOutParams : Pass<"buffer-results-to-out-params", "ModuleOp">  {
   let summary = "Converts memref-typed function results to out-params";
   let description = [{