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
113 lines
4.2 KiB
C++
113 lines
4.2 KiB
C++
//===- MlirTranslateMain.cpp - MLIR Translation entry point ---------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Tools/mlir-translate/MlirTranslateMain.h"
|
|
#include "mlir/IR/AsmState.h"
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
#include "mlir/IR/Dialect.h"
|
|
#include "mlir/IR/Verifier.h"
|
|
#include "mlir/Parser/Parser.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Support/ToolUtilities.h"
|
|
#include "mlir/Tools/mlir-translate/Translation.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
using namespace mlir;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Translation Parser
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
LogicalResult mlir::mlirTranslateMain(int argc, char **argv,
|
|
llvm::StringRef toolName) {
|
|
|
|
static llvm::cl::opt<std::string> inputFilename(
|
|
llvm::cl::Positional, llvm::cl::desc("<input file>"),
|
|
llvm::cl::init("-"));
|
|
|
|
static llvm::cl::opt<std::string> outputFilename(
|
|
"o", llvm::cl::desc("Output filename"), llvm::cl::value_desc("filename"),
|
|
llvm::cl::init("-"));
|
|
|
|
static llvm::cl::opt<bool> allowUnregisteredDialects(
|
|
"allow-unregistered-dialect",
|
|
llvm::cl::desc("Allow operation with no registered dialects"),
|
|
llvm::cl::init(false));
|
|
|
|
static llvm::cl::opt<bool> splitInputFile(
|
|
"split-input-file",
|
|
llvm::cl::desc("Split the input file into pieces and "
|
|
"process each chunk independently"),
|
|
llvm::cl::init(false));
|
|
|
|
static llvm::cl::opt<bool> verifyDiagnostics(
|
|
"verify-diagnostics",
|
|
llvm::cl::desc("Check that emitted diagnostics match "
|
|
"expected-* lines on the corresponding line"),
|
|
llvm::cl::init(false));
|
|
|
|
llvm::InitLLVM y(argc, argv);
|
|
|
|
// Add flags for all the registered translations.
|
|
llvm::cl::opt<const Translation *, false, TranslationParser>
|
|
translationRequested("", llvm::cl::desc("Translation to perform"),
|
|
llvm::cl::Required);
|
|
registerAsmPrinterCLOptions();
|
|
registerMLIRContextCLOptions();
|
|
registerTranslationCLOptions();
|
|
llvm::cl::ParseCommandLineOptions(argc, argv, toolName);
|
|
|
|
std::string errorMessage;
|
|
std::unique_ptr<llvm::MemoryBuffer> input;
|
|
if (auto inputAlignment = translationRequested->getInputAlignment())
|
|
input = openInputFile(inputFilename, *inputAlignment, &errorMessage);
|
|
else
|
|
input = openInputFile(inputFilename, &errorMessage);
|
|
if (!input) {
|
|
llvm::errs() << errorMessage << "\n";
|
|
return failure();
|
|
}
|
|
|
|
auto output = openOutputFile(outputFilename, &errorMessage);
|
|
if (!output) {
|
|
llvm::errs() << errorMessage << "\n";
|
|
return failure();
|
|
}
|
|
|
|
// Processes the memory buffer with a new MLIRContext.
|
|
auto processBuffer = [&](std::unique_ptr<llvm::MemoryBuffer> ownedBuffer,
|
|
raw_ostream &os) {
|
|
MLIRContext context;
|
|
context.allowUnregisteredDialects(allowUnregisteredDialects);
|
|
context.printOpOnDiagnostic(!verifyDiagnostics);
|
|
auto sourceMgr = std::make_shared<llvm::SourceMgr>();
|
|
sourceMgr->AddNewSourceBuffer(std::move(ownedBuffer), SMLoc());
|
|
|
|
if (!verifyDiagnostics) {
|
|
SourceMgrDiagnosticHandler sourceMgrHandler(*sourceMgr, &context);
|
|
return (*translationRequested)(sourceMgr, os, &context);
|
|
}
|
|
|
|
// In the diagnostic verification flow, we ignore whether the translation
|
|
// failed (in most cases, it is expected to fail). Instead, we check if the
|
|
// diagnostics were produced as expected.
|
|
SourceMgrDiagnosticVerifierHandler sourceMgrHandler(*sourceMgr, &context);
|
|
(void)(*translationRequested)(sourceMgr, os, &context);
|
|
return sourceMgrHandler.verify();
|
|
};
|
|
|
|
if (failed(splitAndProcessBuffer(std::move(input), processBuffer,
|
|
output->os(), splitInputFile)))
|
|
return failure();
|
|
|
|
output->keep();
|
|
return success();
|
|
}
|