2996a8d675
MLIRContext allows its users to access directly to the DialectRegistry it contains. While sometimes useful for registering additional dialects on an already existing context, this breaks the encapsulation by essentially giving raw accesses to a part of the context's internal state. Remove this mutable access and instead provide a method to append a given DialectRegistry to the one already contained in the context. Also provide a shortcut mechanism to construct a context from an already existing registry, which seems to be a common use case in the wild. Keep read-only access to the registry contained in the context in case it needs to be copied or used for constructing another context. With this change, DialectRegistry is no longer concerned with loading the dialects and deciding whether to invoke delayed interface registration. Loading is concentrated in the MLIRContext, and the functionality of the registry better reflects its name. Depends On D96137 Reviewed By: mehdi_amini Differential Revision: https://reviews.llvm.org/D96331
138 lines
4.4 KiB
C++
138 lines
4.4 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 <vector>
|
|
|
|
#include "mlir/InitAllDialects.h"
|
|
#include "mlir/Parser.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Reducer/OptReductionPass.h"
|
|
#include "mlir/Reducer/Passes/OpReducer.h"
|
|
#include "mlir/Reducer/ReductionNode.h"
|
|
#include "mlir/Reducer/ReductionTreePass.h"
|
|
#include "mlir/Reducer/Tester.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Support/LogicalResult.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestDialect(DialectRegistry &);
|
|
} // namespace test
|
|
} // namespace mlir
|
|
|
|
static llvm::cl::opt<std::string> inputFilename(llvm::cl::Positional,
|
|
llvm::cl::Required,
|
|
llvm::cl::desc("<input file>"));
|
|
|
|
static llvm::cl::opt<std::string>
|
|
testFilename("test", llvm::cl::Required, llvm::cl::desc("Testing script"));
|
|
|
|
static llvm::cl::list<std::string>
|
|
testArguments("test-args", llvm::cl::ZeroOrMore,
|
|
llvm::cl::desc("Testing script arguments"));
|
|
|
|
static llvm::cl::opt<std::string>
|
|
outputFilename("o",
|
|
llvm::cl::desc("Output filename for the reduced test case"),
|
|
llvm::cl::init("-"));
|
|
|
|
// TODO: Use PassPipelineCLParser to define pass pieplines in the command line.
|
|
static llvm::cl::opt<std::string>
|
|
passTestSpecifier("pass-test",
|
|
llvm::cl::desc("Indicate a specific pass to be tested"));
|
|
|
|
// Parse and verify the input MLIR file.
|
|
static LogicalResult loadModule(MLIRContext &context, OwningModuleRef &module,
|
|
StringRef inputFilename) {
|
|
module = parseSourceFile(inputFilename, &context);
|
|
if (!module)
|
|
return failure();
|
|
|
|
return success();
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
llvm::InitLLVM y(argc, argv);
|
|
|
|
registerMLIRContextCLOptions();
|
|
registerPassManagerCLOptions();
|
|
|
|
llvm::cl::ParseCommandLineOptions(argc, argv,
|
|
"MLIR test case reduction tool.\n");
|
|
|
|
std::string errorMessage;
|
|
|
|
auto testscript = openInputFile(testFilename, &errorMessage);
|
|
if (!testscript)
|
|
llvm::report_fatal_error(errorMessage);
|
|
|
|
auto output = openOutputFile(outputFilename, &errorMessage);
|
|
if (!output)
|
|
llvm::report_fatal_error(errorMessage);
|
|
|
|
mlir::DialectRegistry registry;
|
|
registerAllDialects(registry);
|
|
#ifdef MLIR_INCLUDE_TESTS
|
|
mlir::test::registerTestDialect(registry);
|
|
#endif
|
|
mlir::MLIRContext context(registry);
|
|
|
|
mlir::OwningModuleRef moduleRef;
|
|
if (failed(loadModule(context, moduleRef, inputFilename)))
|
|
llvm::report_fatal_error("Input test case can't be parsed");
|
|
|
|
// Initialize test environment.
|
|
const Tester test(testFilename, testArguments);
|
|
|
|
if (!test.isInteresting(inputFilename))
|
|
llvm::report_fatal_error(
|
|
"Input test case does not exhibit interesting behavior");
|
|
|
|
// Reduction pass pipeline.
|
|
PassManager pm(&context);
|
|
|
|
if (passTestSpecifier == "DCE") {
|
|
|
|
// Opt Reduction Pass with SymbolDCEPass as opt pass.
|
|
pm.addPass(std::make_unique<OptReductionPass>(test, &context,
|
|
createSymbolDCEPass()));
|
|
|
|
} else if (passTestSpecifier == "function-reducer") {
|
|
|
|
// Reduction tree pass with OpReducer variant generation and single path
|
|
// traversal.
|
|
pm.addPass(
|
|
std::make_unique<ReductionTreePass<OpReducer<FuncOp>, SinglePath>>(
|
|
test));
|
|
}
|
|
|
|
ModuleOp m = moduleRef.get().clone();
|
|
|
|
if (failed(pm.run(m)))
|
|
llvm::report_fatal_error("Error running the reduction pass pipeline");
|
|
|
|
m.print(output->os());
|
|
output->keep();
|
|
|
|
return 0;
|
|
}
|