1 //===- mlir-reduce.cpp - The MLIR reducer ---------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the general framework of the MLIR reducer tool. It
10 // parses the command line arguments, parses the initial MLIR test case and sets
11 // up the testing environment. It outputs the most reduced test case variant
12 // after executing the reduction passes.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #include "mlir/Tools/mlir-reduce/MlirReduceMain.h"
17 #include "mlir/IR/PatternMatch.h"
18 #include "mlir/Parser/Parser.h"
19 #include "mlir/Pass/Pass.h"
20 #include "mlir/Pass/PassManager.h"
21 #include "mlir/Reducer/Passes.h"
22 #include "mlir/Rewrite/FrozenRewritePatternSet.h"
23 #include "mlir/Support/FileUtilities.h"
24 #include "mlir/Tools/ParseUtilities.h"
25 #include "llvm/Support/InitLLVM.h"
26 #include "llvm/Support/SourceMgr.h"
27 #include "llvm/Support/ToolOutputFile.h"
28
29 using namespace mlir;
30
31 // Parse and verify the input MLIR file. Returns null on error.
loadModule(MLIRContext & context,StringRef inputFilename,bool insertImplictModule)32 OwningOpRef<Operation *> loadModule(MLIRContext &context,
33 StringRef inputFilename,
34 bool insertImplictModule) {
35 // Set up the input file.
36 std::string errorMessage;
37 auto file = openInputFile(inputFilename, &errorMessage);
38 if (!file) {
39 llvm::errs() << errorMessage << "\n";
40 return nullptr;
41 }
42
43 auto sourceMgr = std::make_shared<llvm::SourceMgr>();
44 sourceMgr->AddNewSourceBuffer(std::move(file), SMLoc());
45 return parseSourceFileForTool(sourceMgr, &context, insertImplictModule);
46 }
47
mlirReduceMain(int argc,char ** argv,MLIRContext & context)48 LogicalResult mlir::mlirReduceMain(int argc, char **argv,
49 MLIRContext &context) {
50 // Override the default '-h' and use the default PrintHelpMessage() which
51 // won't print options in categories.
52 static llvm::cl::opt<bool> help("h", llvm::cl::desc("Alias for -help"),
53 llvm::cl::Hidden);
54
55 static llvm::cl::OptionCategory mlirReduceCategory("mlir-reduce options");
56
57 static llvm::cl::opt<std::string> inputFilename(
58 llvm::cl::Positional, llvm::cl::desc("<input file>"),
59 llvm::cl::cat(mlirReduceCategory));
60
61 static llvm::cl::opt<std::string> outputFilename(
62 "o", llvm::cl::desc("Output filename for the reduced test case"),
63 llvm::cl::init("-"), llvm::cl::cat(mlirReduceCategory));
64
65 static llvm::cl::opt<bool> noImplicitModule{
66 "no-implicit-module",
67 llvm::cl::desc(
68 "Disable implicit addition of a top-level module op during parsing"),
69 llvm::cl::init(false)};
70
71 llvm::cl::HideUnrelatedOptions(mlirReduceCategory);
72
73 llvm::InitLLVM y(argc, argv);
74
75 registerReducerPasses();
76
77 PassPipelineCLParser parser("", "Reduction Passes to Run");
78 llvm::cl::ParseCommandLineOptions(argc, argv,
79 "MLIR test case reduction tool.\n");
80
81 if (help) {
82 llvm::cl::PrintHelpMessage();
83 return success();
84 }
85
86 std::string errorMessage;
87
88 auto output = openOutputFile(outputFilename, &errorMessage);
89 if (!output)
90 return failure();
91
92 OwningOpRef<Operation *> opRef =
93 loadModule(context, inputFilename, !noImplicitModule);
94 if (!opRef)
95 return failure();
96
97 auto errorHandler = [&](const Twine &msg) {
98 return emitError(UnknownLoc::get(&context)) << msg;
99 };
100
101 // Reduction pass pipeline.
102 PassManager pm(&context, opRef.get()->getName().getStringRef());
103 if (failed(parser.addToPipeline(pm, errorHandler)))
104 return failure();
105
106 OwningOpRef<Operation *> op = opRef.get()->clone();
107
108 if (failed(pm.run(op.get())))
109 return failure();
110
111 op.get()->print(output->os());
112 output->keep();
113
114 return success();
115 }
116