60 lines
1.8 KiB
C++
60 lines
1.8 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/IR/Operation.h"
|
|
#include "mlir/IR/OwningOpRef.h"
|
|
#include "mlir/Parser/Parser.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "mlir/Support/LLVM.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <cstdlib>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
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";
|
|
}
|