18546ff8dd
The bytecode reader currently has no mechanism that allows for directly referencing data from the input buffer safely. This commit adds shared_ptr<SourceMgr> overloads that provide an explicit and safe way of extending the lifetime of the input. The usage of these new overloads is adopted in all of our tooling, and is implicitly used in the filename only parser methods. Differential Revision: https://reviews.llvm.org/D139366
117 lines
3.9 KiB
C++
117 lines
3.9 KiB
C++
//===- mlir-reduce.cpp - The MLIR reducer ---------------------------------===//
|
|
//
|
|
// 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 implements the general framework of the MLIR reducer tool. It
|
|
// parses the command line arguments, parses the initial MLIR test case and sets
|
|
// up the testing environment. It outputs the most reduced test case variant
|
|
// after executing the reduction passes.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Tools/mlir-reduce/MlirReduceMain.h"
|
|
#include "mlir/IR/PatternMatch.h"
|
|
#include "mlir/Parser/Parser.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Reducer/Passes.h"
|
|
#include "mlir/Rewrite/FrozenRewritePatternSet.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Support/LogicalResult.h"
|
|
#include "mlir/Tools/ParseUtilities.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
using namespace mlir;
|
|
|
|
// Parse and verify the input MLIR file. Returns null on error.
|
|
OwningOpRef<Operation *> loadModule(MLIRContext &context,
|
|
StringRef inputFilename,
|
|
bool insertImplictModule) {
|
|
// Set up the input file.
|
|
std::string errorMessage;
|
|
auto file = openInputFile(inputFilename, &errorMessage);
|
|
if (!file) {
|
|
llvm::errs() << errorMessage << "\n";
|
|
return nullptr;
|
|
}
|
|
|
|
auto sourceMgr = std::make_shared<llvm::SourceMgr>();
|
|
sourceMgr->AddNewSourceBuffer(std::move(file), SMLoc());
|
|
return parseSourceFileForTool(sourceMgr, &context, insertImplictModule);
|
|
}
|
|
|
|
LogicalResult mlir::mlirReduceMain(int argc, char **argv,
|
|
MLIRContext &context) {
|
|
// Override the default '-h' and use the default PrintHelpMessage() which
|
|
// won't print options in categories.
|
|
static llvm::cl::opt<bool> help("h", llvm::cl::desc("Alias for -help"),
|
|
llvm::cl::Hidden);
|
|
|
|
static llvm::cl::OptionCategory mlirReduceCategory("mlir-reduce options");
|
|
|
|
static llvm::cl::opt<std::string> inputFilename(
|
|
llvm::cl::Positional, llvm::cl::desc("<input file>"),
|
|
llvm::cl::cat(mlirReduceCategory));
|
|
|
|
static llvm::cl::opt<std::string> outputFilename(
|
|
"o", llvm::cl::desc("Output filename for the reduced test case"),
|
|
llvm::cl::init("-"), llvm::cl::cat(mlirReduceCategory));
|
|
|
|
static llvm::cl::opt<bool> noImplicitModule{
|
|
"no-implicit-module",
|
|
llvm::cl::desc(
|
|
"Disable implicit addition of a top-level module op during parsing"),
|
|
llvm::cl::init(false)};
|
|
|
|
llvm::cl::HideUnrelatedOptions(mlirReduceCategory);
|
|
|
|
llvm::InitLLVM y(argc, argv);
|
|
|
|
registerReducerPasses();
|
|
|
|
PassPipelineCLParser parser("", "Reduction Passes to Run");
|
|
llvm::cl::ParseCommandLineOptions(argc, argv,
|
|
"MLIR test case reduction tool.\n");
|
|
|
|
if (help) {
|
|
llvm::cl::PrintHelpMessage();
|
|
return success();
|
|
}
|
|
|
|
std::string errorMessage;
|
|
|
|
auto output = openOutputFile(outputFilename, &errorMessage);
|
|
if (!output)
|
|
return failure();
|
|
|
|
OwningOpRef<Operation *> opRef =
|
|
loadModule(context, inputFilename, !noImplicitModule);
|
|
if (!opRef)
|
|
return failure();
|
|
|
|
auto errorHandler = [&](const Twine &msg) {
|
|
return emitError(UnknownLoc::get(&context)) << msg;
|
|
};
|
|
|
|
// Reduction pass pipeline.
|
|
PassManager pm(&context, opRef.get()->getName().getStringRef());
|
|
if (failed(parser.addToPipeline(pm, errorHandler)))
|
|
return failure();
|
|
|
|
OwningOpRef<Operation *> op = opRef.get()->clone();
|
|
|
|
if (failed(pm.run(op.get())))
|
|
return failure();
|
|
|
|
op.get()->print(output->os());
|
|
output->keep();
|
|
|
|
return success();
|
|
}
|