5fcf907b34
This commit renames 4 pattern rewriter API functions: * `updateRootInPlace` -> `modifyOpInPlace` * `startRootUpdate` -> `startOpModification` * `finalizeRootUpdate` -> `finalizeOpModification` * `cancelRootUpdate` -> `cancelOpModification` The term "root" is a misnomer. The root is the op that a rewrite pattern matches against (https://mlir.llvm.org/docs/PatternRewriter/#root-operation-name-optional). A rewriter must be notified of all in-place op modifications, not just in-place modifications of the root (https://mlir.llvm.org/docs/PatternRewriter/#pattern-rewriter). The old function names were confusing and have contributed to various broken rewrite patterns. Note: The new function names use the term "modify" instead of "update" for consistency with the `RewriterBase::Listener` terminology (`notifyOperationModified`).
49 lines
1.7 KiB
C++
49 lines
1.7 KiB
C++
//===- StandalonePasses.cpp - Standalone passes -----------------*- C++ -*-===//
|
|
//
|
|
// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "mlir/Dialect/Func/IR/FuncOps.h"
|
|
#include "mlir/IR/PatternMatch.h"
|
|
#include "mlir/Rewrite/FrozenRewritePatternSet.h"
|
|
#include "mlir/Support/LogicalResult.h"
|
|
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
|
|
|
|
#include "Standalone/StandalonePasses.h"
|
|
|
|
namespace mlir::standalone {
|
|
#define GEN_PASS_DEF_STANDALONESWITCHBARFOO
|
|
#include "Standalone/StandalonePasses.h.inc"
|
|
|
|
namespace {
|
|
class StandaloneSwitchBarFooRewriter : public OpRewritePattern<func::FuncOp> {
|
|
public:
|
|
using OpRewritePattern<func::FuncOp>::OpRewritePattern;
|
|
LogicalResult matchAndRewrite(func::FuncOp op,
|
|
PatternRewriter &rewriter) const final {
|
|
if (op.getSymName() == "bar") {
|
|
rewriter.modifyOpInPlace(op, [&op]() { op.setSymName("foo"); });
|
|
return success();
|
|
}
|
|
return failure();
|
|
}
|
|
};
|
|
|
|
class StandaloneSwitchBarFoo
|
|
: public impl::StandaloneSwitchBarFooBase<StandaloneSwitchBarFoo> {
|
|
public:
|
|
using impl::StandaloneSwitchBarFooBase<
|
|
StandaloneSwitchBarFoo>::StandaloneSwitchBarFooBase;
|
|
void runOnOperation() final {
|
|
RewritePatternSet patterns(&getContext());
|
|
patterns.add<StandaloneSwitchBarFooRewriter>(&getContext());
|
|
FrozenRewritePatternSet patternSet(std::move(patterns));
|
|
if (failed(applyPatternsAndFoldGreedily(getOperation(), patternSet)))
|
|
signalPassFailure();
|
|
}
|
|
};
|
|
} // namespace
|
|
} // namespace mlir::standalone
|