d08bb68080
Currently, the opt-reduction-pass only supports inputting the optimization pipeline via the command line, which becomes cumbersome when the pipeline is long. To address this, this PR introduces the opt-pass-file option. This allows users to save the pipeline in a file and provide the filename to parse the pipeline.
102 lines
3.2 KiB
C++
102 lines
3.2 KiB
C++
//===- OptReductionPass.cpp - Optimization Reduction Pass Wrapper ---------===//
|
|
//
|
|
// 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 file defines the Opt Reduction Pass Wrapper. It creates a MLIR pass to
|
|
// run any optimization pass within it and only replaces the output module with
|
|
// the transformed version if it is smaller and interesting.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Pass/PassRegistry.h"
|
|
#include "mlir/Reducer/Passes.h"
|
|
#include "mlir/Reducer/Tester.h"
|
|
|
|
#include "llvm/Support/DebugLog.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
namespace mlir {
|
|
#define GEN_PASS_DEF_OPTREDUCTIONPASS
|
|
#include "mlir/Reducer/Passes.h.inc"
|
|
} // namespace mlir
|
|
|
|
#define DEBUG_TYPE "mlir-reduce"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
|
|
class OptReductionPass : public impl::OptReductionPassBase<OptReductionPass> {
|
|
public:
|
|
using Base::Base;
|
|
|
|
/// Runs the pass instance in the pass pipeline.
|
|
void runOnOperation() override;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
/// Runs the pass instance in the pass pipeline.
|
|
void OptReductionPass::runOnOperation() {
|
|
LDBG() << "\nOptimization Reduction pass: ";
|
|
|
|
Tester test(testerName, testerArgs);
|
|
Operation *topOp = this->getOperation();
|
|
|
|
std::string pipelineStr = optPass;
|
|
if (pipelineStr.empty()) {
|
|
if (!optPassFile.empty()) {
|
|
auto fileOrErr = llvm::MemoryBuffer::getFile(optPassFile);
|
|
if (std::error_code ec = fileOrErr.getError()) {
|
|
topOp->emitError() << "Could not open pass pipeline file: "
|
|
<< optPassFile << " (" << ec.message() << ")";
|
|
return signalPassFailure();
|
|
}
|
|
pipelineStr = fileOrErr.get()->getBuffer().trim().str();
|
|
}
|
|
}
|
|
|
|
PassManager passManager(topOp->getName());
|
|
if (failed(parsePassPipeline(pipelineStr, passManager))) {
|
|
topOp->emitError() << "\nfailed to parse pass pipeline";
|
|
return signalPassFailure();
|
|
}
|
|
|
|
std::pair<Tester::Interestingness, int> original = test.isInteresting(topOp);
|
|
if (original.first != Tester::Interestingness::True) {
|
|
topOp->emitError() << "\nthe original input is not interested";
|
|
return signalPassFailure();
|
|
}
|
|
Operation *topOpVariant = topOp->clone();
|
|
|
|
LogicalResult pipelineResult = passManager.run(topOpVariant);
|
|
if (failed(pipelineResult)) {
|
|
topOp->emitError() << "\nfailed to run pass pipeline";
|
|
return signalPassFailure();
|
|
}
|
|
|
|
std::pair<Tester::Interestingness, int> reduced =
|
|
test.isInteresting(topOpVariant);
|
|
|
|
if (reduced.first == Tester::Interestingness::True &&
|
|
reduced.second < original.second) {
|
|
topOp->getRegion(0).getBlocks().clear();
|
|
topOp->getRegion(0).getBlocks().splice(
|
|
topOp->getRegion(0).getBlocks().begin(),
|
|
topOpVariant->getRegion(0).getBlocks());
|
|
|
|
LDBG() << "\nSuccessful Transformed version\n";
|
|
} else {
|
|
LDBG() << "\nUnsuccessful Transformed version\n";
|
|
}
|
|
|
|
topOpVariant->destroy();
|
|
|
|
LDBG() << "Pass Complete\n";
|
|
}
|