remove finalize_bufferize (#108)

makslevental · web-flow · commit b19f9b4246a0 · 2024-11-27T00:15:29.000Z
diff --git a/mlir/extras/runtime/passes.py b/mlir/extras/runtime/passes.py
@@ -142,7 +142,7 @@ def bufferize(self):
         return (
             self.Func(Pipeline().empty_tensor_to_alloc_tensor())
             .one_shot_bufferize()
-            .Func(Pipeline().finalizing_bufferize().buffer_deallocation())
+            .Func(Pipeline().buffer_deallocation())
         )
 
     def lower_to_llvm(self):

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,7 @@ def bufferize(self):`
`142`	`142`	`return (`
`143`	`143`	`self.Func(Pipeline().empty_tensor_to_alloc_tensor())`
`144`	`144`	`.one_shot_bufferize()`
`145`		`- .Func(Pipeline().finalizing_bufferize().buffer_deallocation())`
	`145`	`+ .Func(Pipeline().buffer_deallocation())`
`146`	`146`	`)`
`147`	`147`
`148`	`148`	`def lower_to_llvm(self):`