forked from OSchip/llvm-project
57 lines
2.1 KiB
C++
57 lines
2.1 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"
|
|
|
|
#define DEBUG_TYPE "mlir-reduce"
|
|
|
|
using namespace mlir;
|
|
|
|
OptReductionPass::OptReductionPass(const Tester &test, MLIRContext *context,
|
|
std::unique_ptr<Pass> optPass)
|
|
: context(context), test(test), optPass(std::move(optPass)) {}
|
|
|
|
OptReductionPass::OptReductionPass(const OptReductionPass &srcPass)
|
|
: OptReductionBase<OptReductionPass>(srcPass), test(srcPass.test),
|
|
optPass(srcPass.optPass.get()) {}
|
|
|
|
/// Runs the pass instance in the pass pipeline.
|
|
void OptReductionPass::runOnOperation() {
|
|
LLVM_DEBUG(llvm::dbgs() << "\nOptimization Reduction pass: ");
|
|
LLVM_DEBUG(llvm::dbgs() << optPass.get()->getName() << "\nTesting:\n");
|
|
|
|
ModuleOp module = this->getOperation();
|
|
ModuleOp moduleVariant = module.clone();
|
|
PassManager pmTransform(context);
|
|
pmTransform.addPass(std::move(optPass));
|
|
|
|
if (failed(pmTransform.run(moduleVariant)))
|
|
return;
|
|
|
|
ReductionNode original(module, nullptr);
|
|
original.measureAndTest(test);
|
|
|
|
ReductionNode reduced(moduleVariant, nullptr);
|
|
reduced.measureAndTest(test);
|
|
|
|
if (reduced.isInteresting() && reduced.getSize() < original.getSize()) {
|
|
ReductionTreeUtils::updateGoldenModule(module, reduced.getModule().clone());
|
|
LLVM_DEBUG(llvm::dbgs() << "\nSuccessful Transformed version\n\n");
|
|
} else {
|
|
LLVM_DEBUG(llvm::dbgs() << "\nUnsuccessful Transformed version\n\n");
|
|
}
|
|
|
|
LLVM_DEBUG(llvm::dbgs() << "Pass Complete\n\n");
|
|
}
|