149 lines · cpp
1//===- Attributes.cpp - Generate attributes -------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/TableGen/Error.h"10#include "llvm/TableGen/Record.h"11#include "llvm/TableGen/TableGenBackend.h"12using namespace llvm;13 14#define DEBUG_TYPE "attr-enum"15 16namespace {17 18class Attributes {19public:20 Attributes(const RecordKeeper &R) : Records(R) {}21 void run(raw_ostream &OS);22 23private:24 void emitTargetIndependentNames(raw_ostream &OS);25 void emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr);26 void emitAttributeProperties(raw_ostream &OF);27 28 const RecordKeeper &Records;29};30 31} // End anonymous namespace.32 33void Attributes::emitTargetIndependentNames(raw_ostream &OS) {34 OS << "#ifdef GET_ATTR_NAMES\n";35 OS << "#undef GET_ATTR_NAMES\n";36 37 OS << "#ifndef ATTRIBUTE_ALL\n";38 OS << "#define ATTRIBUTE_ALL(FIRST, SECOND)\n";39 OS << "#endif\n\n";40 41 auto Emit = [&](ArrayRef<StringRef> KindNames, StringRef MacroName) {42 OS << "#ifndef " << MacroName << "\n";43 OS << "#define " << MacroName44 << "(FIRST, SECOND) ATTRIBUTE_ALL(FIRST, SECOND)\n";45 OS << "#endif\n\n";46 for (StringRef KindName : KindNames) {47 for (auto *A : Records.getAllDerivedDefinitions(KindName)) {48 OS << MacroName << "(" << A->getName() << ","49 << A->getValueAsString("AttrString") << ")\n";50 }51 }52 OS << "#undef " << MacroName << "\n\n";53 };54 55 // Emit attribute enums in the same order llvm::Attribute::operator< expects.56 Emit({"EnumAttr", "TypeAttr", "IntAttr", "ConstantRangeAttr",57 "ConstantRangeListAttr"},58 "ATTRIBUTE_ENUM");59 Emit({"StrBoolAttr"}, "ATTRIBUTE_STRBOOL");60 Emit({"ComplexStrAttr"}, "ATTRIBUTE_COMPLEXSTR");61 62 OS << "#undef ATTRIBUTE_ALL\n";63 OS << "#endif\n\n";64 65 OS << "#ifdef GET_ATTR_ENUM\n";66 OS << "#undef GET_ATTR_ENUM\n";67 unsigned Value = 1; // Leave zero for AttrKind::None.68 for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr",69 "ConstantRangeAttr", "ConstantRangeListAttr"}) {70 OS << "First" << KindName << " = " << Value << ",\n";71 for (auto *A : Records.getAllDerivedDefinitions(KindName)) {72 OS << A->getName() << " = " << Value << ",\n";73 Value++;74 }75 OS << "Last" << KindName << " = " << (Value - 1) << ",\n";76 }77 OS << "#endif\n\n";78}79 80void Attributes::emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr) {81 OS << "#ifdef GET_ATTR_COMPAT_FUNC\n";82 OS << "#undef GET_ATTR_COMPAT_FUNC\n";83 84 OS << "static inline bool hasCompatibleFnAttrs(const Function &Caller,\n"85 << " const Function &Callee) {\n";86 OS << " bool Ret = true;\n\n";87 88 for (const Record *Rule : Records.getAllDerivedDefinitions("CompatRule")) {89 StringRef FuncName = Rule->getValueAsString("CompatFunc");90 OS << " Ret &= " << FuncName << "(Caller, Callee";91 StringRef AttrName = Rule->getValueAsString("AttrName");92 if (!AttrName.empty())93 OS << ", \"" << AttrName << "\"";94 OS << ");\n";95 }96 97 OS << "\n";98 OS << " return Ret;\n";99 OS << "}\n\n";100 101 OS << "static inline void mergeFnAttrs(Function &Caller,\n"102 << " const Function &Callee) {\n";103 104 for (const Record *Rule : Records.getAllDerivedDefinitions("MergeRule")) {105 StringRef FuncName = Rule->getValueAsString("MergeFunc");106 OS << " " << FuncName << "(Caller, Callee);\n";107 }108 109 OS << "}\n\n";110 111 OS << "#endif\n";112}113 114void Attributes::emitAttributeProperties(raw_ostream &OS) {115 OS << "#ifdef GET_ATTR_PROP_TABLE\n";116 OS << "#undef GET_ATTR_PROP_TABLE\n";117 OS << "static const uint8_t AttrPropTable[] = {\n";118 for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr",119 "ConstantRangeAttr", "ConstantRangeListAttr"}) {120 bool AllowIntersectAnd = KindName == "EnumAttr";121 bool AllowIntersectMin = KindName == "IntAttr";122 for (auto *A : Records.getAllDerivedDefinitions(KindName)) {123 OS << "0";124 for (const Init *P : *A->getValueAsListInit("Properties")) {125 if (!AllowIntersectAnd &&126 cast<DefInit>(P)->getDef()->getName() == "IntersectAnd")127 PrintFatalError("'IntersectAnd' only compatible with 'EnumAttr'");128 if (!AllowIntersectMin &&129 cast<DefInit>(P)->getDef()->getName() == "IntersectMin")130 PrintFatalError("'IntersectMin' only compatible with 'IntAttr'");131 132 OS << " | AttributeProperty::" << cast<DefInit>(P)->getDef()->getName();133 }134 OS << ",\n";135 }136 }137 OS << "};\n";138 OS << "#endif\n";139}140 141void Attributes::run(raw_ostream &OS) {142 emitTargetIndependentNames(OS);143 emitFnAttrCompatCheck(OS, false);144 emitAttributeProperties(OS);145}146 147static TableGen::Emitter::OptClass<Attributes> X("gen-attrs",148 "Generate attributes");149