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

[mlir][Transforms] GreedyPatternRewriteDriver: log successful folding (#77796)

Browse Source 
Similar to successful pattern applications, dump the rewritten IR after
each successful folding when running with `-debug`.
This commit is contained in:
Matthias Springer
2024-01-12 15:50:52 +01:00
committed by GitHub
parent 7700ea1031
commit dec908a285
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -136,6 +136,21 @@ protected:
};
#endif // MLIR_ENABLE_EXPENSIVE_PATTERN_API_CHECKS
#ifndef NDEBUG
static Operation *getDumpRootOp(Operation *op) {
// Dump the parent op so that materialized constants are visible. If the op
// is a top-level op, dump it directly.
if (Operation *parentOp = op->getParentOp())
return parentOp;
return op;
}
static void logSuccessfulFolding(Operation *op) {
llvm::dbgs() << "// *** IR Dump After Successful Folding ***\n";
op->dump();
llvm::dbgs() << "\n\n";
}
#endif // NDEBUG
//===----------------------------------------------------------------------===//
// Worklist
//===----------------------------------------------------------------------===//
@@ -434,10 +449,14 @@ bool GreedyPatternRewriteDriver::processWorklist() {
SmallVector<OpFoldResult> foldResults;
if (succeeded(op->fold(foldResults))) {
LLVM_DEBUG(logResultWithLine("success", "operation was folded"));
#ifndef NDEBUG
Operation *dumpRootOp = getDumpRootOp(op);
#endif // NDEBUG
if (foldResults.empty()) {
// Op was modified in-place.
notifyOperationModified(op);
changed = true;
LLVM_DEBUG(logSuccessfulFolding(dumpRootOp));
#if MLIR_ENABLE_EXPENSIVE_PATTERN_API_CHECKS
if (config.scope && failed(verify(config.scope->getParentOp())))
llvm::report_fatal_error("IR failed to verify after folding");
@@ -492,6 +511,7 @@ bool GreedyPatternRewriteDriver::processWorklist() {
if (materializationSucceeded) {
replaceOp(op, replacements);
changed = true;
LLVM_DEBUG(logSuccessfulFolding(dumpRootOp));
#if MLIR_ENABLE_EXPENSIVE_PATTERN_API_CHECKS
if (config.scope && failed(verify(config.scope->getParentOp())))
llvm::report_fatal_error("IR failed to verify after folding");