forked from OSchip/llvm-project
84 lines
3.0 KiB
C++
84 lines
3.0 KiB
C++
//===- mlir-tblgen.cpp - Top-Level TableGen implementation for MLIR -------===//
|
|
//
|
|
// 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 contains the main function for MLIR's TableGen.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/TableGen/GenInfo.h"
|
|
#include "mlir/TableGen/GenNameParser.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/FormatVariadic.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include "llvm/TableGen/Error.h"
|
|
#include "llvm/TableGen/Main.h"
|
|
#include "llvm/TableGen/Record.h"
|
|
#include "llvm/TableGen/TableGenBackend.h"
|
|
|
|
using namespace llvm;
|
|
using namespace mlir;
|
|
|
|
static llvm::ManagedStatic<std::vector<GenInfo>> generatorRegistry;
|
|
|
|
mlir::GenRegistration::GenRegistration(StringRef arg, StringRef description,
|
|
GenFunction function) {
|
|
generatorRegistry->emplace_back(arg, description, function);
|
|
}
|
|
|
|
GenNameParser::GenNameParser(llvm::cl::Option &opt)
|
|
: llvm::cl::parser<const GenInfo *>(opt) {
|
|
for (const auto &kv : *generatorRegistry) {
|
|
addLiteralOption(kv.getGenArgument(), &kv, kv.getGenDescription());
|
|
}
|
|
}
|
|
|
|
void GenNameParser::printOptionInfo(const llvm::cl::Option &o,
|
|
size_t globalWidth) const {
|
|
GenNameParser *tp = const_cast<GenNameParser *>(this);
|
|
llvm::array_pod_sort(tp->Values.begin(), tp->Values.end(),
|
|
[](const GenNameParser::OptionInfo *vT1,
|
|
const GenNameParser::OptionInfo *vT2) {
|
|
return vT1->Name.compare(vT2->Name);
|
|
});
|
|
using llvm::cl::parser;
|
|
parser<const GenInfo *>::printOptionInfo(o, globalWidth);
|
|
}
|
|
|
|
// Generator that prints records.
|
|
GenRegistration printRecords("print-records", "Print all records to stdout",
|
|
[](const RecordKeeper &records, raw_ostream &os) {
|
|
os << records;
|
|
return false;
|
|
});
|
|
|
|
// Generator to invoke.
|
|
const mlir::GenInfo *generator;
|
|
|
|
// TableGenMain requires a function pointer so this function is passed in which
|
|
// simply wraps the call to the generator.
|
|
static bool mlirTableGenMain(raw_ostream &os, RecordKeeper &records) {
|
|
if (!generator) {
|
|
os << records;
|
|
return false;
|
|
}
|
|
return generator->invoke(records, os);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
llvm::InitLLVM y(argc, argv);
|
|
llvm::cl::opt<const mlir::GenInfo *, false, mlir::GenNameParser> generator(
|
|
"", llvm::cl::desc("Generator to run"));
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
::generator = generator.getValue();
|
|
|
|
return TableGenMain(argv[0], &mlirTableGenMain);
|
|
}
|