1532ee6916
A new ErrorCode enumeration is present in PluginInterface which can be used when returning an llvm::Error from offload and PluginInterface functions. This enum must be kept up to sync with liboffload's ol_errc_t enum, so both are automatically generated from liboffload's enum definition. Some error codes have also been shuffled around to allow for future work. Note that this patch only adds the machinery; actual error codes will be added in a future patch. ~~Depends on #137339 , please ignore first commit of this MR.~~ This has been merged.
96 lines
3.0 KiB
C++
96 lines
3.0 KiB
C++
//===- offload-tblgen/APIGen.cpp - Tablegen backend for Offload functions -===//
|
|
//
|
|
// 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 is a Tablegen backend that handles generation of various small files
|
|
// pertaining to the API functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/FormatVariadic.h"
|
|
#include "llvm/TableGen/Record.h"
|
|
|
|
#include "GenCommon.hpp"
|
|
#include "RecordTypes.hpp"
|
|
|
|
using namespace llvm;
|
|
using namespace offload::tblgen;
|
|
|
|
// Emit a list of just the API function names
|
|
void EmitOffloadFuncNames(const RecordKeeper &Records, raw_ostream &OS) {
|
|
OS << GenericHeader;
|
|
OS << R"(
|
|
#ifndef OFFLOAD_FUNC
|
|
#error Please define the macro OFFLOAD_FUNC(Function)
|
|
#endif
|
|
|
|
)";
|
|
for (auto *R : Records.getAllDerivedDefinitions("Function")) {
|
|
FunctionRec FR{R};
|
|
OS << formatv("OFFLOAD_FUNC({0})", FR.getName()) << "\n";
|
|
}
|
|
for (auto *R : Records.getAllDerivedDefinitions("Function")) {
|
|
FunctionRec FR{R};
|
|
OS << formatv("OFFLOAD_FUNC({0}WithCodeLoc)", FR.getName()) << "\n";
|
|
}
|
|
|
|
OS << "\n#undef OFFLOAD_FUNC\n";
|
|
}
|
|
|
|
void EmitOffloadExports(const RecordKeeper &Records, raw_ostream &OS) {
|
|
OS << "VERS1.0 {\n";
|
|
OS << TAB_1 "global:\n";
|
|
|
|
for (auto *R : Records.getAllDerivedDefinitions("Function")) {
|
|
OS << formatv(TAB_2 "{0};\n", FunctionRec(R).getName());
|
|
}
|
|
for (auto *R : Records.getAllDerivedDefinitions("Function")) {
|
|
OS << formatv(TAB_2 "{0}WithCodeLoc;\n", FunctionRec(R).getName());
|
|
}
|
|
OS << TAB_1 "local:\n";
|
|
OS << TAB_2 "*;\n";
|
|
OS << "};\n";
|
|
}
|
|
|
|
// Emit declarations for every implementation function
|
|
void EmitOffloadImplFuncDecls(const RecordKeeper &Records, raw_ostream &OS) {
|
|
OS << GenericHeader;
|
|
for (auto *R : Records.getAllDerivedDefinitions("Function")) {
|
|
FunctionRec F{R};
|
|
OS << formatv("{0}_impl_result_t {1}_impl(", PrefixLower, F.getName());
|
|
auto Params = F.getParams();
|
|
for (auto &Param : Params) {
|
|
OS << Param.getType() << " " << Param.getName();
|
|
if (Param != Params.back()) {
|
|
OS << ", ";
|
|
}
|
|
}
|
|
OS << ");\n\n";
|
|
}
|
|
}
|
|
|
|
// Emit macro calls for each error enum
|
|
void EmitOffloadErrcodes(const RecordKeeper &Records, raw_ostream &OS) {
|
|
OS << GenericHeader;
|
|
OS << R"(
|
|
#ifndef OFFLOAD_ERRC
|
|
#error Please define the macro OFFLOAD_ERRCODE(Name, Desc, Value)
|
|
#endif
|
|
|
|
// Error codes are shared between PluginInterface and liboffload.
|
|
// To add new error codes, add them to offload/liboffload/API/Common.td and run the GenerateOffload target.
|
|
|
|
)";
|
|
|
|
auto ErrorCodeEnum = EnumRec{Records.getDef("ErrorCode")};
|
|
uint32_t EtorVal = 0;
|
|
for (const auto &EnumVal : ErrorCodeEnum.getValues()) {
|
|
OS << formatv(TAB_1 "OFFLOAD_ERRC({0}, \"{1}\", {2})\n", EnumVal.getName(),
|
|
EnumVal.getDesc(), EtorVal++);
|
|
}
|
|
}
|