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

[llvmlistbot at llvm.org]

Author: Matthias Springer
Date: 2025-04-28T17:17:41+02:00
New Revision: d1a84b99691e94aaf9b344cc84f1b8d98ebd3518

URL: https://github.com/llvm/llvm-project/commit/d1a84b99691e94aaf9b344cc84f1b8d98ebd3518
DIFF: https://github.com/llvm/llvm-project/commit/d1a84b99691e94aaf9b344cc84f1b8d98ebd3518.diff

LOG: [mlir][linalg] Fix memory leak in `pack-dynamic-inner-tile.mlir` (#137589)

This fixes the test when running with ASAN.

Added: 
    

Modified: 
    mlir/test/Integration/Dialect/Linalg/CPU/pack-dynamic-inner-tile.mlir

Removed: 
    


################################################################################
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">