10b57cec5SDimitry Andric //===- ClangOpenCLBuiltinEmitter.cpp - Generate Clang OpenCL Builtin handling 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // The LLVM Compiler Infrastructure 40b57cec5SDimitry Andric // 50b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 60b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 70b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 100b57cec5SDimitry Andric // 11fe6060f1SDimitry Andric // These backends consume the definitions of OpenCL builtin functions in 12fe6060f1SDimitry Andric // clang/lib/Sema/OpenCLBuiltins.td and produce builtin handling code for 13fe6060f1SDimitry Andric // inclusion in SemaLookup.cpp, or a test file that calls all declared builtins. 14fe6060f1SDimitry Andric // 15fe6060f1SDimitry Andric //===----------------------------------------------------------------------===// 16fe6060f1SDimitry Andric 17fe6060f1SDimitry Andric #include "TableGenBackends.h" 18fe6060f1SDimitry Andric #include "llvm/ADT/MapVector.h" 19fe6060f1SDimitry Andric #include "llvm/ADT/STLExtras.h" 2081ad6265SDimitry Andric #include "llvm/ADT/SmallSet.h" 21fe6060f1SDimitry Andric #include "llvm/ADT/SmallString.h" 22fe6060f1SDimitry Andric #include "llvm/ADT/StringExtras.h" 23fe6060f1SDimitry Andric #include "llvm/ADT/StringMap.h" 24fe6060f1SDimitry Andric #include "llvm/ADT/StringRef.h" 25fe6060f1SDimitry Andric #include "llvm/ADT/StringSwitch.h" 26fe6060f1SDimitry Andric #include "llvm/Support/ErrorHandling.h" 27fe6060f1SDimitry Andric #include "llvm/Support/raw_ostream.h" 28fe6060f1SDimitry Andric #include "llvm/TableGen/Error.h" 29fe6060f1SDimitry Andric #include "llvm/TableGen/Record.h" 30fe6060f1SDimitry Andric #include "llvm/TableGen/StringMatcher.h" 31fe6060f1SDimitry Andric #include "llvm/TableGen/TableGenBackend.h" 32fe6060f1SDimitry Andric 33fe6060f1SDimitry Andric using namespace llvm; 34fe6060f1SDimitry Andric 35fe6060f1SDimitry Andric namespace { 36fe6060f1SDimitry Andric 37fe6060f1SDimitry Andric // A list of signatures that are shared by one or more builtin functions. 38fe6060f1SDimitry Andric struct BuiltinTableEntries { 39fe6060f1SDimitry Andric SmallVector<StringRef, 4> Names; 40fe6060f1SDimitry Andric std::vector<std::pair<const Record *, unsigned>> Signatures; 41fe6060f1SDimitry Andric }; 42fe6060f1SDimitry Andric 430b57cec5SDimitry Andric // This tablegen backend emits code for checking whether a function is an 440b57cec5SDimitry Andric // OpenCL builtin function. If so, all overloads of this function are 450b57cec5SDimitry Andric // added to the LookupResult. The generated include file is used by 460b57cec5SDimitry Andric // SemaLookup.cpp 470b57cec5SDimitry Andric // 480b57cec5SDimitry Andric // For a successful lookup of e.g. the "cos" builtin, isOpenCLBuiltin("cos") 490b57cec5SDimitry Andric // returns a pair <Index, Len>. 50a7dea167SDimitry Andric // BuiltinTable[Index] to BuiltinTable[Index + Len] contains the pairs 510b57cec5SDimitry Andric // <SigIndex, SigLen> of the overloads of "cos". 52a7dea167SDimitry Andric // SignatureTable[SigIndex] to SignatureTable[SigIndex + SigLen] contains 53a7dea167SDimitry Andric // one of the signatures of "cos". The SignatureTable entry can be 54a7dea167SDimitry Andric // referenced by other functions, e.g. "sin", to exploit the fact that 55a7dea167SDimitry Andric // many OpenCL builtins share the same signature. 56a7dea167SDimitry Andric // 57a7dea167SDimitry Andric // The file generated by this TableGen emitter contains the following: 58a7dea167SDimitry Andric // 59a7dea167SDimitry Andric // * Structs and enums to represent types and function signatures. 60a7dea167SDimitry Andric // 61480093f4SDimitry Andric // * const char *FunctionExtensionTable[] 62480093f4SDimitry Andric // List of space-separated OpenCL extensions. A builtin references an 63480093f4SDimitry Andric // entry in this table when the builtin requires a particular (set of) 64480093f4SDimitry Andric // extension(s) to be enabled. 65480093f4SDimitry Andric // 66a7dea167SDimitry Andric // * OpenCLTypeStruct TypeTable[] 67a7dea167SDimitry Andric // Type information for return types and arguments. 68a7dea167SDimitry Andric // 69a7dea167SDimitry Andric // * unsigned SignatureTable[] 70a7dea167SDimitry Andric // A list of types representing function signatures. Each entry is an index 71a7dea167SDimitry Andric // into the above TypeTable. Multiple entries following each other form a 72a7dea167SDimitry Andric // signature, where the first entry is the return type and subsequent 73a7dea167SDimitry Andric // entries are the argument types. 74a7dea167SDimitry Andric // 75a7dea167SDimitry Andric // * OpenCLBuiltinStruct BuiltinTable[] 76a7dea167SDimitry Andric // Each entry represents one overload of an OpenCL builtin function and 77a7dea167SDimitry Andric // consists of an index into the SignatureTable and the number of arguments. 78a7dea167SDimitry Andric // 79a7dea167SDimitry Andric // * std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name) 80a7dea167SDimitry Andric // Find out whether a string matches an existing OpenCL builtin function 81a7dea167SDimitry Andric // name and return an index into BuiltinTable and the number of overloads. 82a7dea167SDimitry Andric // 83fe6060f1SDimitry Andric // * void OCL2Qual(Sema&, OpenCLTypeStruct, std::vector<QualType>&) 84a7dea167SDimitry Andric // Convert an OpenCLTypeStruct type to a list of QualType instances. 85a7dea167SDimitry Andric // One OpenCLTypeStruct can represent multiple types, primarily when using 86a7dea167SDimitry Andric // GenTypes. 87a7dea167SDimitry Andric // 880b57cec5SDimitry Andric class BuiltinNameEmitter { 890b57cec5SDimitry Andric public: 900b57cec5SDimitry Andric BuiltinNameEmitter(RecordKeeper &Records, raw_ostream &OS) 910b57cec5SDimitry Andric : Records(Records), OS(OS) {} 920b57cec5SDimitry Andric 930b57cec5SDimitry Andric // Entrypoint to generate the functions and structures for checking 940b57cec5SDimitry Andric // whether a function is an OpenCL builtin function. 950b57cec5SDimitry Andric void Emit(); 960b57cec5SDimitry Andric 970b57cec5SDimitry Andric private: 98480093f4SDimitry Andric // A list of indices into the builtin function table. 99480093f4SDimitry Andric using BuiltinIndexListTy = SmallVector<unsigned, 11>; 100480093f4SDimitry Andric 1010b57cec5SDimitry Andric // Contains OpenCL builtin functions and related information, stored as 1020b57cec5SDimitry Andric // Record instances. They are coming from the associated TableGen file. 1030b57cec5SDimitry Andric RecordKeeper &Records; 1040b57cec5SDimitry Andric 1050b57cec5SDimitry Andric // The output file. 1060b57cec5SDimitry Andric raw_ostream &OS; 1070b57cec5SDimitry Andric 108a7dea167SDimitry Andric // Helper function for BuiltinNameEmitter::EmitDeclarations. Generate enum 109a7dea167SDimitry Andric // definitions in the Output string parameter, and save their Record instances 110a7dea167SDimitry Andric // in the List parameter. 111a7dea167SDimitry Andric // \param Types (in) List containing the Types to extract. 112a7dea167SDimitry Andric // \param TypesSeen (inout) List containing the Types already extracted. 113a7dea167SDimitry Andric // \param Output (out) String containing the enums to emit in the output file. 114a7dea167SDimitry Andric // \param List (out) List containing the extracted Types, except the Types in 115a7dea167SDimitry Andric // TypesSeen. 116a7dea167SDimitry Andric void ExtractEnumTypes(std::vector<Record *> &Types, 117a7dea167SDimitry Andric StringMap<bool> &TypesSeen, std::string &Output, 118a7dea167SDimitry Andric std::vector<const Record *> &List); 119a7dea167SDimitry Andric 120a7dea167SDimitry Andric // Emit the enum or struct used in the generated file. 121a7dea167SDimitry Andric // Populate the TypeList at the same time. 1220b57cec5SDimitry Andric void EmitDeclarations(); 1230b57cec5SDimitry Andric 124a7dea167SDimitry Andric // Parse the Records generated by TableGen to populate the SignaturesList, 125a7dea167SDimitry Andric // FctOverloadMap and TypeMap. 1260b57cec5SDimitry Andric void GetOverloads(); 1270b57cec5SDimitry Andric 128480093f4SDimitry Andric // Compare two lists of signatures and check that e.g. the OpenCL version, 129480093f4SDimitry Andric // function attributes, and extension are equal for each signature. 130480093f4SDimitry Andric // \param Candidate (in) Entry in the SignatureListMap to check. 131480093f4SDimitry Andric // \param SignatureList (in) List of signatures of the considered function. 132480093f4SDimitry Andric // \returns true if the two lists of signatures are identical. 133480093f4SDimitry Andric bool CanReuseSignature( 134480093f4SDimitry Andric BuiltinIndexListTy *Candidate, 135480093f4SDimitry Andric std::vector<std::pair<const Record *, unsigned>> &SignatureList); 136480093f4SDimitry Andric 137480093f4SDimitry Andric // Group functions with the same list of signatures by populating the 138480093f4SDimitry Andric // SignatureListMap. 139480093f4SDimitry Andric // Some builtin functions have the same list of signatures, for example the 140480093f4SDimitry Andric // "sin" and "cos" functions. To save space in the BuiltinTable, the 141480093f4SDimitry Andric // "isOpenCLBuiltin" function will have the same output for these two 142480093f4SDimitry Andric // function names. 143480093f4SDimitry Andric void GroupBySignature(); 144480093f4SDimitry Andric 145480093f4SDimitry Andric // Emit the FunctionExtensionTable that lists all function extensions. 146480093f4SDimitry Andric void EmitExtensionTable(); 147480093f4SDimitry Andric 148a7dea167SDimitry Andric // Emit the TypeTable containing all types used by OpenCL builtins. 149a7dea167SDimitry Andric void EmitTypeTable(); 150a7dea167SDimitry Andric 151a7dea167SDimitry Andric // Emit the SignatureTable. This table contains all the possible signatures. 152a7dea167SDimitry Andric // A signature is stored as a list of indexes of the TypeTable. 153a7dea167SDimitry Andric // The first index references the return type (mandatory), and the followings 154a7dea167SDimitry Andric // reference its arguments. 1550b57cec5SDimitry Andric // E.g.: 156a7dea167SDimitry Andric // 15, 2, 15 can represent a function with the signature: 157a7dea167SDimitry Andric // int func(float, int) 158a7dea167SDimitry Andric // The "int" type being at the index 15 in the TypeTable. 1590b57cec5SDimitry Andric void EmitSignatureTable(); 1600b57cec5SDimitry Andric 161a7dea167SDimitry Andric // Emit the BuiltinTable table. This table contains all the overloads of 1620b57cec5SDimitry Andric // each function, and is a struct OpenCLBuiltinDecl. 1630b57cec5SDimitry Andric // E.g.: 164a7dea167SDimitry Andric // // 891 convert_float2_rtn 165480093f4SDimitry Andric // { 58, 2, 3, 100, 0 }, 166a7dea167SDimitry Andric // This means that the signature of this convert_float2_rtn overload has 167a7dea167SDimitry Andric // 1 argument (+1 for the return type), stored at index 58 in 168480093f4SDimitry Andric // the SignatureTable. This prototype requires extension "3" in the 169480093f4SDimitry Andric // FunctionExtensionTable. The last two values represent the minimum (1.0) 170480093f4SDimitry Andric // and maximum (0, meaning no max version) OpenCL version in which this 171480093f4SDimitry Andric // overload is supported. 1720b57cec5SDimitry Andric void EmitBuiltinTable(); 1730b57cec5SDimitry Andric 1740b57cec5SDimitry Andric // Emit a StringMatcher function to check whether a function name is an 1750b57cec5SDimitry Andric // OpenCL builtin function name. 1760b57cec5SDimitry Andric void EmitStringMatcher(); 1770b57cec5SDimitry Andric 1780b57cec5SDimitry Andric // Emit a function returning the clang QualType instance associated with 1790b57cec5SDimitry Andric // the TableGen Record Type. 1800b57cec5SDimitry Andric void EmitQualTypeFinder(); 1810b57cec5SDimitry Andric 1820b57cec5SDimitry Andric // Contains a list of the available signatures, without the name of the 1830b57cec5SDimitry Andric // function. Each pair consists of a signature and a cumulative index. 1840b57cec5SDimitry Andric // E.g.: <<float, float>, 0>, 1850b57cec5SDimitry Andric // <<float, int, int, 2>>, 1860b57cec5SDimitry Andric // <<float>, 5>, 1870b57cec5SDimitry Andric // ... 1880b57cec5SDimitry Andric // <<double, double>, 35>. 189a7dea167SDimitry Andric std::vector<std::pair<std::vector<Record *>, unsigned>> SignaturesList; 1900b57cec5SDimitry Andric 1910b57cec5SDimitry Andric // Map the name of a builtin function to its prototypes (instances of the 1920b57cec5SDimitry Andric // TableGen "Builtin" class). 1930b57cec5SDimitry Andric // Each prototype is registered as a pair of: 1940b57cec5SDimitry Andric // <pointer to the "Builtin" instance, 195a7dea167SDimitry Andric // cumulative index of the associated signature in the SignaturesList> 1960b57cec5SDimitry Andric // E.g.: The function cos: (float cos(float), double cos(double), ...) 1970b57cec5SDimitry Andric // <"cos", <<ptrToPrototype0, 5>, 198a7dea167SDimitry Andric // <ptrToPrototype1, 35>, 1990b57cec5SDimitry Andric // <ptrToPrototype2, 79>> 2000b57cec5SDimitry Andric // ptrToPrototype1 has the following signature: <double, double> 2010b57cec5SDimitry Andric MapVector<StringRef, std::vector<std::pair<const Record *, unsigned>>> 202a7dea167SDimitry Andric FctOverloadMap; 203a7dea167SDimitry Andric 204a7dea167SDimitry Andric // Contains the map of OpenCL types to their index in the TypeTable. 205a7dea167SDimitry Andric MapVector<const Record *, unsigned> TypeMap; 206a7dea167SDimitry Andric 207480093f4SDimitry Andric // List of OpenCL function extensions mapping extension strings to 208480093f4SDimitry Andric // an index into the FunctionExtensionTable. 209480093f4SDimitry Andric StringMap<unsigned> FunctionExtensionIndex; 210480093f4SDimitry Andric 211a7dea167SDimitry Andric // List of OpenCL type names in the same order as in enum OpenCLTypeID. 212a7dea167SDimitry Andric // This list does not contain generic types. 213a7dea167SDimitry Andric std::vector<const Record *> TypeList; 214a7dea167SDimitry Andric 215a7dea167SDimitry Andric // Same as TypeList, but for generic types only. 216a7dea167SDimitry Andric std::vector<const Record *> GenTypeList; 217480093f4SDimitry Andric 218480093f4SDimitry Andric // Map an ordered vector of signatures to their original Record instances, 219480093f4SDimitry Andric // and to a list of function names that share these signatures. 220480093f4SDimitry Andric // 221480093f4SDimitry Andric // For example, suppose the "cos" and "sin" functions have only three 222480093f4SDimitry Andric // signatures, and these signatures are at index Ix in the SignatureTable: 223480093f4SDimitry Andric // cos | sin | Signature | Index 224480093f4SDimitry Andric // float cos(float) | float sin(float) | Signature1 | I1 225480093f4SDimitry Andric // double cos(double) | double sin(double) | Signature2 | I2 226480093f4SDimitry Andric // half cos(half) | half sin(half) | Signature3 | I3 227480093f4SDimitry Andric // 228480093f4SDimitry Andric // Then we will create a mapping of the vector of signatures: 229480093f4SDimitry Andric // SignatureListMap[<I1, I2, I3>] = < 230480093f4SDimitry Andric // <"cos", "sin">, 231480093f4SDimitry Andric // <Signature1, Signature2, Signature3>> 232480093f4SDimitry Andric // The function "tan", having the same signatures, would be mapped to the 233480093f4SDimitry Andric // same entry (<I1, I2, I3>). 234480093f4SDimitry Andric MapVector<BuiltinIndexListTy *, BuiltinTableEntries> SignatureListMap; 2350b57cec5SDimitry Andric }; 236fe6060f1SDimitry Andric 237349cc55cSDimitry Andric /// Base class for emitting a file (e.g. header or test) from OpenCLBuiltins.td 238349cc55cSDimitry Andric class OpenCLBuiltinFileEmitterBase { 239fe6060f1SDimitry Andric public: 240349cc55cSDimitry Andric OpenCLBuiltinFileEmitterBase(RecordKeeper &Records, raw_ostream &OS) 241fe6060f1SDimitry Andric : Records(Records), OS(OS) {} 242349cc55cSDimitry Andric virtual ~OpenCLBuiltinFileEmitterBase() = default; 243fe6060f1SDimitry Andric 244fe6060f1SDimitry Andric // Entrypoint to generate the functions for testing all OpenCL builtin 245fe6060f1SDimitry Andric // functions. 246349cc55cSDimitry Andric virtual void emit() = 0; 247fe6060f1SDimitry Andric 248349cc55cSDimitry Andric protected: 249fe6060f1SDimitry Andric struct TypeFlags { 250fe6060f1SDimitry Andric TypeFlags() : IsConst(false), IsVolatile(false), IsPointer(false) {} 251fe6060f1SDimitry Andric bool IsConst : 1; 252fe6060f1SDimitry Andric bool IsVolatile : 1; 253fe6060f1SDimitry Andric bool IsPointer : 1; 254fe6060f1SDimitry Andric StringRef AddrSpace; 255fe6060f1SDimitry Andric }; 256fe6060f1SDimitry Andric 257fe6060f1SDimitry Andric // Return a string representation of the given type, such that it can be 258fe6060f1SDimitry Andric // used as a type in OpenCL C code. 259fe6060f1SDimitry Andric std::string getTypeString(const Record *Type, TypeFlags Flags, 260fe6060f1SDimitry Andric int VectorSize) const; 261fe6060f1SDimitry Andric 262fe6060f1SDimitry Andric // Return the type(s) and vector size(s) for the given type. For 263fe6060f1SDimitry Andric // non-GenericTypes, the resulting vectors will contain 1 element. For 264fe6060f1SDimitry Andric // GenericTypes, the resulting vectors typically contain multiple elements. 265fe6060f1SDimitry Andric void getTypeLists(Record *Type, TypeFlags &Flags, 266fe6060f1SDimitry Andric std::vector<Record *> &TypeList, 267fe6060f1SDimitry Andric std::vector<int64_t> &VectorList) const; 268fe6060f1SDimitry Andric 269fe6060f1SDimitry Andric // Expand the TableGen Records representing a builtin function signature into 270fe6060f1SDimitry Andric // one or more function signatures. Return them as a vector of a vector of 271fe6060f1SDimitry Andric // strings, with each string containing an OpenCL C type and optional 272fe6060f1SDimitry Andric // qualifiers. 273fe6060f1SDimitry Andric // 274fe6060f1SDimitry Andric // The Records may contain GenericTypes, which expand into multiple 275fe6060f1SDimitry Andric // signatures. Repeated occurrences of GenericType in a signature expand to 276fe6060f1SDimitry Andric // the same types. For example [char, FGenType, FGenType] expands to: 277fe6060f1SDimitry Andric // [char, float, float] 278fe6060f1SDimitry Andric // [char, float2, float2] 279fe6060f1SDimitry Andric // [char, float3, float3] 280fe6060f1SDimitry Andric // ... 281fe6060f1SDimitry Andric void 282fe6060f1SDimitry Andric expandTypesInSignature(const std::vector<Record *> &Signature, 283fe6060f1SDimitry Andric SmallVectorImpl<SmallVector<std::string, 2>> &Types); 284fe6060f1SDimitry Andric 285349cc55cSDimitry Andric // Emit extension enabling pragmas. 286349cc55cSDimitry Andric void emitExtensionSetup(); 287349cc55cSDimitry Andric 288349cc55cSDimitry Andric // Emit an #if guard for a Builtin's extension. Return the corresponding 289349cc55cSDimitry Andric // closing #endif, or an empty string if no extension #if guard was emitted. 290349cc55cSDimitry Andric std::string emitExtensionGuard(const Record *Builtin); 291349cc55cSDimitry Andric 292349cc55cSDimitry Andric // Emit an #if guard for a Builtin's language version. Return the 293349cc55cSDimitry Andric // corresponding closing #endif, or an empty string if no version #if guard 294349cc55cSDimitry Andric // was emitted. 295349cc55cSDimitry Andric std::string emitVersionGuard(const Record *Builtin); 296349cc55cSDimitry Andric 29781ad6265SDimitry Andric // Emit an #if guard for all type extensions required for the given type 29881ad6265SDimitry Andric // strings. Return the corresponding closing #endif, or an empty string 29981ad6265SDimitry Andric // if no extension #if guard was emitted. 30081ad6265SDimitry Andric StringRef 30181ad6265SDimitry Andric emitTypeExtensionGuards(const SmallVectorImpl<std::string> &Signature); 30281ad6265SDimitry Andric 30381ad6265SDimitry Andric // Map type strings to type extensions (e.g. "half2" -> "cl_khr_fp16"). 30481ad6265SDimitry Andric StringMap<StringRef> TypeExtMap; 30581ad6265SDimitry Andric 306fe6060f1SDimitry Andric // Contains OpenCL builtin functions and related information, stored as 307fe6060f1SDimitry Andric // Record instances. They are coming from the associated TableGen file. 308fe6060f1SDimitry Andric RecordKeeper &Records; 309fe6060f1SDimitry Andric 310fe6060f1SDimitry Andric // The output file. 311fe6060f1SDimitry Andric raw_ostream &OS; 312fe6060f1SDimitry Andric }; 313fe6060f1SDimitry Andric 314349cc55cSDimitry Andric // OpenCL builtin test generator. This class processes the same TableGen input 315349cc55cSDimitry Andric // as BuiltinNameEmitter, but generates a .cl file that contains a call to each 316349cc55cSDimitry Andric // builtin function described in the .td input. 317349cc55cSDimitry Andric class OpenCLBuiltinTestEmitter : public OpenCLBuiltinFileEmitterBase { 318349cc55cSDimitry Andric public: 319349cc55cSDimitry Andric OpenCLBuiltinTestEmitter(RecordKeeper &Records, raw_ostream &OS) 320349cc55cSDimitry Andric : OpenCLBuiltinFileEmitterBase(Records, OS) {} 321349cc55cSDimitry Andric 322349cc55cSDimitry Andric // Entrypoint to generate the functions for testing all OpenCL builtin 323349cc55cSDimitry Andric // functions. 324349cc55cSDimitry Andric void emit() override; 325349cc55cSDimitry Andric }; 326349cc55cSDimitry Andric 32706c3fb27SDimitry Andric // OpenCL builtin header generator. This class processes the same TableGen 32806c3fb27SDimitry Andric // input as BuiltinNameEmitter, but generates a .h file that contains a 32906c3fb27SDimitry Andric // prototype for each builtin function described in the .td input. 33006c3fb27SDimitry Andric class OpenCLBuiltinHeaderEmitter : public OpenCLBuiltinFileEmitterBase { 33106c3fb27SDimitry Andric public: 33206c3fb27SDimitry Andric OpenCLBuiltinHeaderEmitter(RecordKeeper &Records, raw_ostream &OS) 33306c3fb27SDimitry Andric : OpenCLBuiltinFileEmitterBase(Records, OS) {} 33406c3fb27SDimitry Andric 33506c3fb27SDimitry Andric // Entrypoint to generate the header. 33606c3fb27SDimitry Andric void emit() override; 33706c3fb27SDimitry Andric }; 33806c3fb27SDimitry Andric 3390b57cec5SDimitry Andric } // namespace 3400b57cec5SDimitry Andric 3410b57cec5SDimitry Andric void BuiltinNameEmitter::Emit() { 342*5f757f3fSDimitry Andric emitSourceFileHeader("OpenCL Builtin handling", OS, Records); 3430b57cec5SDimitry Andric 3440b57cec5SDimitry Andric OS << "#include \"llvm/ADT/StringRef.h\"\n"; 3450b57cec5SDimitry Andric OS << "using namespace clang;\n\n"; 3460b57cec5SDimitry Andric 347a7dea167SDimitry Andric // Emit enums and structs. 3480b57cec5SDimitry Andric EmitDeclarations(); 3490b57cec5SDimitry Andric 350480093f4SDimitry Andric // Parse the Records to populate the internal lists. 3510b57cec5SDimitry Andric GetOverloads(); 352480093f4SDimitry Andric GroupBySignature(); 3530b57cec5SDimitry Andric 354a7dea167SDimitry Andric // Emit tables. 355480093f4SDimitry Andric EmitExtensionTable(); 356a7dea167SDimitry Andric EmitTypeTable(); 3570b57cec5SDimitry Andric EmitSignatureTable(); 3580b57cec5SDimitry Andric EmitBuiltinTable(); 3590b57cec5SDimitry Andric 360480093f4SDimitry Andric // Emit functions. 3610b57cec5SDimitry Andric EmitStringMatcher(); 3620b57cec5SDimitry Andric EmitQualTypeFinder(); 3630b57cec5SDimitry Andric } 3640b57cec5SDimitry Andric 365a7dea167SDimitry Andric void BuiltinNameEmitter::ExtractEnumTypes(std::vector<Record *> &Types, 366a7dea167SDimitry Andric StringMap<bool> &TypesSeen, 367a7dea167SDimitry Andric std::string &Output, 368a7dea167SDimitry Andric std::vector<const Record *> &List) { 369a7dea167SDimitry Andric raw_string_ostream SS(Output); 370a7dea167SDimitry Andric 3710b57cec5SDimitry Andric for (const auto *T : Types) { 37206c3fb27SDimitry Andric if (!TypesSeen.contains(T->getValueAsString("Name"))) { 373a7dea167SDimitry Andric SS << " OCLT_" + T->getValueAsString("Name") << ",\n"; 374a7dea167SDimitry Andric // Save the type names in the same order as their enum value. Note that 375a7dea167SDimitry Andric // the Record can be a VectorType or something else, only the name is 376a7dea167SDimitry Andric // important. 377a7dea167SDimitry Andric List.push_back(T); 3780b57cec5SDimitry Andric TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true)); 3790b57cec5SDimitry Andric } 380a7dea167SDimitry Andric } 381a7dea167SDimitry Andric SS.flush(); 382a7dea167SDimitry Andric } 383a7dea167SDimitry Andric 384a7dea167SDimitry Andric void BuiltinNameEmitter::EmitDeclarations() { 385a7dea167SDimitry Andric // Enum of scalar type names (float, int, ...) and generic type sets. 386a7dea167SDimitry Andric OS << "enum OpenCLTypeID {\n"; 387a7dea167SDimitry Andric 388a7dea167SDimitry Andric StringMap<bool> TypesSeen; 389a7dea167SDimitry Andric std::string GenTypeEnums; 390a7dea167SDimitry Andric std::string TypeEnums; 391a7dea167SDimitry Andric 392a7dea167SDimitry Andric // Extract generic types and non-generic types separately, to keep 393a7dea167SDimitry Andric // gentypes at the end of the enum which simplifies the special handling 394a7dea167SDimitry Andric // for gentypes in SemaLookup. 395a7dea167SDimitry Andric std::vector<Record *> GenTypes = 396a7dea167SDimitry Andric Records.getAllDerivedDefinitions("GenericType"); 397a7dea167SDimitry Andric ExtractEnumTypes(GenTypes, TypesSeen, GenTypeEnums, GenTypeList); 398a7dea167SDimitry Andric 399a7dea167SDimitry Andric std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type"); 400a7dea167SDimitry Andric ExtractEnumTypes(Types, TypesSeen, TypeEnums, TypeList); 401a7dea167SDimitry Andric 402a7dea167SDimitry Andric OS << TypeEnums; 403a7dea167SDimitry Andric OS << GenTypeEnums; 4040b57cec5SDimitry Andric OS << "};\n"; 4050b57cec5SDimitry Andric 406a7dea167SDimitry Andric // Structure definitions. 4070b57cec5SDimitry Andric OS << R"( 408a7dea167SDimitry Andric // Image access qualifier. 409a7dea167SDimitry Andric enum OpenCLAccessQual : unsigned char { 410a7dea167SDimitry Andric OCLAQ_None, 411a7dea167SDimitry Andric OCLAQ_ReadOnly, 412a7dea167SDimitry Andric OCLAQ_WriteOnly, 413a7dea167SDimitry Andric OCLAQ_ReadWrite 414a7dea167SDimitry Andric }; 4150b57cec5SDimitry Andric 416a7dea167SDimitry Andric // Represents a return type or argument type. 417a7dea167SDimitry Andric struct OpenCLTypeStruct { 418a7dea167SDimitry Andric // A type (e.g. float, int, ...). 419a7dea167SDimitry Andric const OpenCLTypeID ID; 420a7dea167SDimitry Andric // Vector size (if applicable; 0 for scalars and generic types). 421a7dea167SDimitry Andric const unsigned VectorWidth; 422a7dea167SDimitry Andric // 0 if the type is not a pointer. 4235ffd83dbSDimitry Andric const bool IsPointer : 1; 424a7dea167SDimitry Andric // 0 if the type is not const. 4255ffd83dbSDimitry Andric const bool IsConst : 1; 426a7dea167SDimitry Andric // 0 if the type is not volatile. 4275ffd83dbSDimitry Andric const bool IsVolatile : 1; 428a7dea167SDimitry Andric // Access qualifier. 429a7dea167SDimitry Andric const OpenCLAccessQual AccessQualifier; 430a7dea167SDimitry Andric // Address space of the pointer (if applicable). 431a7dea167SDimitry Andric const LangAS AS; 4320b57cec5SDimitry Andric }; 4330b57cec5SDimitry Andric 4340b57cec5SDimitry Andric // One overload of an OpenCL builtin function. 435a7dea167SDimitry Andric struct OpenCLBuiltinStruct { 436a7dea167SDimitry Andric // Index of the signature in the OpenCLTypeStruct table. 437a7dea167SDimitry Andric const unsigned SigTableIndex; 438a7dea167SDimitry Andric // Entries between index SigTableIndex and (SigTableIndex + NumTypes - 1) in 439a7dea167SDimitry Andric // the SignatureTable represent the complete signature. The first type at 440a7dea167SDimitry Andric // index SigTableIndex is the return type. 441a7dea167SDimitry Andric const unsigned NumTypes; 442480093f4SDimitry Andric // Function attribute __attribute__((pure)) 4435ffd83dbSDimitry Andric const bool IsPure : 1; 444480093f4SDimitry Andric // Function attribute __attribute__((const)) 4455ffd83dbSDimitry Andric const bool IsConst : 1; 446480093f4SDimitry Andric // Function attribute __attribute__((convergent)) 4475ffd83dbSDimitry Andric const bool IsConv : 1; 448480093f4SDimitry Andric // OpenCL extension(s) required for this overload. 449480093f4SDimitry Andric const unsigned short Extension; 450fe6060f1SDimitry Andric // OpenCL versions in which this overload is available. 451fe6060f1SDimitry Andric const unsigned short Versions; 4520b57cec5SDimitry Andric }; 4530b57cec5SDimitry Andric 4540b57cec5SDimitry Andric )"; 4550b57cec5SDimitry Andric } 4560b57cec5SDimitry Andric 457a7dea167SDimitry Andric // Verify that the combination of GenTypes in a signature is supported. 458a7dea167SDimitry Andric // To simplify the logic for creating overloads in SemaLookup, only allow 459a7dea167SDimitry Andric // a signature to contain different GenTypes if these GenTypes represent 460a7dea167SDimitry Andric // the same number of actual scalar or vector types. 461a7dea167SDimitry Andric // 462a7dea167SDimitry Andric // Exit with a fatal error if an unsupported construct is encountered. 463a7dea167SDimitry Andric static void VerifySignature(const std::vector<Record *> &Signature, 464a7dea167SDimitry Andric const Record *BuiltinRec) { 465a7dea167SDimitry Andric unsigned GenTypeVecSizes = 1; 466a7dea167SDimitry Andric unsigned GenTypeTypes = 1; 467a7dea167SDimitry Andric 468a7dea167SDimitry Andric for (const auto *T : Signature) { 469a7dea167SDimitry Andric // Check all GenericType arguments in this signature. 470a7dea167SDimitry Andric if (T->isSubClassOf("GenericType")) { 471a7dea167SDimitry Andric // Check number of vector sizes. 472a7dea167SDimitry Andric unsigned NVecSizes = 473a7dea167SDimitry Andric T->getValueAsDef("VectorList")->getValueAsListOfInts("List").size(); 474a7dea167SDimitry Andric if (NVecSizes != GenTypeVecSizes && NVecSizes != 1) { 475a7dea167SDimitry Andric if (GenTypeVecSizes > 1) { 476a7dea167SDimitry Andric // We already saw a gentype with a different number of vector sizes. 477a7dea167SDimitry Andric PrintFatalError(BuiltinRec->getLoc(), 478a7dea167SDimitry Andric "number of vector sizes should be equal or 1 for all gentypes " 479a7dea167SDimitry Andric "in a declaration"); 480a7dea167SDimitry Andric } 481a7dea167SDimitry Andric GenTypeVecSizes = NVecSizes; 482a7dea167SDimitry Andric } 483a7dea167SDimitry Andric 484a7dea167SDimitry Andric // Check number of data types. 485a7dea167SDimitry Andric unsigned NTypes = 486a7dea167SDimitry Andric T->getValueAsDef("TypeList")->getValueAsListOfDefs("List").size(); 487a7dea167SDimitry Andric if (NTypes != GenTypeTypes && NTypes != 1) { 488a7dea167SDimitry Andric if (GenTypeTypes > 1) { 489a7dea167SDimitry Andric // We already saw a gentype with a different number of types. 490a7dea167SDimitry Andric PrintFatalError(BuiltinRec->getLoc(), 491a7dea167SDimitry Andric "number of types should be equal or 1 for all gentypes " 492a7dea167SDimitry Andric "in a declaration"); 493a7dea167SDimitry Andric } 494a7dea167SDimitry Andric GenTypeTypes = NTypes; 495a7dea167SDimitry Andric } 496a7dea167SDimitry Andric } 497a7dea167SDimitry Andric } 498a7dea167SDimitry Andric } 499a7dea167SDimitry Andric 5000b57cec5SDimitry Andric void BuiltinNameEmitter::GetOverloads() { 501a7dea167SDimitry Andric // Populate the TypeMap. 502a7dea167SDimitry Andric std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type"); 503a7dea167SDimitry Andric unsigned I = 0; 504a7dea167SDimitry Andric for (const auto &T : Types) { 505a7dea167SDimitry Andric TypeMap.insert(std::make_pair(T, I++)); 506a7dea167SDimitry Andric } 507a7dea167SDimitry Andric 508a7dea167SDimitry Andric // Populate the SignaturesList and the FctOverloadMap. 5090b57cec5SDimitry Andric unsigned CumulativeSignIndex = 0; 5100b57cec5SDimitry Andric std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin"); 5110b57cec5SDimitry Andric for (const auto *B : Builtins) { 5120b57cec5SDimitry Andric StringRef BName = B->getValueAsString("Name"); 51306c3fb27SDimitry Andric if (!FctOverloadMap.contains(BName)) { 514a7dea167SDimitry Andric FctOverloadMap.insert(std::make_pair( 5150b57cec5SDimitry Andric BName, std::vector<std::pair<const Record *, unsigned>>{})); 5160b57cec5SDimitry Andric } 5170b57cec5SDimitry Andric 5180b57cec5SDimitry Andric auto Signature = B->getValueAsListOfDefs("Signature"); 519a7dea167SDimitry Andric // Reuse signatures to avoid unnecessary duplicates. 5200b57cec5SDimitry Andric auto it = 521349cc55cSDimitry Andric llvm::find_if(SignaturesList, 5220b57cec5SDimitry Andric [&](const std::pair<std::vector<Record *>, unsigned> &a) { 5230b57cec5SDimitry Andric return a.first == Signature; 5240b57cec5SDimitry Andric }); 5250b57cec5SDimitry Andric unsigned SignIndex; 526a7dea167SDimitry Andric if (it == SignaturesList.end()) { 527a7dea167SDimitry Andric VerifySignature(Signature, B); 528a7dea167SDimitry Andric SignaturesList.push_back(std::make_pair(Signature, CumulativeSignIndex)); 5290b57cec5SDimitry Andric SignIndex = CumulativeSignIndex; 5300b57cec5SDimitry Andric CumulativeSignIndex += Signature.size(); 5310b57cec5SDimitry Andric } else { 5320b57cec5SDimitry Andric SignIndex = it->second; 5330b57cec5SDimitry Andric } 534a7dea167SDimitry Andric FctOverloadMap[BName].push_back(std::make_pair(B, SignIndex)); 5350b57cec5SDimitry Andric } 5360b57cec5SDimitry Andric } 5370b57cec5SDimitry Andric 538480093f4SDimitry Andric void BuiltinNameEmitter::EmitExtensionTable() { 539480093f4SDimitry Andric OS << "static const char *FunctionExtensionTable[] = {\n"; 540480093f4SDimitry Andric unsigned Index = 0; 541480093f4SDimitry Andric std::vector<Record *> FuncExtensions = 542480093f4SDimitry Andric Records.getAllDerivedDefinitions("FunctionExtension"); 543480093f4SDimitry Andric 544480093f4SDimitry Andric for (const auto &FE : FuncExtensions) { 545480093f4SDimitry Andric // Emit OpenCL extension table entry. 546480093f4SDimitry Andric OS << " // " << Index << ": " << FE->getName() << "\n" 547480093f4SDimitry Andric << " \"" << FE->getValueAsString("ExtName") << "\",\n"; 548480093f4SDimitry Andric 549480093f4SDimitry Andric // Record index of this extension. 550480093f4SDimitry Andric FunctionExtensionIndex[FE->getName()] = Index++; 551480093f4SDimitry Andric } 552480093f4SDimitry Andric OS << "};\n\n"; 553480093f4SDimitry Andric } 554480093f4SDimitry Andric 555a7dea167SDimitry Andric void BuiltinNameEmitter::EmitTypeTable() { 556a7dea167SDimitry Andric OS << "static const OpenCLTypeStruct TypeTable[] = {\n"; 557a7dea167SDimitry Andric for (const auto &T : TypeMap) { 558a7dea167SDimitry Andric const char *AccessQual = 559a7dea167SDimitry Andric StringSwitch<const char *>(T.first->getValueAsString("AccessQualifier")) 560a7dea167SDimitry Andric .Case("RO", "OCLAQ_ReadOnly") 561a7dea167SDimitry Andric .Case("WO", "OCLAQ_WriteOnly") 562a7dea167SDimitry Andric .Case("RW", "OCLAQ_ReadWrite") 563a7dea167SDimitry Andric .Default("OCLAQ_None"); 564a7dea167SDimitry Andric 565a7dea167SDimitry Andric OS << " // " << T.second << "\n" 566a7dea167SDimitry Andric << " {OCLT_" << T.first->getValueAsString("Name") << ", " 567a7dea167SDimitry Andric << T.first->getValueAsInt("VecWidth") << ", " 568a7dea167SDimitry Andric << T.first->getValueAsBit("IsPointer") << ", " 569a7dea167SDimitry Andric << T.first->getValueAsBit("IsConst") << ", " 570a7dea167SDimitry Andric << T.first->getValueAsBit("IsVolatile") << ", " 571a7dea167SDimitry Andric << AccessQual << ", " 572a7dea167SDimitry Andric << T.first->getValueAsString("AddrSpace") << "},\n"; 5730b57cec5SDimitry Andric } 574a7dea167SDimitry Andric OS << "};\n\n"; 575a7dea167SDimitry Andric } 576a7dea167SDimitry Andric 577a7dea167SDimitry Andric void BuiltinNameEmitter::EmitSignatureTable() { 578a7dea167SDimitry Andric // Store a type (e.g. int, float, int2, ...). The type is stored as an index 579a7dea167SDimitry Andric // of a struct OpenCLType table. Multiple entries following each other form a 580a7dea167SDimitry Andric // signature. 5815ffd83dbSDimitry Andric OS << "static const unsigned short SignatureTable[] = {\n"; 582a7dea167SDimitry Andric for (const auto &P : SignaturesList) { 583a7dea167SDimitry Andric OS << " // " << P.second << "\n "; 584a7dea167SDimitry Andric for (const Record *R : P.first) { 5855ffd83dbSDimitry Andric unsigned Entry = TypeMap.find(R)->second; 5865ffd83dbSDimitry Andric if (Entry > USHRT_MAX) { 5875ffd83dbSDimitry Andric // Report an error when seeing an entry that is too large for the 5885ffd83dbSDimitry Andric // current index type (unsigned short). When hitting this, the type 5895ffd83dbSDimitry Andric // of SignatureTable will need to be changed. 5905ffd83dbSDimitry Andric PrintFatalError("Entry in SignatureTable exceeds limit."); 5915ffd83dbSDimitry Andric } 5925ffd83dbSDimitry Andric OS << Entry << ", "; 593a7dea167SDimitry Andric } 594a7dea167SDimitry Andric OS << "\n"; 5950b57cec5SDimitry Andric } 5960b57cec5SDimitry Andric OS << "};\n\n"; 5970b57cec5SDimitry Andric } 5980b57cec5SDimitry Andric 599fe6060f1SDimitry Andric // Encode a range MinVersion..MaxVersion into a single bit mask that can be 600fe6060f1SDimitry Andric // checked against LangOpts using isOpenCLVersionContainedInMask(). 601fe6060f1SDimitry Andric // This must be kept in sync with OpenCLVersionID in OpenCLOptions.h. 602fe6060f1SDimitry Andric // (Including OpenCLOptions.h here would be a layering violation.) 603fe6060f1SDimitry Andric static unsigned short EncodeVersions(unsigned int MinVersion, 604fe6060f1SDimitry Andric unsigned int MaxVersion) { 605fe6060f1SDimitry Andric unsigned short Encoded = 0; 606fe6060f1SDimitry Andric 607fe6060f1SDimitry Andric // A maximum version of 0 means available in all later versions. 608fe6060f1SDimitry Andric if (MaxVersion == 0) { 609fe6060f1SDimitry Andric MaxVersion = UINT_MAX; 610fe6060f1SDimitry Andric } 611fe6060f1SDimitry Andric 612fe6060f1SDimitry Andric unsigned VersionIDs[] = {100, 110, 120, 200, 300}; 613bdd1243dSDimitry Andric for (unsigned I = 0; I < std::size(VersionIDs); I++) { 614fe6060f1SDimitry Andric if (VersionIDs[I] >= MinVersion && VersionIDs[I] < MaxVersion) { 615fe6060f1SDimitry Andric Encoded |= 1 << I; 616fe6060f1SDimitry Andric } 617fe6060f1SDimitry Andric } 618fe6060f1SDimitry Andric 619fe6060f1SDimitry Andric return Encoded; 620fe6060f1SDimitry Andric } 621fe6060f1SDimitry Andric 6220b57cec5SDimitry Andric void BuiltinNameEmitter::EmitBuiltinTable() { 623a7dea167SDimitry Andric unsigned Index = 0; 624a7dea167SDimitry Andric 625a7dea167SDimitry Andric OS << "static const OpenCLBuiltinStruct BuiltinTable[] = {\n"; 626480093f4SDimitry Andric for (const auto &SLM : SignatureListMap) { 627a7dea167SDimitry Andric 628480093f4SDimitry Andric OS << " // " << (Index + 1) << ": "; 629480093f4SDimitry Andric for (const auto &Name : SLM.second.Names) { 630480093f4SDimitry Andric OS << Name << ", "; 631480093f4SDimitry Andric } 632480093f4SDimitry Andric OS << "\n"; 633a7dea167SDimitry Andric 634480093f4SDimitry Andric for (const auto &Overload : SLM.second.Signatures) { 635480093f4SDimitry Andric StringRef ExtName = Overload.first->getValueAsDef("Extension")->getName(); 636fe6060f1SDimitry Andric unsigned int MinVersion = 637fe6060f1SDimitry Andric Overload.first->getValueAsDef("MinVersion")->getValueAsInt("ID"); 638fe6060f1SDimitry Andric unsigned int MaxVersion = 639fe6060f1SDimitry Andric Overload.first->getValueAsDef("MaxVersion")->getValueAsInt("ID"); 640fe6060f1SDimitry Andric 641a7dea167SDimitry Andric OS << " { " << Overload.second << ", " 642a7dea167SDimitry Andric << Overload.first->getValueAsListOfDefs("Signature").size() << ", " 643480093f4SDimitry Andric << (Overload.first->getValueAsBit("IsPure")) << ", " 644480093f4SDimitry Andric << (Overload.first->getValueAsBit("IsConst")) << ", " 645480093f4SDimitry Andric << (Overload.first->getValueAsBit("IsConv")) << ", " 646480093f4SDimitry Andric << FunctionExtensionIndex[ExtName] << ", " 647fe6060f1SDimitry Andric << EncodeVersions(MinVersion, MaxVersion) << " },\n"; 648a7dea167SDimitry Andric Index++; 6490b57cec5SDimitry Andric } 6500b57cec5SDimitry Andric } 6510b57cec5SDimitry Andric OS << "};\n\n"; 6520b57cec5SDimitry Andric } 6530b57cec5SDimitry Andric 654480093f4SDimitry Andric bool BuiltinNameEmitter::CanReuseSignature( 655480093f4SDimitry Andric BuiltinIndexListTy *Candidate, 656480093f4SDimitry Andric std::vector<std::pair<const Record *, unsigned>> &SignatureList) { 657480093f4SDimitry Andric assert(Candidate->size() == SignatureList.size() && 658480093f4SDimitry Andric "signature lists should have the same size"); 659480093f4SDimitry Andric 660480093f4SDimitry Andric auto &CandidateSigs = 661480093f4SDimitry Andric SignatureListMap.find(Candidate)->second.Signatures; 662480093f4SDimitry Andric for (unsigned Index = 0; Index < Candidate->size(); Index++) { 663480093f4SDimitry Andric const Record *Rec = SignatureList[Index].first; 664480093f4SDimitry Andric const Record *Rec2 = CandidateSigs[Index].first; 665480093f4SDimitry Andric if (Rec->getValueAsBit("IsPure") == Rec2->getValueAsBit("IsPure") && 666480093f4SDimitry Andric Rec->getValueAsBit("IsConst") == Rec2->getValueAsBit("IsConst") && 667480093f4SDimitry Andric Rec->getValueAsBit("IsConv") == Rec2->getValueAsBit("IsConv") && 668480093f4SDimitry Andric Rec->getValueAsDef("MinVersion")->getValueAsInt("ID") == 669480093f4SDimitry Andric Rec2->getValueAsDef("MinVersion")->getValueAsInt("ID") && 670480093f4SDimitry Andric Rec->getValueAsDef("MaxVersion")->getValueAsInt("ID") == 671480093f4SDimitry Andric Rec2->getValueAsDef("MaxVersion")->getValueAsInt("ID") && 672480093f4SDimitry Andric Rec->getValueAsDef("Extension")->getName() == 673480093f4SDimitry Andric Rec2->getValueAsDef("Extension")->getName()) { 674480093f4SDimitry Andric return true; 675480093f4SDimitry Andric } 676480093f4SDimitry Andric } 677480093f4SDimitry Andric return false; 678480093f4SDimitry Andric } 679480093f4SDimitry Andric 680480093f4SDimitry Andric void BuiltinNameEmitter::GroupBySignature() { 681480093f4SDimitry Andric // List of signatures known to be emitted. 682480093f4SDimitry Andric std::vector<BuiltinIndexListTy *> KnownSignatures; 683480093f4SDimitry Andric 684480093f4SDimitry Andric for (auto &Fct : FctOverloadMap) { 685480093f4SDimitry Andric bool FoundReusableSig = false; 686480093f4SDimitry Andric 687480093f4SDimitry Andric // Gather all signatures for the current function. 688480093f4SDimitry Andric auto *CurSignatureList = new BuiltinIndexListTy(); 689480093f4SDimitry Andric for (const auto &Signature : Fct.second) { 690480093f4SDimitry Andric CurSignatureList->push_back(Signature.second); 691480093f4SDimitry Andric } 692480093f4SDimitry Andric // Sort the list to facilitate future comparisons. 6935ffd83dbSDimitry Andric llvm::sort(*CurSignatureList); 694480093f4SDimitry Andric 695480093f4SDimitry Andric // Check if we have already seen another function with the same list of 696480093f4SDimitry Andric // signatures. If so, just add the name of the function. 697480093f4SDimitry Andric for (auto *Candidate : KnownSignatures) { 698480093f4SDimitry Andric if (Candidate->size() == CurSignatureList->size() && 699480093f4SDimitry Andric *Candidate == *CurSignatureList) { 700480093f4SDimitry Andric if (CanReuseSignature(Candidate, Fct.second)) { 701480093f4SDimitry Andric SignatureListMap.find(Candidate)->second.Names.push_back(Fct.first); 702480093f4SDimitry Andric FoundReusableSig = true; 703480093f4SDimitry Andric } 704480093f4SDimitry Andric } 705480093f4SDimitry Andric } 706480093f4SDimitry Andric 707480093f4SDimitry Andric if (FoundReusableSig) { 708480093f4SDimitry Andric delete CurSignatureList; 709480093f4SDimitry Andric } else { 710480093f4SDimitry Andric // Add a new entry. 711480093f4SDimitry Andric SignatureListMap[CurSignatureList] = { 712480093f4SDimitry Andric SmallVector<StringRef, 4>(1, Fct.first), Fct.second}; 713480093f4SDimitry Andric KnownSignatures.push_back(CurSignatureList); 714480093f4SDimitry Andric } 715480093f4SDimitry Andric } 716480093f4SDimitry Andric 717480093f4SDimitry Andric for (auto *I : KnownSignatures) { 718480093f4SDimitry Andric delete I; 719480093f4SDimitry Andric } 720480093f4SDimitry Andric } 721480093f4SDimitry Andric 7220b57cec5SDimitry Andric void BuiltinNameEmitter::EmitStringMatcher() { 7230b57cec5SDimitry Andric std::vector<StringMatcher::StringPair> ValidBuiltins; 7240b57cec5SDimitry Andric unsigned CumulativeIndex = 1; 725480093f4SDimitry Andric 726480093f4SDimitry Andric for (const auto &SLM : SignatureListMap) { 727480093f4SDimitry Andric const auto &Ovl = SLM.second.Signatures; 728480093f4SDimitry Andric 729480093f4SDimitry Andric // A single signature list may be used by different builtins. Return the 730480093f4SDimitry Andric // same <index, length> pair for each of those builtins. 731480093f4SDimitry Andric for (const auto &FctName : SLM.second.Names) { 7320b57cec5SDimitry Andric std::string RetStmt; 7330b57cec5SDimitry Andric raw_string_ostream SS(RetStmt); 734480093f4SDimitry Andric SS << "return std::make_pair(" << CumulativeIndex << ", " << Ovl.size() 7350b57cec5SDimitry Andric << ");"; 7360b57cec5SDimitry Andric SS.flush(); 7375ffd83dbSDimitry Andric ValidBuiltins.push_back( 7385ffd83dbSDimitry Andric StringMatcher::StringPair(std::string(FctName), RetStmt)); 739480093f4SDimitry Andric } 740480093f4SDimitry Andric CumulativeIndex += Ovl.size(); 7410b57cec5SDimitry Andric } 7420b57cec5SDimitry Andric 7430b57cec5SDimitry Andric OS << R"( 744a7dea167SDimitry Andric // Find out whether a string matches an existing OpenCL builtin function name. 745a7dea167SDimitry Andric // Returns: A pair <0, 0> if no name matches. 746a7dea167SDimitry Andric // A pair <Index, Len> indexing the BuiltinTable if the name is 747a7dea167SDimitry Andric // matching an OpenCL builtin function. 748a7dea167SDimitry Andric static std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name) { 7490b57cec5SDimitry Andric 7500b57cec5SDimitry Andric )"; 7510b57cec5SDimitry Andric 752a7dea167SDimitry Andric StringMatcher("Name", ValidBuiltins, OS).Emit(0, true); 7530b57cec5SDimitry Andric 7540b57cec5SDimitry Andric OS << " return std::make_pair(0, 0);\n"; 755a7dea167SDimitry Andric OS << "} // isOpenCLBuiltin\n"; 7560b57cec5SDimitry Andric } 7570b57cec5SDimitry Andric 75881ad6265SDimitry Andric // Emit an if-statement with an isMacroDefined call for each extension in 75981ad6265SDimitry Andric // the space-separated list of extensions. 76081ad6265SDimitry Andric static void EmitMacroChecks(raw_ostream &OS, StringRef Extensions) { 76181ad6265SDimitry Andric SmallVector<StringRef, 2> ExtVec; 76281ad6265SDimitry Andric Extensions.split(ExtVec, " "); 76381ad6265SDimitry Andric OS << " if ("; 76481ad6265SDimitry Andric for (StringRef Ext : ExtVec) { 76581ad6265SDimitry Andric if (Ext != ExtVec.front()) 76681ad6265SDimitry Andric OS << " && "; 76781ad6265SDimitry Andric OS << "S.getPreprocessor().isMacroDefined(\"" << Ext << "\")"; 76881ad6265SDimitry Andric } 76981ad6265SDimitry Andric OS << ") {\n "; 77081ad6265SDimitry Andric } 77181ad6265SDimitry Andric 7720b57cec5SDimitry Andric void BuiltinNameEmitter::EmitQualTypeFinder() { 7730b57cec5SDimitry Andric OS << R"( 7740b57cec5SDimitry Andric 775fe6060f1SDimitry Andric static QualType getOpenCLEnumType(Sema &S, llvm::StringRef Name); 776fe6060f1SDimitry Andric static QualType getOpenCLTypedefType(Sema &S, llvm::StringRef Name); 777fe6060f1SDimitry Andric 778a7dea167SDimitry Andric // Convert an OpenCLTypeStruct type to a list of QualTypes. 779a7dea167SDimitry Andric // Generic types represent multiple types and vector sizes, thus a vector 780a7dea167SDimitry Andric // is returned. The conversion is done in two steps: 781a7dea167SDimitry Andric // Step 1: A switch statement fills a vector with scalar base types for the 782a7dea167SDimitry Andric // Cartesian product of (vector sizes) x (types) for generic types, 783a7dea167SDimitry Andric // or a single scalar type for non generic types. 784a7dea167SDimitry Andric // Step 2: Qualifiers and other type properties such as vector size are 785a7dea167SDimitry Andric // applied. 786fe6060f1SDimitry Andric static void OCL2Qual(Sema &S, const OpenCLTypeStruct &Ty, 787a7dea167SDimitry Andric llvm::SmallVectorImpl<QualType> &QT) { 788fe6060f1SDimitry Andric ASTContext &Context = S.Context; 789a7dea167SDimitry Andric // Number of scalar types in the GenType. 790a7dea167SDimitry Andric unsigned GenTypeNumTypes; 791a7dea167SDimitry Andric // Pointer to the list of vector sizes for the GenType. 792a7dea167SDimitry Andric llvm::ArrayRef<unsigned> GenVectorSizes; 7930b57cec5SDimitry Andric )"; 7940b57cec5SDimitry Andric 795a7dea167SDimitry Andric // Generate list of vector sizes for each generic type. 796a7dea167SDimitry Andric for (const auto *VectList : Records.getAllDerivedDefinitions("IntList")) { 797a7dea167SDimitry Andric OS << " constexpr unsigned List" 798a7dea167SDimitry Andric << VectList->getValueAsString("Name") << "[] = {"; 799a7dea167SDimitry Andric for (const auto V : VectList->getValueAsListOfInts("List")) { 800a7dea167SDimitry Andric OS << V << ", "; 801a7dea167SDimitry Andric } 802a7dea167SDimitry Andric OS << "};\n"; 803a7dea167SDimitry Andric } 804a7dea167SDimitry Andric 805a7dea167SDimitry Andric // Step 1. 806a7dea167SDimitry Andric // Start of switch statement over all types. 807a7dea167SDimitry Andric OS << "\n switch (Ty.ID) {\n"; 808a7dea167SDimitry Andric 809a7dea167SDimitry Andric // Switch cases for image types (Image2d, Image3d, ...) 810a7dea167SDimitry Andric std::vector<Record *> ImageTypes = 811a7dea167SDimitry Andric Records.getAllDerivedDefinitions("ImageType"); 812a7dea167SDimitry Andric 813a7dea167SDimitry Andric // Map an image type name to its 3 access-qualified types (RO, WO, RW). 814fe6060f1SDimitry Andric StringMap<SmallVector<Record *, 3>> ImageTypesMap; 815a7dea167SDimitry Andric for (auto *IT : ImageTypes) { 816a7dea167SDimitry Andric auto Entry = ImageTypesMap.find(IT->getValueAsString("Name")); 817a7dea167SDimitry Andric if (Entry == ImageTypesMap.end()) { 818a7dea167SDimitry Andric SmallVector<Record *, 3> ImageList; 819a7dea167SDimitry Andric ImageList.push_back(IT); 820a7dea167SDimitry Andric ImageTypesMap.insert( 821a7dea167SDimitry Andric std::make_pair(IT->getValueAsString("Name"), ImageList)); 822a7dea167SDimitry Andric } else { 823a7dea167SDimitry Andric Entry->second.push_back(IT); 824a7dea167SDimitry Andric } 825a7dea167SDimitry Andric } 826a7dea167SDimitry Andric 827a7dea167SDimitry Andric // Emit the cases for the image types. For an image type name, there are 3 828a7dea167SDimitry Andric // corresponding QualTypes ("RO", "WO", "RW"). The "AccessQualifier" field 829a7dea167SDimitry Andric // tells which one is needed. Emit a switch statement that puts the 830a7dea167SDimitry Andric // corresponding QualType into "QT". 831a7dea167SDimitry Andric for (const auto &ITE : ImageTypesMap) { 832fe6060f1SDimitry Andric OS << " case OCLT_" << ITE.getKey() << ":\n" 833a7dea167SDimitry Andric << " switch (Ty.AccessQualifier) {\n" 834a7dea167SDimitry Andric << " case OCLAQ_None:\n" 835a7dea167SDimitry Andric << " llvm_unreachable(\"Image without access qualifier\");\n"; 836fe6060f1SDimitry Andric for (const auto &Image : ITE.getValue()) { 83781ad6265SDimitry Andric StringRef Exts = 83881ad6265SDimitry Andric Image->getValueAsDef("Extension")->getValueAsString("ExtName"); 839a7dea167SDimitry Andric OS << StringSwitch<const char *>( 840a7dea167SDimitry Andric Image->getValueAsString("AccessQualifier")) 841a7dea167SDimitry Andric .Case("RO", " case OCLAQ_ReadOnly:\n") 842a7dea167SDimitry Andric .Case("WO", " case OCLAQ_WriteOnly:\n") 84381ad6265SDimitry Andric .Case("RW", " case OCLAQ_ReadWrite:\n"); 84481ad6265SDimitry Andric if (!Exts.empty()) { 84581ad6265SDimitry Andric OS << " "; 84681ad6265SDimitry Andric EmitMacroChecks(OS, Exts); 84781ad6265SDimitry Andric } 84881ad6265SDimitry Andric OS << " QT.push_back(" 849fe6060f1SDimitry Andric << Image->getValueAsDef("QTExpr")->getValueAsString("TypeExpr") 85081ad6265SDimitry Andric << ");\n"; 85181ad6265SDimitry Andric if (!Exts.empty()) { 85281ad6265SDimitry Andric OS << " }\n"; 85381ad6265SDimitry Andric } 85481ad6265SDimitry Andric OS << " break;\n"; 855a7dea167SDimitry Andric } 856a7dea167SDimitry Andric OS << " }\n" 857a7dea167SDimitry Andric << " break;\n"; 858a7dea167SDimitry Andric } 859a7dea167SDimitry Andric 860a7dea167SDimitry Andric // Switch cases for generic types. 861a7dea167SDimitry Andric for (const auto *GenType : Records.getAllDerivedDefinitions("GenericType")) { 862fe6060f1SDimitry Andric OS << " case OCLT_" << GenType->getValueAsString("Name") << ": {\n"; 863a7dea167SDimitry Andric 864a7dea167SDimitry Andric // Build the Cartesian product of (vector sizes) x (types). Only insert 865a7dea167SDimitry Andric // the plain scalar types for now; other type information such as vector 866a7dea167SDimitry Andric // size and type qualifiers will be added after the switch statement. 867fe6060f1SDimitry Andric std::vector<Record *> BaseTypes = 868fe6060f1SDimitry Andric GenType->getValueAsDef("TypeList")->getValueAsListOfDefs("List"); 869fe6060f1SDimitry Andric 870fe6060f1SDimitry Andric // Collect all QualTypes for a single vector size into TypeList. 871fe6060f1SDimitry Andric OS << " SmallVector<QualType, " << BaseTypes.size() << "> TypeList;\n"; 872fe6060f1SDimitry Andric for (const auto *T : BaseTypes) { 87381ad6265SDimitry Andric StringRef Exts = 874fe6060f1SDimitry Andric T->getValueAsDef("Extension")->getValueAsString("ExtName"); 87581ad6265SDimitry Andric if (!Exts.empty()) { 87681ad6265SDimitry Andric EmitMacroChecks(OS, Exts); 877fe6060f1SDimitry Andric } 878fe6060f1SDimitry Andric OS << " TypeList.push_back(" 879fe6060f1SDimitry Andric << T->getValueAsDef("QTExpr")->getValueAsString("TypeExpr") << ");\n"; 88081ad6265SDimitry Andric if (!Exts.empty()) { 881fe6060f1SDimitry Andric OS << " }\n"; 882a7dea167SDimitry Andric } 883a7dea167SDimitry Andric } 884fe6060f1SDimitry Andric OS << " GenTypeNumTypes = TypeList.size();\n"; 885fe6060f1SDimitry Andric 886fe6060f1SDimitry Andric // Duplicate the TypeList for every vector size. 887fe6060f1SDimitry Andric std::vector<int64_t> VectorList = 888fe6060f1SDimitry Andric GenType->getValueAsDef("VectorList")->getValueAsListOfInts("List"); 889fe6060f1SDimitry Andric OS << " QT.reserve(" << VectorList.size() * BaseTypes.size() << ");\n" 890fe6060f1SDimitry Andric << " for (unsigned I = 0; I < " << VectorList.size() << "; I++) {\n" 891fe6060f1SDimitry Andric << " QT.append(TypeList);\n" 892fe6060f1SDimitry Andric << " }\n"; 893fe6060f1SDimitry Andric 894a7dea167SDimitry Andric // GenVectorSizes is the list of vector sizes for this GenType. 895a7dea167SDimitry Andric OS << " GenVectorSizes = List" 896a7dea167SDimitry Andric << GenType->getValueAsDef("VectorList")->getValueAsString("Name") 897fe6060f1SDimitry Andric << ";\n" 898fe6060f1SDimitry Andric << " break;\n" 899fe6060f1SDimitry Andric << " }\n"; 900a7dea167SDimitry Andric } 901a7dea167SDimitry Andric 902a7dea167SDimitry Andric // Switch cases for non generic, non image types (int, int4, float, ...). 903a7dea167SDimitry Andric // Only insert the plain scalar type; vector information and type qualifiers 904a7dea167SDimitry Andric // are added in step 2. 9050b57cec5SDimitry Andric std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type"); 9060b57cec5SDimitry Andric StringMap<bool> TypesSeen; 9070b57cec5SDimitry Andric 9080b57cec5SDimitry Andric for (const auto *T : Types) { 909a7dea167SDimitry Andric // Check this is not an image type 91006c3fb27SDimitry Andric if (ImageTypesMap.contains(T->getValueAsString("Name"))) 911a7dea167SDimitry Andric continue; 9120b57cec5SDimitry Andric // Check we have not seen this Type 91306c3fb27SDimitry Andric if (TypesSeen.contains(T->getValueAsString("Name"))) 9140b57cec5SDimitry Andric continue; 9150b57cec5SDimitry Andric TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true)); 9160b57cec5SDimitry Andric 9170b57cec5SDimitry Andric // Check the Type does not have an "abstract" QualType 918fe6060f1SDimitry Andric auto QT = T->getValueAsDef("QTExpr"); 919a7dea167SDimitry Andric if (QT->getValueAsBit("IsAbstract") == 1) 9200b57cec5SDimitry Andric continue; 921a7dea167SDimitry Andric // Emit the cases for non generic, non image types. 9220b57cec5SDimitry Andric OS << " case OCLT_" << T->getValueAsString("Name") << ":\n"; 923fe6060f1SDimitry Andric 92481ad6265SDimitry Andric StringRef Exts = T->getValueAsDef("Extension")->getValueAsString("ExtName"); 92581ad6265SDimitry Andric // If this type depends on an extension, ensure the extension macros are 926fe6060f1SDimitry Andric // defined. 92781ad6265SDimitry Andric if (!Exts.empty()) { 92881ad6265SDimitry Andric EmitMacroChecks(OS, Exts); 929fe6060f1SDimitry Andric } 930fe6060f1SDimitry Andric OS << " QT.push_back(" << QT->getValueAsString("TypeExpr") << ");\n"; 93181ad6265SDimitry Andric if (!Exts.empty()) { 932fe6060f1SDimitry Andric OS << " }\n"; 933fe6060f1SDimitry Andric } 9340b57cec5SDimitry Andric OS << " break;\n"; 9350b57cec5SDimitry Andric } 9360b57cec5SDimitry Andric 937a7dea167SDimitry Andric // End of switch statement. 938480093f4SDimitry Andric OS << " } // end of switch (Ty.ID)\n\n"; 939a7dea167SDimitry Andric 940a7dea167SDimitry Andric // Step 2. 941a7dea167SDimitry Andric // Add ExtVector types if this was a generic type, as the switch statement 942a7dea167SDimitry Andric // above only populated the list with scalar types. This completes the 943a7dea167SDimitry Andric // construction of the Cartesian product of (vector sizes) x (types). 944a7dea167SDimitry Andric OS << " // Construct the different vector types for each generic type.\n"; 945a7dea167SDimitry Andric OS << " if (Ty.ID >= " << TypeList.size() << ") {"; 9460b57cec5SDimitry Andric OS << R"( 947a7dea167SDimitry Andric for (unsigned I = 0; I < QT.size(); I++) { 948a7dea167SDimitry Andric // For scalars, size is 1. 949a7dea167SDimitry Andric if (GenVectorSizes[I / GenTypeNumTypes] != 1) { 950a7dea167SDimitry Andric QT[I] = Context.getExtVectorType(QT[I], 951a7dea167SDimitry Andric GenVectorSizes[I / GenTypeNumTypes]); 9520b57cec5SDimitry Andric } 953a7dea167SDimitry Andric } 9540b57cec5SDimitry Andric } 9550b57cec5SDimitry Andric )"; 956a7dea167SDimitry Andric 957a7dea167SDimitry Andric // Assign the right attributes to the types (e.g. vector size). 958a7dea167SDimitry Andric OS << R"( 959a7dea167SDimitry Andric // Set vector size for non-generic vector types. 960a7dea167SDimitry Andric if (Ty.VectorWidth > 1) { 961a7dea167SDimitry Andric for (unsigned Index = 0; Index < QT.size(); Index++) { 962a7dea167SDimitry Andric QT[Index] = Context.getExtVectorType(QT[Index], Ty.VectorWidth); 963a7dea167SDimitry Andric } 9640b57cec5SDimitry Andric } 9650b57cec5SDimitry Andric 966a7dea167SDimitry Andric if (Ty.IsVolatile != 0) { 967a7dea167SDimitry Andric for (unsigned Index = 0; Index < QT.size(); Index++) { 968a7dea167SDimitry Andric QT[Index] = Context.getVolatileType(QT[Index]); 969a7dea167SDimitry Andric } 970a7dea167SDimitry Andric } 9710b57cec5SDimitry Andric 972a7dea167SDimitry Andric if (Ty.IsConst != 0) { 973a7dea167SDimitry Andric for (unsigned Index = 0; Index < QT.size(); Index++) { 974a7dea167SDimitry Andric QT[Index] = Context.getConstType(QT[Index]); 975a7dea167SDimitry Andric } 976a7dea167SDimitry Andric } 977a7dea167SDimitry Andric 978a7dea167SDimitry Andric // Transform the type to a pointer as the last step, if necessary. 979a7dea167SDimitry Andric // Builtin functions only have pointers on [const|volatile], no 980a7dea167SDimitry Andric // [const|volatile] pointers, so this is ok to do it as a last step. 981a7dea167SDimitry Andric if (Ty.IsPointer != 0) { 982a7dea167SDimitry Andric for (unsigned Index = 0; Index < QT.size(); Index++) { 983a7dea167SDimitry Andric QT[Index] = Context.getAddrSpaceQualType(QT[Index], Ty.AS); 984a7dea167SDimitry Andric QT[Index] = Context.getPointerType(QT[Index]); 985a7dea167SDimitry Andric } 986a7dea167SDimitry Andric } 987a7dea167SDimitry Andric )"; 988a7dea167SDimitry Andric 989a7dea167SDimitry Andric // End of the "OCL2Qual" function. 990a7dea167SDimitry Andric OS << "\n} // OCL2Qual\n"; 991a7dea167SDimitry Andric } 992a7dea167SDimitry Andric 993349cc55cSDimitry Andric std::string OpenCLBuiltinFileEmitterBase::getTypeString(const Record *Type, 994fe6060f1SDimitry Andric TypeFlags Flags, 995fe6060f1SDimitry Andric int VectorSize) const { 996fe6060f1SDimitry Andric std::string S; 997fe6060f1SDimitry Andric if (Type->getValueAsBit("IsConst") || Flags.IsConst) { 998fe6060f1SDimitry Andric S += "const "; 999fe6060f1SDimitry Andric } 1000fe6060f1SDimitry Andric if (Type->getValueAsBit("IsVolatile") || Flags.IsVolatile) { 1001fe6060f1SDimitry Andric S += "volatile "; 1002fe6060f1SDimitry Andric } 1003fe6060f1SDimitry Andric 1004fe6060f1SDimitry Andric auto PrintAddrSpace = [&S](StringRef AddrSpace) { 1005fe6060f1SDimitry Andric S += StringSwitch<const char *>(AddrSpace) 1006fe6060f1SDimitry Andric .Case("clang::LangAS::opencl_private", "__private") 1007fe6060f1SDimitry Andric .Case("clang::LangAS::opencl_global", "__global") 1008fe6060f1SDimitry Andric .Case("clang::LangAS::opencl_constant", "__constant") 1009fe6060f1SDimitry Andric .Case("clang::LangAS::opencl_local", "__local") 1010fe6060f1SDimitry Andric .Case("clang::LangAS::opencl_generic", "__generic") 1011fe6060f1SDimitry Andric .Default("__private"); 1012fe6060f1SDimitry Andric S += " "; 1013fe6060f1SDimitry Andric }; 1014fe6060f1SDimitry Andric if (Flags.IsPointer) { 1015fe6060f1SDimitry Andric PrintAddrSpace(Flags.AddrSpace); 1016fe6060f1SDimitry Andric } else if (Type->getValueAsBit("IsPointer")) { 1017fe6060f1SDimitry Andric PrintAddrSpace(Type->getValueAsString("AddrSpace")); 1018fe6060f1SDimitry Andric } 1019fe6060f1SDimitry Andric 1020fe6060f1SDimitry Andric StringRef Acc = Type->getValueAsString("AccessQualifier"); 1021fe6060f1SDimitry Andric if (Acc != "") { 1022fe6060f1SDimitry Andric S += StringSwitch<const char *>(Acc) 1023fe6060f1SDimitry Andric .Case("RO", "__read_only ") 1024fe6060f1SDimitry Andric .Case("WO", "__write_only ") 1025fe6060f1SDimitry Andric .Case("RW", "__read_write "); 1026fe6060f1SDimitry Andric } 1027fe6060f1SDimitry Andric 1028fe6060f1SDimitry Andric S += Type->getValueAsString("Name").str(); 1029fe6060f1SDimitry Andric if (VectorSize > 1) { 1030fe6060f1SDimitry Andric S += std::to_string(VectorSize); 1031fe6060f1SDimitry Andric } 1032fe6060f1SDimitry Andric 1033fe6060f1SDimitry Andric if (Type->getValueAsBit("IsPointer") || Flags.IsPointer) { 1034fe6060f1SDimitry Andric S += " *"; 1035fe6060f1SDimitry Andric } 1036fe6060f1SDimitry Andric 1037fe6060f1SDimitry Andric return S; 1038fe6060f1SDimitry Andric } 1039fe6060f1SDimitry Andric 1040349cc55cSDimitry Andric void OpenCLBuiltinFileEmitterBase::getTypeLists( 1041fe6060f1SDimitry Andric Record *Type, TypeFlags &Flags, std::vector<Record *> &TypeList, 1042fe6060f1SDimitry Andric std::vector<int64_t> &VectorList) const { 1043fe6060f1SDimitry Andric bool isGenType = Type->isSubClassOf("GenericType"); 1044fe6060f1SDimitry Andric if (isGenType) { 1045fe6060f1SDimitry Andric TypeList = Type->getValueAsDef("TypeList")->getValueAsListOfDefs("List"); 1046fe6060f1SDimitry Andric VectorList = 1047fe6060f1SDimitry Andric Type->getValueAsDef("VectorList")->getValueAsListOfInts("List"); 1048fe6060f1SDimitry Andric return; 1049fe6060f1SDimitry Andric } 1050fe6060f1SDimitry Andric 1051fe6060f1SDimitry Andric if (Type->isSubClassOf("PointerType") || Type->isSubClassOf("ConstType") || 1052fe6060f1SDimitry Andric Type->isSubClassOf("VolatileType")) { 1053fe6060f1SDimitry Andric StringRef SubTypeName = Type->getValueAsString("Name"); 1054fe6060f1SDimitry Andric Record *PossibleGenType = Records.getDef(SubTypeName); 1055fe6060f1SDimitry Andric if (PossibleGenType && PossibleGenType->isSubClassOf("GenericType")) { 1056fe6060f1SDimitry Andric // When PointerType, ConstType, or VolatileType is applied to a 1057fe6060f1SDimitry Andric // GenericType, the flags need to be taken from the subtype, not from the 1058fe6060f1SDimitry Andric // GenericType. 1059fe6060f1SDimitry Andric Flags.IsPointer = Type->getValueAsBit("IsPointer"); 1060fe6060f1SDimitry Andric Flags.IsConst = Type->getValueAsBit("IsConst"); 1061fe6060f1SDimitry Andric Flags.IsVolatile = Type->getValueAsBit("IsVolatile"); 1062fe6060f1SDimitry Andric Flags.AddrSpace = Type->getValueAsString("AddrSpace"); 1063fe6060f1SDimitry Andric getTypeLists(PossibleGenType, Flags, TypeList, VectorList); 1064fe6060f1SDimitry Andric return; 1065fe6060f1SDimitry Andric } 1066fe6060f1SDimitry Andric } 1067fe6060f1SDimitry Andric 1068fe6060f1SDimitry Andric // Not a GenericType, so just insert the single type. 1069fe6060f1SDimitry Andric TypeList.push_back(Type); 1070fe6060f1SDimitry Andric VectorList.push_back(Type->getValueAsInt("VecWidth")); 1071fe6060f1SDimitry Andric } 1072fe6060f1SDimitry Andric 1073349cc55cSDimitry Andric void OpenCLBuiltinFileEmitterBase::expandTypesInSignature( 1074fe6060f1SDimitry Andric const std::vector<Record *> &Signature, 1075fe6060f1SDimitry Andric SmallVectorImpl<SmallVector<std::string, 2>> &Types) { 1076fe6060f1SDimitry Andric // Find out if there are any GenTypes in this signature, and if so, calculate 1077fe6060f1SDimitry Andric // into how many signatures they will expand. 1078fe6060f1SDimitry Andric unsigned NumSignatures = 1; 1079fe6060f1SDimitry Andric SmallVector<SmallVector<std::string, 4>, 4> ExpandedGenTypes; 1080fe6060f1SDimitry Andric for (const auto &Arg : Signature) { 1081fe6060f1SDimitry Andric SmallVector<std::string, 4> ExpandedArg; 1082fe6060f1SDimitry Andric std::vector<Record *> TypeList; 1083fe6060f1SDimitry Andric std::vector<int64_t> VectorList; 1084fe6060f1SDimitry Andric TypeFlags Flags; 1085fe6060f1SDimitry Andric 1086fe6060f1SDimitry Andric getTypeLists(Arg, Flags, TypeList, VectorList); 1087fe6060f1SDimitry Andric 1088fe6060f1SDimitry Andric // Insert the Cartesian product of the types and vector sizes. 1089fe6060f1SDimitry Andric for (const auto &Vector : VectorList) { 1090fe6060f1SDimitry Andric for (const auto &Type : TypeList) { 109181ad6265SDimitry Andric std::string FullType = getTypeString(Type, Flags, Vector); 109281ad6265SDimitry Andric ExpandedArg.push_back(FullType); 109381ad6265SDimitry Andric 109481ad6265SDimitry Andric // If the type requires an extension, add a TypeExtMap entry mapping 109581ad6265SDimitry Andric // the full type name to the extension. 109681ad6265SDimitry Andric StringRef Ext = 109781ad6265SDimitry Andric Type->getValueAsDef("Extension")->getValueAsString("ExtName"); 109806c3fb27SDimitry Andric if (!Ext.empty() && !TypeExtMap.contains(FullType)) { 109981ad6265SDimitry Andric TypeExtMap.insert({FullType, Ext}); 110081ad6265SDimitry Andric } 1101fe6060f1SDimitry Andric } 1102fe6060f1SDimitry Andric } 1103fe6060f1SDimitry Andric NumSignatures = std::max<unsigned>(NumSignatures, ExpandedArg.size()); 1104fe6060f1SDimitry Andric ExpandedGenTypes.push_back(ExpandedArg); 1105fe6060f1SDimitry Andric } 1106fe6060f1SDimitry Andric 1107fe6060f1SDimitry Andric // Now the total number of signatures is known. Populate the return list with 1108fe6060f1SDimitry Andric // all signatures. 1109fe6060f1SDimitry Andric for (unsigned I = 0; I < NumSignatures; I++) { 1110fe6060f1SDimitry Andric SmallVector<std::string, 2> Args; 1111fe6060f1SDimitry Andric 1112fe6060f1SDimitry Andric // Process a single signature. 1113fe6060f1SDimitry Andric for (unsigned ArgNum = 0; ArgNum < Signature.size(); ArgNum++) { 1114fe6060f1SDimitry Andric // For differently-sized GenTypes in a parameter list, the smaller 1115fe6060f1SDimitry Andric // GenTypes just repeat, so index modulo the number of expanded types. 1116fe6060f1SDimitry Andric size_t TypeIndex = I % ExpandedGenTypes[ArgNum].size(); 1117fe6060f1SDimitry Andric Args.push_back(ExpandedGenTypes[ArgNum][TypeIndex]); 1118fe6060f1SDimitry Andric } 1119fe6060f1SDimitry Andric Types.push_back(Args); 1120fe6060f1SDimitry Andric } 1121fe6060f1SDimitry Andric } 1122fe6060f1SDimitry Andric 1123349cc55cSDimitry Andric void OpenCLBuiltinFileEmitterBase::emitExtensionSetup() { 1124fe6060f1SDimitry Andric OS << R"( 1125fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_fp16 : enable 1126fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_fp64 : enable 1127fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_int64_base_atomics : enable 1128fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_int64_extended_atomics : enable 1129fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_gl_msaa_sharing : enable 1130fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_mipmap_image_writes : enable 1131fe6060f1SDimitry Andric #pragma OPENCL EXTENSION cl_khr_3d_image_writes : enable 1132fe6060f1SDimitry Andric 1133fe6060f1SDimitry Andric )"; 1134349cc55cSDimitry Andric } 1135349cc55cSDimitry Andric 1136349cc55cSDimitry Andric std::string 1137349cc55cSDimitry Andric OpenCLBuiltinFileEmitterBase::emitExtensionGuard(const Record *Builtin) { 1138349cc55cSDimitry Andric StringRef Extensions = 1139349cc55cSDimitry Andric Builtin->getValueAsDef("Extension")->getValueAsString("ExtName"); 1140349cc55cSDimitry Andric if (Extensions.empty()) 1141349cc55cSDimitry Andric return ""; 1142349cc55cSDimitry Andric 1143349cc55cSDimitry Andric OS << "#if"; 1144349cc55cSDimitry Andric 1145349cc55cSDimitry Andric SmallVector<StringRef, 2> ExtVec; 1146349cc55cSDimitry Andric Extensions.split(ExtVec, " "); 1147349cc55cSDimitry Andric bool isFirst = true; 1148349cc55cSDimitry Andric for (StringRef Ext : ExtVec) { 1149349cc55cSDimitry Andric if (!isFirst) { 1150349cc55cSDimitry Andric OS << " &&"; 1151349cc55cSDimitry Andric } 1152349cc55cSDimitry Andric OS << " defined(" << Ext << ")"; 1153349cc55cSDimitry Andric isFirst = false; 1154349cc55cSDimitry Andric } 1155349cc55cSDimitry Andric OS << "\n"; 1156349cc55cSDimitry Andric 1157349cc55cSDimitry Andric return "#endif // Extension\n"; 1158349cc55cSDimitry Andric } 1159349cc55cSDimitry Andric 1160349cc55cSDimitry Andric std::string 1161349cc55cSDimitry Andric OpenCLBuiltinFileEmitterBase::emitVersionGuard(const Record *Builtin) { 1162349cc55cSDimitry Andric std::string OptionalEndif; 1163349cc55cSDimitry Andric auto PrintOpenCLVersion = [this](int Version) { 1164349cc55cSDimitry Andric OS << "CL_VERSION_" << (Version / 100) << "_" << ((Version % 100) / 10); 1165349cc55cSDimitry Andric }; 1166349cc55cSDimitry Andric int MinVersion = Builtin->getValueAsDef("MinVersion")->getValueAsInt("ID"); 1167349cc55cSDimitry Andric if (MinVersion != 100) { 1168349cc55cSDimitry Andric // OpenCL 1.0 is the default minimum version. 1169349cc55cSDimitry Andric OS << "#if __OPENCL_C_VERSION__ >= "; 1170349cc55cSDimitry Andric PrintOpenCLVersion(MinVersion); 1171349cc55cSDimitry Andric OS << "\n"; 1172349cc55cSDimitry Andric OptionalEndif = "#endif // MinVersion\n" + OptionalEndif; 1173349cc55cSDimitry Andric } 1174349cc55cSDimitry Andric int MaxVersion = Builtin->getValueAsDef("MaxVersion")->getValueAsInt("ID"); 1175349cc55cSDimitry Andric if (MaxVersion) { 1176349cc55cSDimitry Andric OS << "#if __OPENCL_C_VERSION__ < "; 1177349cc55cSDimitry Andric PrintOpenCLVersion(MaxVersion); 1178349cc55cSDimitry Andric OS << "\n"; 1179349cc55cSDimitry Andric OptionalEndif = "#endif // MaxVersion\n" + OptionalEndif; 1180349cc55cSDimitry Andric } 1181349cc55cSDimitry Andric return OptionalEndif; 1182349cc55cSDimitry Andric } 1183349cc55cSDimitry Andric 118481ad6265SDimitry Andric StringRef OpenCLBuiltinFileEmitterBase::emitTypeExtensionGuards( 118581ad6265SDimitry Andric const SmallVectorImpl<std::string> &Signature) { 118681ad6265SDimitry Andric SmallSet<StringRef, 2> ExtSet; 118781ad6265SDimitry Andric 118881ad6265SDimitry Andric // Iterate over all types to gather the set of required TypeExtensions. 118981ad6265SDimitry Andric for (const auto &Ty : Signature) { 119081ad6265SDimitry Andric StringRef TypeExt = TypeExtMap.lookup(Ty); 119181ad6265SDimitry Andric if (!TypeExt.empty()) { 119281ad6265SDimitry Andric // The TypeExtensions are space-separated in the .td file. 119381ad6265SDimitry Andric SmallVector<StringRef, 2> ExtVec; 119481ad6265SDimitry Andric TypeExt.split(ExtVec, " "); 119581ad6265SDimitry Andric for (const auto Ext : ExtVec) { 119681ad6265SDimitry Andric ExtSet.insert(Ext); 119781ad6265SDimitry Andric } 119881ad6265SDimitry Andric } 119981ad6265SDimitry Andric } 120081ad6265SDimitry Andric 120181ad6265SDimitry Andric // Emit the #if only when at least one extension is required. 120281ad6265SDimitry Andric if (ExtSet.empty()) 120381ad6265SDimitry Andric return ""; 120481ad6265SDimitry Andric 120581ad6265SDimitry Andric OS << "#if "; 120681ad6265SDimitry Andric bool isFirst = true; 120781ad6265SDimitry Andric for (const auto Ext : ExtSet) { 120881ad6265SDimitry Andric if (!isFirst) 120981ad6265SDimitry Andric OS << " && "; 121081ad6265SDimitry Andric OS << "defined(" << Ext << ")"; 121181ad6265SDimitry Andric isFirst = false; 121281ad6265SDimitry Andric } 121381ad6265SDimitry Andric OS << "\n"; 121481ad6265SDimitry Andric return "#endif // TypeExtension\n"; 121581ad6265SDimitry Andric } 121681ad6265SDimitry Andric 1217349cc55cSDimitry Andric void OpenCLBuiltinTestEmitter::emit() { 1218*5f757f3fSDimitry Andric emitSourceFileHeader("OpenCL Builtin exhaustive testing", OS, Records); 1219349cc55cSDimitry Andric 1220349cc55cSDimitry Andric emitExtensionSetup(); 1221fe6060f1SDimitry Andric 1222fe6060f1SDimitry Andric // Ensure each test has a unique name by numbering them. 1223fe6060f1SDimitry Andric unsigned TestID = 0; 1224fe6060f1SDimitry Andric 1225fe6060f1SDimitry Andric // Iterate over all builtins. 1226fe6060f1SDimitry Andric std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin"); 1227fe6060f1SDimitry Andric for (const auto *B : Builtins) { 1228fe6060f1SDimitry Andric StringRef Name = B->getValueAsString("Name"); 1229fe6060f1SDimitry Andric 1230fe6060f1SDimitry Andric SmallVector<SmallVector<std::string, 2>, 4> FTypes; 1231fe6060f1SDimitry Andric expandTypesInSignature(B->getValueAsListOfDefs("Signature"), FTypes); 1232fe6060f1SDimitry Andric 1233fe6060f1SDimitry Andric OS << "// Test " << Name << "\n"; 1234fe6060f1SDimitry Andric 1235349cc55cSDimitry Andric std::string OptionalExtensionEndif = emitExtensionGuard(B); 1236349cc55cSDimitry Andric std::string OptionalVersionEndif = emitVersionGuard(B); 1237349cc55cSDimitry Andric 1238fe6060f1SDimitry Andric for (const auto &Signature : FTypes) { 123981ad6265SDimitry Andric StringRef OptionalTypeExtEndif = emitTypeExtensionGuards(Signature); 124081ad6265SDimitry Andric 1241fe6060f1SDimitry Andric // Emit function declaration. 1242fe6060f1SDimitry Andric OS << Signature[0] << " test" << TestID++ << "_" << Name << "("; 1243fe6060f1SDimitry Andric if (Signature.size() > 1) { 1244fe6060f1SDimitry Andric for (unsigned I = 1; I < Signature.size(); I++) { 1245fe6060f1SDimitry Andric if (I != 1) 1246fe6060f1SDimitry Andric OS << ", "; 1247fe6060f1SDimitry Andric OS << Signature[I] << " arg" << I; 1248fe6060f1SDimitry Andric } 1249fe6060f1SDimitry Andric } 1250fe6060f1SDimitry Andric OS << ") {\n"; 1251fe6060f1SDimitry Andric 1252fe6060f1SDimitry Andric // Emit function body. 1253fe6060f1SDimitry Andric OS << " "; 1254fe6060f1SDimitry Andric if (Signature[0] != "void") { 1255fe6060f1SDimitry Andric OS << "return "; 1256fe6060f1SDimitry Andric } 1257fe6060f1SDimitry Andric OS << Name << "("; 1258fe6060f1SDimitry Andric for (unsigned I = 1; I < Signature.size(); I++) { 1259fe6060f1SDimitry Andric if (I != 1) 1260fe6060f1SDimitry Andric OS << ", "; 1261fe6060f1SDimitry Andric OS << "arg" << I; 1262fe6060f1SDimitry Andric } 1263fe6060f1SDimitry Andric OS << ");\n"; 1264fe6060f1SDimitry Andric 1265fe6060f1SDimitry Andric // End of function body. 1266fe6060f1SDimitry Andric OS << "}\n"; 126781ad6265SDimitry Andric OS << OptionalTypeExtEndif; 1268fe6060f1SDimitry Andric } 1269349cc55cSDimitry Andric 1270349cc55cSDimitry Andric OS << OptionalVersionEndif; 1271349cc55cSDimitry Andric OS << OptionalExtensionEndif; 1272fe6060f1SDimitry Andric } 1273fe6060f1SDimitry Andric } 1274fe6060f1SDimitry Andric 127506c3fb27SDimitry Andric void OpenCLBuiltinHeaderEmitter::emit() { 1276*5f757f3fSDimitry Andric emitSourceFileHeader("OpenCL Builtin declarations", OS, Records); 127706c3fb27SDimitry Andric 127806c3fb27SDimitry Andric emitExtensionSetup(); 127906c3fb27SDimitry Andric 128006c3fb27SDimitry Andric OS << R"( 128106c3fb27SDimitry Andric #define __ovld __attribute__((overloadable)) 128206c3fb27SDimitry Andric #define __conv __attribute__((convergent)) 128306c3fb27SDimitry Andric #define __purefn __attribute__((pure)) 128406c3fb27SDimitry Andric #define __cnfn __attribute__((const)) 128506c3fb27SDimitry Andric 128606c3fb27SDimitry Andric )"; 128706c3fb27SDimitry Andric 128806c3fb27SDimitry Andric // Iterate over all builtins; sort to follow order of definition in .td file. 128906c3fb27SDimitry Andric std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin"); 129006c3fb27SDimitry Andric llvm::sort(Builtins, LessRecord()); 129106c3fb27SDimitry Andric 129206c3fb27SDimitry Andric for (const auto *B : Builtins) { 129306c3fb27SDimitry Andric StringRef Name = B->getValueAsString("Name"); 129406c3fb27SDimitry Andric 129506c3fb27SDimitry Andric std::string OptionalExtensionEndif = emitExtensionGuard(B); 129606c3fb27SDimitry Andric std::string OptionalVersionEndif = emitVersionGuard(B); 129706c3fb27SDimitry Andric 129806c3fb27SDimitry Andric SmallVector<SmallVector<std::string, 2>, 4> FTypes; 129906c3fb27SDimitry Andric expandTypesInSignature(B->getValueAsListOfDefs("Signature"), FTypes); 130006c3fb27SDimitry Andric 130106c3fb27SDimitry Andric for (const auto &Signature : FTypes) { 130206c3fb27SDimitry Andric StringRef OptionalTypeExtEndif = emitTypeExtensionGuards(Signature); 130306c3fb27SDimitry Andric 130406c3fb27SDimitry Andric // Emit function declaration. 130506c3fb27SDimitry Andric OS << Signature[0] << " __ovld "; 130606c3fb27SDimitry Andric if (B->getValueAsBit("IsConst")) 130706c3fb27SDimitry Andric OS << "__cnfn "; 130806c3fb27SDimitry Andric if (B->getValueAsBit("IsPure")) 130906c3fb27SDimitry Andric OS << "__purefn "; 131006c3fb27SDimitry Andric if (B->getValueAsBit("IsConv")) 131106c3fb27SDimitry Andric OS << "__conv "; 131206c3fb27SDimitry Andric 131306c3fb27SDimitry Andric OS << Name << "("; 131406c3fb27SDimitry Andric if (Signature.size() > 1) { 131506c3fb27SDimitry Andric for (unsigned I = 1; I < Signature.size(); I++) { 131606c3fb27SDimitry Andric if (I != 1) 131706c3fb27SDimitry Andric OS << ", "; 131806c3fb27SDimitry Andric OS << Signature[I]; 131906c3fb27SDimitry Andric } 132006c3fb27SDimitry Andric } 132106c3fb27SDimitry Andric OS << ");\n"; 132206c3fb27SDimitry Andric 132306c3fb27SDimitry Andric OS << OptionalTypeExtEndif; 132406c3fb27SDimitry Andric } 132506c3fb27SDimitry Andric 132606c3fb27SDimitry Andric OS << OptionalVersionEndif; 132706c3fb27SDimitry Andric OS << OptionalExtensionEndif; 132806c3fb27SDimitry Andric } 132906c3fb27SDimitry Andric 133006c3fb27SDimitry Andric OS << "\n// Disable any extensions we may have enabled previously.\n" 133106c3fb27SDimitry Andric "#pragma OPENCL EXTENSION all : disable\n"; 133206c3fb27SDimitry Andric } 133306c3fb27SDimitry Andric 1334a7dea167SDimitry Andric void clang::EmitClangOpenCLBuiltins(RecordKeeper &Records, raw_ostream &OS) { 13350b57cec5SDimitry Andric BuiltinNameEmitter NameChecker(Records, OS); 13360b57cec5SDimitry Andric NameChecker.Emit(); 13370b57cec5SDimitry Andric } 1338fe6060f1SDimitry Andric 133906c3fb27SDimitry Andric void clang::EmitClangOpenCLBuiltinHeader(RecordKeeper &Records, 134006c3fb27SDimitry Andric raw_ostream &OS) { 134106c3fb27SDimitry Andric OpenCLBuiltinHeaderEmitter HeaderFileGenerator(Records, OS); 134206c3fb27SDimitry Andric HeaderFileGenerator.emit(); 134306c3fb27SDimitry Andric } 134406c3fb27SDimitry Andric 1345fe6060f1SDimitry Andric void clang::EmitClangOpenCLBuiltinTests(RecordKeeper &Records, 1346fe6060f1SDimitry Andric raw_ostream &OS) { 1347fe6060f1SDimitry Andric OpenCLBuiltinTestEmitter TestFileGenerator(Records, OS); 1348fe6060f1SDimitry Andric TestFileGenerator.emit(); 1349fe6060f1SDimitry Andric } 1350