Skip to content

Commit 9f25fbb

Browse files
[mlir][Transform] Remove notifyOperationErased workaround
D144193 (#66771) has been merged.
1 parent 407c7f7 commit 9f25fbb

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

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

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1278,14 +1278,11 @@ void transform::TrackingListener::notifyMatchFailure(
12781278
}
12791279

12801280
void transform::TrackingListener::notifyOperationErased(Operation *op) {
1281-
// TODO: Walk can be removed when D144193 has landed.
1282-
op->walk([&](Operation *op) {
1283-
// Remove mappings for result values.
1284-
for (OpResult value : op->getResults())
1285-
(void)replacePayloadValue(value, nullptr);
1286-
// Remove mapping for op.
1287-
(void)replacePayloadOp(op, nullptr);
1288-
});
1281+
// Remove mappings for result values.
1282+
for (OpResult value : op->getResults())
1283+
(void)replacePayloadValue(value, nullptr);
1284+
// Remove mapping for op.
1285+
(void)replacePayloadOp(op, nullptr);
12891286
}
12901287

12911288
void transform::TrackingListener::notifyOperationReplaced(

0 commit comments

Comments
 (0)