cd1acf2ae3
The Canonicalizer pass has a dependency to UB dialect which shouldn't have. It also no longer needs to directly depend on the UB dialect since the Vector dialect (which uses UB dialect for poison index operations introduced by35df525) already declares this dependency(878d3594).
86 lines
3.3 KiB
C++
86 lines
3.3 KiB
C++
//===- Canonicalizer.cpp - Canonicalize MLIR operations -------------------===//
|
|
//
|
|
// Part of the LLVM Project, 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This transformation pass converts operations into their canonical forms by
|
|
// folding constants, applying operation identity transformations etc.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Transforms/Passes.h"
|
|
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
|
|
|
|
namespace mlir {
|
|
#define GEN_PASS_DEF_CANONICALIZER
|
|
#include "mlir/Transforms/Passes.h.inc"
|
|
} // namespace mlir
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
/// Canonicalize operations in nested regions.
|
|
struct Canonicalizer : public impl::CanonicalizerBase<Canonicalizer> {
|
|
Canonicalizer() = default;
|
|
Canonicalizer(const GreedyRewriteConfig &config,
|
|
ArrayRef<std::string> disabledPatterns,
|
|
ArrayRef<std::string> enabledPatterns)
|
|
: config(config) {
|
|
this->topDownProcessingEnabled = config.getUseTopDownTraversal();
|
|
this->regionSimplifyLevel = config.getRegionSimplificationLevel();
|
|
this->maxIterations = config.getMaxIterations();
|
|
this->maxNumRewrites = config.getMaxNumRewrites();
|
|
this->disabledPatterns = disabledPatterns;
|
|
this->enabledPatterns = enabledPatterns;
|
|
}
|
|
|
|
/// Initialize the canonicalizer by building the set of patterns used during
|
|
/// execution.
|
|
LogicalResult initialize(MLIRContext *context) override {
|
|
// Set the config from possible pass options set in the meantime.
|
|
config.setUseTopDownTraversal(topDownProcessingEnabled);
|
|
config.setRegionSimplificationLevel(regionSimplifyLevel);
|
|
config.setMaxIterations(maxIterations);
|
|
config.setMaxNumRewrites(maxNumRewrites);
|
|
|
|
RewritePatternSet owningPatterns(context);
|
|
for (auto *dialect : context->getLoadedDialects())
|
|
dialect->getCanonicalizationPatterns(owningPatterns);
|
|
for (RegisteredOperationName op : context->getRegisteredOperations())
|
|
op.getCanonicalizationPatterns(owningPatterns, context);
|
|
|
|
patterns = std::make_shared<FrozenRewritePatternSet>(
|
|
std::move(owningPatterns), disabledPatterns, enabledPatterns);
|
|
return success();
|
|
}
|
|
void runOnOperation() override {
|
|
LogicalResult converged =
|
|
applyPatternsGreedily(getOperation(), *patterns, config);
|
|
// Canonicalization is best-effort. Non-convergence is not a pass failure.
|
|
if (testConvergence && failed(converged))
|
|
signalPassFailure();
|
|
}
|
|
GreedyRewriteConfig config;
|
|
std::shared_ptr<const FrozenRewritePatternSet> patterns;
|
|
};
|
|
} // namespace
|
|
|
|
/// Create a Canonicalizer pass.
|
|
std::unique_ptr<Pass> mlir::createCanonicalizerPass() {
|
|
return std::make_unique<Canonicalizer>();
|
|
}
|
|
|
|
/// Creates an instance of the Canonicalizer pass with the specified config.
|
|
std::unique_ptr<Pass>
|
|
mlir::createCanonicalizerPass(const GreedyRewriteConfig &config,
|
|
ArrayRef<std::string> disabledPatterns,
|
|
ArrayRef<std::string> enabledPatterns) {
|
|
return std::make_unique<Canonicalizer>(config, disabledPatterns,
|
|
enabledPatterns);
|
|
}
|