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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/TableGen/Record.h" 10 #include "llvm/TableGen/TableGenBackend.h" 11 #include <vector> 12 using namespace llvm; 13 14 #define DEBUG_TYPE "attr-enum" 15 16 namespace { 17 18 class Attributes { 19 public: 20 Attributes(RecordKeeper &R) : Records(R) {} 21 void run(raw_ostream &OS); 22 23 private: 24 void emitTargetIndependentNames(raw_ostream &OS); 25 void emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr); 26 void emitAttributeProperties(raw_ostream &OF); 27 28 RecordKeeper &Records; 29 }; 30 31 } // End anonymous namespace. 32 33 void 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 " << MacroName 44 << "(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"}, "ATTRIBUTE_ENUM"); 57 Emit({"StrBoolAttr"}, "ATTRIBUTE_STRBOOL"); 58 Emit({"ComplexStrAttr"}, "ATTRIBUTE_COMPLEXSTR"); 59 60 OS << "#undef ATTRIBUTE_ALL\n"; 61 OS << "#endif\n\n"; 62 63 OS << "#ifdef GET_ATTR_ENUM\n"; 64 OS << "#undef GET_ATTR_ENUM\n"; 65 unsigned Value = 1; // Leave zero for AttrKind::None. 66 for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr"}) { 67 OS << "First" << KindName << " = " << Value << ",\n"; 68 for (auto *A : Records.getAllDerivedDefinitions(KindName)) { 69 OS << A->getName() << " = " << Value << ",\n"; 70 Value++; 71 } 72 OS << "Last" << KindName << " = " << (Value - 1) << ",\n"; 73 } 74 OS << "#endif\n\n"; 75 } 76 77 void Attributes::emitFnAttrCompatCheck(raw_ostream &OS, bool IsStringAttr) { 78 OS << "#ifdef GET_ATTR_COMPAT_FUNC\n"; 79 OS << "#undef GET_ATTR_COMPAT_FUNC\n"; 80 81 OS << "static inline bool hasCompatibleFnAttrs(const Function &Caller,\n" 82 << " const Function &Callee) {\n"; 83 OS << " bool Ret = true;\n\n"; 84 85 std::vector<Record *> CompatRules = 86 Records.getAllDerivedDefinitions("CompatRule"); 87 88 for (auto *Rule : CompatRules) { 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 std::vector<Record *> MergeRules = 102 Records.getAllDerivedDefinitions("MergeRule"); 103 OS << "static inline void mergeFnAttrs(Function &Caller,\n" 104 << " const Function &Callee) {\n"; 105 106 for (auto *Rule : MergeRules) { 107 StringRef FuncName = Rule->getValueAsString("MergeFunc"); 108 OS << " " << FuncName << "(Caller, Callee);\n"; 109 } 110 111 OS << "}\n\n"; 112 113 OS << "#endif\n"; 114 } 115 116 void Attributes::emitAttributeProperties(raw_ostream &OS) { 117 OS << "#ifdef GET_ATTR_PROP_TABLE\n"; 118 OS << "#undef GET_ATTR_PROP_TABLE\n"; 119 OS << "static const uint8_t AttrPropTable[] = {\n"; 120 for (StringRef KindName : {"EnumAttr", "TypeAttr", "IntAttr"}) { 121 for (auto *A : Records.getAllDerivedDefinitions(KindName)) { 122 OS << "0"; 123 for (Init *P : *A->getValueAsListInit("Properties")) 124 OS << " | AttributeProperty::" << cast<DefInit>(P)->getDef()->getName(); 125 OS << ",\n"; 126 } 127 } 128 OS << "};\n"; 129 OS << "#endif\n"; 130 } 131 132 void Attributes::run(raw_ostream &OS) { 133 emitTargetIndependentNames(OS); 134 emitFnAttrCompatCheck(OS, false); 135 emitAttributeProperties(OS); 136 } 137 138 static TableGen::Emitter::OptClass<Attributes> X("gen-attrs", 139 "Generate attributes"); 140