llvm-project/mlir/tools/mlir-reduce/OptReductionPass.cpp

71 lines
2.4 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/Reducer/OptReductionPass.h"
#include "mlir/Pass/PassRegistry.h"
#include "mlir/Reducer/Passes.h"
#include "mlir/Reducer/Tester.h"
#define DEBUG_TYPE "mlir-reduce"
using namespace mlir;
/// Runs the pass instance in the pass pipeline.
void OptReductionPass::runOnOperation() {
LLVM_DEBUG(llvm::dbgs() << "\nOptimization Reduction pass: ");
Tester test(testerName, testerArgs);
ModuleOp module = this->getOperation();
ModuleOp moduleVariant = module.clone();
PassManager passManager(module.getContext());
if (failed(parsePassPipeline(optPass, passManager))) {
LLVM_DEBUG(llvm::dbgs() << "\nFailed to parse pass pipeline");
return;
}
std::pair<Tester::Interestingness, int> original = test.isInteresting(module);
if (original.first != Tester::Interestingness::True) {
LLVM_DEBUG(llvm::dbgs() << "\nThe original input is not interested");
return;
}
if (failed(passManager.run(moduleVariant))) {
LLVM_DEBUG(llvm::dbgs() << "\nFailed to run pass pipeline");
return;
}
std::pair<Tester::Interestingness, int> reduced =
test.isInteresting(moduleVariant);
if (reduced.first == Tester::Interestingness::True &&
reduced.second < original.second) {
module.getBody()->clear();
module.getBody()->getOperations().splice(
module.getBody()->begin(), moduleVariant.getBody()->getOperations());
LLVM_DEBUG(llvm::dbgs() << "\nSuccessful Transformed version\n\n");
} else {
LLVM_DEBUG(llvm::dbgs() << "\nUnsuccessful Transformed version\n\n");
}
moduleVariant->destroy();
LLVM_DEBUG(llvm::dbgs() << "Pass Complete\n\n");
}
std::unique_ptr<Pass> mlir::createOptReductionPass() {
return std::make_unique<OptReductionPass>();
}