[Mlir-commits] [mlir] 102273a - [mlir][Transform] Remove `notifyOperationErased` workaround (#84134)

[llvmlistbot at llvm.org]

Author: Matthias Springer
Date: 2024-03-15T10:29:36+09:00
New Revision: 102273a9b4886a11c78b28a77156730817d290b1

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

LOG: [mlir][Transform] Remove `notifyOperationErased` workaround (#84134)

D144193 (#66771) has been merged.

Added: 
    

Modified: 
    mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp b/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
index 71a9d61198e3fb..fe2eea535ffdcf 100644
--- a/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
+++ b/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
@@ -1278,14 +1278,11 @@ void transform::TrackingListener::notifyMatchFailure(
 }
 
 void transform::TrackingListener::notifyOperationErased(Operation *op) {
-  // TODO: Walk can be removed when D144193 has landed.
-  op->walk([&](Operation *op) {
-    // Remove mappings for result values.
-    for (OpResult value : op->getResults())
-      (void)replacePayloadValue(value, nullptr);
-    // Remove mapping for op.
-    (void)replacePayloadOp(op, nullptr);
-  });
+  // Remove mappings for result values.
+  for (OpResult value : op->getResults())
+    (void)replacePayloadValue(value, nullptr);
+  // Remove mapping for op.
+  (void)replacePayloadOp(op, nullptr);
 }
 
 void transform::TrackingListener::notifyOperationReplaced(