2a65d0388c
This is an alternate approach to the patches proposed in D153897 and D153794. Rather than exporting a single header that can be included on the GPU in all circumstances, this patch chooses to instead generate a separate set of headers that only provides the declarations. This can then be used by external tooling to set up what's on the GPU. This leaves room for header hacks for offloading languages without needing to worry about the `libc` implementation. Currently this generates a set of headers that only contain the declarations. These will then be installed to a new clang resource directory called `llvm_libc_wrappers/` which will house the shim code. We can then automaticlaly include this from `clang` when offloading to wrap around the headers while specifying what's on the GPU. Reviewed By: jdoerfert, JonChesterfield Differential Revision: https://reviews.llvm.org/D154036
68 lines
2.2 KiB
C++
68 lines
2.2 KiB
C++
//===-- "main" function of libc-hdrgen ------------------------------------===//
|
|
//
|
|
// 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 "Generator.h"
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/TableGen/Main.h"
|
|
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
namespace {
|
|
|
|
llvm::cl::opt<std::string>
|
|
HeaderDefFile("def", llvm::cl::desc("Path to the .h.def file."),
|
|
llvm::cl::value_desc("<filename>"), llvm::cl::Required);
|
|
llvm::cl::opt<std::string> StandardHeader(
|
|
"header",
|
|
llvm::cl::desc("The standard header file which is to be generated."),
|
|
llvm::cl::value_desc("<header file>"));
|
|
llvm::cl::list<std::string> EntrypointNamesOption(
|
|
"e", llvm::cl::value_desc("<list of entrypoints>"),
|
|
llvm::cl::desc(
|
|
"Each --e is one entrypoint (generated from entrypoints.txt)"),
|
|
llvm::cl::OneOrMore);
|
|
llvm::cl::list<std::string> ReplacementValues(
|
|
"args", llvm::cl::desc("Command separated <argument name>=<value> pairs."),
|
|
llvm::cl::value_desc("<name=value>[,name=value]"));
|
|
llvm::cl::opt<bool> ExportDecls(
|
|
"export-decls",
|
|
llvm::cl::desc("Output a new header containing only the entrypoints."));
|
|
|
|
void ParseArgValuePairs(std::unordered_map<std::string, std::string> &Map) {
|
|
for (std::string &R : ReplacementValues) {
|
|
auto Pair = llvm::StringRef(R).split('=');
|
|
Map[std::string(Pair.first)] = std::string(Pair.second);
|
|
}
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
namespace llvm_libc {
|
|
|
|
bool HeaderGeneratorMain(llvm::raw_ostream &OS, llvm::RecordKeeper &Records) {
|
|
std::unordered_map<std::string, std::string> ArgMap;
|
|
ParseArgValuePairs(ArgMap);
|
|
Generator G(HeaderDefFile, EntrypointNamesOption, StandardHeader, ArgMap);
|
|
if (ExportDecls)
|
|
G.generateDecls(OS, Records);
|
|
else
|
|
G.generate(OS, Records);
|
|
|
|
return false;
|
|
}
|
|
|
|
} // namespace llvm_libc
|
|
|
|
int main(int argc, char *argv[]) {
|
|
llvm::cl::ParseCommandLineOptions(argc, argv);
|
|
return TableGenMain(argv[0], &llvm_libc::HeaderGeneratorMain);
|
|
}
|