Skip to content

Commit d1a84b9

Browse files
[mlir][linalg] Fix memory leak in pack-dynamic-inner-tile.mlir (#137589)
This fixes the test when running with ASAN.
1 parent 6b973f2 commit d1a84b9

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,13 @@ module @transforms attributes { transform.with_named_sequence } {
125125
%bufferize = transform.bufferization.one_shot_bufferize %module
126126
{bufferize_function_boundaries=true} : (!transform.any_op) -> !transform.any_op
127127

128-
// 5. Canonicalize
128+
// 5. Deallocate buffers.
129129
%func_op_bufferized = transform.structured.match ops{["func.func"]} in %bufferize : (!transform.any_op) -> !transform.op<"func.func">
130-
transform.apply_patterns to %func_op_bufferized {
130+
%func_op_deallocated = transform.apply_registered_pass "buffer-deallocation-pipeline" to %func_op_bufferized
131+
: (!transform.op<"func.func">) -> !transform.op<"func.func">
132+
133+
// 6. Canonicalize
134+
transform.apply_patterns to %func_op_deallocated {
131135
transform.apply_patterns.canonicalization
132136
} : !transform.op<"func.func">
133137

0 commit comments

Comments
 (0)