[Mlir-commits] [mlir] [mlir][linalg] Fix memory leak in `pack-dynamic-inner-tile.mlir` (PR #137589)

[Matthias Springer]

https://github.com/matthias-springer created https://github.com/llvm/llvm-project/pull/137589

None

>From 56c95b429ef7084309f36268986d67540fff39f9 Mon Sep 17 00:00:00 2001
From: Matthias Springer <mspringer at nvidia.com>
Date: Mon, 28 Apr 2025 09:32:46 +0200
Subject: [PATCH] [mlir][linalg] Fix memory leak in
 `pack-dynamic-inner-tile.mlir`

---
 .../Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir       | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/mlir/test/Integration/Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir b/mlir/test/Integration/Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir
index 7b410ccee9633..187d7b862604b 100644
--- a/mlir/test/Integration/Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir
+++ b/mlir/test/Integration/Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir
@@ -125,9 +125,13 @@ module @transforms attributes { transform.with_named_sequence } {
     %bufferize = transform.bufferization.one_shot_bufferize %module
       {bufferize_function_boundaries=true} : (!transform.any_op) -> !transform.any_op
 
-    // 5. Canonicalize
+    // 5. Deallocate buffers.
     %func_op_bufferized = transform.structured.match ops{["func.func"]} in %bufferize : (!transform.any_op) -> !transform.op<"func.func">
-    transform.apply_patterns to %func_op_bufferized {
+    %func_op_deallocated = transform.apply_registered_pass "buffer-deallocation-pipeline" to %func_op_bufferized
+      : (!transform.op<"func.func">) -> !transform.op<"func.func">
+
+    // 6. Canonicalize
+    transform.apply_patterns to %func_op_deallocated {
       transform.apply_patterns.canonicalization
     } : !transform.op<"func.func">