e7f8b45953
These may serve as example and showcase of the MLIR binary footprint. Right now a release build of these tools on a linux machine shows: - mlir-cat: 2MB This includes the Core IR, the textual parser/printer, the support for bytecode. - mlir-minimal-opt: 3MB This adds all the tooling for an mlir-opt tool: the pass infrastructure and all the instrumentation associated with it. - mlir-miminal-opt-canonicalize: 4.8MB This add the canonicalizer pass, which pulls in all the pattern/rewrite machinery, including the PDL compiler and intepreter. Differential Revision: https://reviews.llvm.org/D156218
58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
//===- mlir-cat.cpp ---------------------------------------------*- C++ -*-===//
|
|
//
|
|
// This file is licensed 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/IR/MLIRContext.h"
|
|
#include "mlir/Parser/Parser.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Support/LLVM.h"
|
|
#include "mlir/Support/LogicalResult.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace mlir;
|
|
|
|
/// This example parse its input, either from a file or its standard input (in
|
|
/// bytecode or textual assembly) and print it back.
|
|
|
|
int main(int argc, char **argv) {
|
|
// Set up the input file.
|
|
StringRef inputFile;
|
|
if (argc <= 1) {
|
|
llvm::errs() << "Reading from stdin...\n";
|
|
inputFile = "-";
|
|
} else {
|
|
inputFile = argv[1];
|
|
}
|
|
std::string errorMessage;
|
|
auto file = openInputFile(inputFile, &errorMessage);
|
|
if (!file) {
|
|
llvm::errs() << errorMessage << "\n";
|
|
exit(1);
|
|
}
|
|
llvm::SourceMgr sourceMgr;
|
|
sourceMgr.AddNewSourceBuffer(std::move(file), SMLoc());
|
|
|
|
auto output = openOutputFile("-", &errorMessage);
|
|
if (!output) {
|
|
llvm::errs() << errorMessage << "\n";
|
|
exit(1);
|
|
}
|
|
|
|
DialectRegistry registry;
|
|
MLIRContext context(registry, MLIRContext::Threading::DISABLED);
|
|
context.allowUnregisteredDialects(true);
|
|
OwningOpRef<Operation *> op = parseSourceFile(sourceMgr, &context);
|
|
if (!op) {
|
|
llvm::errs() << "Failed to parse input file";
|
|
exit(1);
|
|
}
|
|
output->os() << *(op.get()) << "\n";
|
|
}
|