From b19f9b4246a0e092a2f30eaa6dde8584adb64bc5 Mon Sep 17 00:00:00 2001 From: Maksim Levental Date: Tue, 26 Nov 2024 18:15:29 -0600 Subject: [PATCH] remove finalize_bufferize (#108) --- mlir/extras/runtime/passes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlir/extras/runtime/passes.py b/mlir/extras/runtime/passes.py index 632b6cb..6c0f1fb 100644 --- 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):