intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-16 21:55:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir] specify the values when notifying about op replacement

Browse Source 
It is useful for PatternRewriter listeners to know the values that are
replacing the op in addition to only the fact of the op being replaced
for being able to keep track of changes or for debugging.

Reviewed By: Mogball

Differential Revision: https://reviews.llvm.org/D134748
This commit is contained in:
Alex Zinenko
2022-09-27 16:08:44 +00:00
parent 401481daac
commit e8aaf75810
3 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlir/include/mlir/IR/PatternMatch.h
View File

@@ -546,9 +546,9 @@ protected:
/// they would like to be notified about certain types of mutations.
/// Notify the rewriter that the specified operation is about to be replaced
/// with another set of operations. This is called before the uses of the
/// operation have been changed.
virtual void notifyRootReplaced(Operation *op) {}
/// with the set of values potentially produced by new operations. This is
/// called before the uses of the operation have been changed.
virtual void notifyRootReplaced(Operation *op, ValueRange replacement) {}
/// This is called on an operation that a rewrite is removing, right before
/// the operation is deleted. At this point, the operation has zero uses.

4
mlir/lib/IR/PatternMatch.cpp
View File

@@ -216,7 +216,7 @@ void RewriterBase::replaceOpWithIf(
"incorrect number of values to replace operation");
// Notify the rewriter subclass that we're about to replace this root.
notifyRootReplaced(op);
notifyRootReplaced(op, newValues);
// Replace each use of the results when the functor is true.
bool replacedAllUses = true;
@@ -244,7 +244,7 @@ void RewriterBase::replaceOpWithinBlock(Operation *op, ValueRange newValues,
/// the operation.
void RewriterBase::replaceOp(Operation *op, ValueRange newValues) {
// Notify the rewriter subclass that we're about to replace this root.
notifyRootReplaced(op);
notifyRootReplaced(op, newValues);
assert(op->getNumResults() == newValues.size() &&
"incorrect # of replacement values");

7
mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp
View File

@@ -69,7 +69,7 @@ protected:
// When the root of a pattern is about to be replaced, it can trigger
// simplifications to its users - make sure to add them to the worklist
// before the root is changed.
void notifyRootReplaced(Operation *op) override;
void notifyRootReplaced(Operation *op, ValueRange replacement) override;
/// PatternRewriter hook for erasing a dead operation.
void eraseOp(Operation *op) override;
@@ -348,7 +348,8 @@ void GreedyPatternRewriteDriver::notifyOperationRemoved(Operation *op) {
});
}
void GreedyPatternRewriteDriver::notifyRootReplaced(Operation *op) {
void GreedyPatternRewriteDriver::notifyRootReplaced(Operation *op,
ValueRange replacement) {
LLVM_DEBUG({
logger.startLine() << "** Replace : '" << op->getName() << "'(" << op
<< ")\n";
@@ -437,7 +438,7 @@ protected:
// When a root is going to be replaced, its removal will be notified as well.
// So there is nothing to do here.
void notifyRootReplaced(Operation *op) override {}
void notifyRootReplaced(Operation *op, ValueRange replacement) override {}
private:
/// The low-level pattern applicator.