diff --git a/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp b/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp --- a/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp +++ b/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp @@ -315,6 +315,8 @@ logger.startLine() << "** Modified: '" << op->getName() << "'(" << op << ")\n"; }); + if (config.listener) + config.listener->notifyOperationModified(op); addToWorklist(op); }