1//===- ClangOpenCLBuiltinEmitter.cpp - Generate Clang OpenCL Builtin handling
2//
3//                     The LLVM Compiler Infrastructure
4//
5// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
6// See https://llvm.org/LICENSE.txt for license information.
7// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
8//
9//===----------------------------------------------------------------------===//
10//
11// This tablegen backend emits code for checking whether a function is an
12// OpenCL builtin function. If so, all overloads of this function are
13// added to the LookupResult. The generated include file is used by
14// SemaLookup.cpp
15//
16// For a successful lookup of e.g. the "cos" builtin, isOpenCLBuiltin("cos")
17// returns a pair <Index, Len>.
18// BuiltinTable[Index] to BuiltinTable[Index + Len] contains the pairs
19// <SigIndex, SigLen> of the overloads of "cos".
20// SignatureTable[SigIndex] to SignatureTable[SigIndex + SigLen] contains
21// one of the signatures of "cos". The SignatureTable entry can be
22// referenced by other functions, e.g. "sin", to exploit the fact that
23// many OpenCL builtins share the same signature.
24//
25// The file generated by this TableGen emitter contains the following:
26//
27//  * Structs and enums to represent types and function signatures.
28//
29//  * const char *FunctionExtensionTable[]
30//    List of space-separated OpenCL extensions.  A builtin references an
31//    entry in this table when the builtin requires a particular (set of)
32//    extension(s) to be enabled.
33//
34//  * OpenCLTypeStruct TypeTable[]
35//    Type information for return types and arguments.
36//
37//  * unsigned SignatureTable[]
38//    A list of types representing function signatures.  Each entry is an index
39//    into the above TypeTable.  Multiple entries following each other form a
40//    signature, where the first entry is the return type and subsequent
41//    entries are the argument types.
42//
43//  * OpenCLBuiltinStruct BuiltinTable[]
44//    Each entry represents one overload of an OpenCL builtin function and
45//    consists of an index into the SignatureTable and the number of arguments.
46//
47//  * std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name)
48//    Find out whether a string matches an existing OpenCL builtin function
49//    name and return an index into BuiltinTable and the number of overloads.
50//
51//  * void OCL2Qual(ASTContext&, OpenCLTypeStruct, std::vector<QualType>&)
52//    Convert an OpenCLTypeStruct type to a list of QualType instances.
53//    One OpenCLTypeStruct can represent multiple types, primarily when using
54//    GenTypes.
55//
56//===----------------------------------------------------------------------===//
57
58#include "TableGenBackends.h"
59#include "llvm/ADT/MapVector.h"
60#include "llvm/ADT/STLExtras.h"
61#include "llvm/ADT/SmallString.h"
62#include "llvm/ADT/StringExtras.h"
63#include "llvm/ADT/StringRef.h"
64#include "llvm/ADT/StringSet.h"
65#include "llvm/ADT/StringSwitch.h"
66#include "llvm/Support/ErrorHandling.h"
67#include "llvm/Support/raw_ostream.h"
68#include "llvm/TableGen/Error.h"
69#include "llvm/TableGen/Record.h"
70#include "llvm/TableGen/StringMatcher.h"
71#include "llvm/TableGen/TableGenBackend.h"
72#include <set>
73
74using namespace llvm;
75
76namespace {
77
78// A list of signatures that are shared by one or more builtin functions.
79struct BuiltinTableEntries {
80  SmallVector<StringRef, 4> Names;
81  std::vector<std::pair<const Record *, unsigned>> Signatures;
82};
83
84class BuiltinNameEmitter {
85public:
86  BuiltinNameEmitter(RecordKeeper &Records, raw_ostream &OS)
87      : Records(Records), OS(OS) {}
88
89  // Entrypoint to generate the functions and structures for checking
90  // whether a function is an OpenCL builtin function.
91  void Emit();
92
93private:
94  // A list of indices into the builtin function table.
95  using BuiltinIndexListTy = SmallVector<unsigned, 11>;
96
97  // Contains OpenCL builtin functions and related information, stored as
98  // Record instances. They are coming from the associated TableGen file.
99  RecordKeeper &Records;
100
101  // The output file.
102  raw_ostream &OS;
103
104  // Helper function for BuiltinNameEmitter::EmitDeclarations.  Generate enum
105  // definitions in the Output string parameter, and save their Record instances
106  // in the List parameter.
107  // \param Types (in) List containing the Types to extract.
108  // \param TypesSeen (inout) List containing the Types already extracted.
109  // \param Output (out) String containing the enums to emit in the output file.
110  // \param List (out) List containing the extracted Types, except the Types in
111  //        TypesSeen.
112  void ExtractEnumTypes(std::vector<Record *> &Types,
113                        StringMap<bool> &TypesSeen, std::string &Output,
114                        std::vector<const Record *> &List);
115
116  // Emit the enum or struct used in the generated file.
117  // Populate the TypeList at the same time.
118  void EmitDeclarations();
119
120  // Parse the Records generated by TableGen to populate the SignaturesList,
121  // FctOverloadMap and TypeMap.
122  void GetOverloads();
123
124  // Compare two lists of signatures and check that e.g. the OpenCL version,
125  // function attributes, and extension are equal for each signature.
126  // \param Candidate (in) Entry in the SignatureListMap to check.
127  // \param SignatureList (in) List of signatures of the considered function.
128  // \returns true if the two lists of signatures are identical.
129  bool CanReuseSignature(
130      BuiltinIndexListTy *Candidate,
131      std::vector<std::pair<const Record *, unsigned>> &SignatureList);
132
133  // Group functions with the same list of signatures by populating the
134  // SignatureListMap.
135  // Some builtin functions have the same list of signatures, for example the
136  // "sin" and "cos" functions. To save space in the BuiltinTable, the
137  // "isOpenCLBuiltin" function will have the same output for these two
138  // function names.
139  void GroupBySignature();
140
141  // Emit the FunctionExtensionTable that lists all function extensions.
142  void EmitExtensionTable();
143
144  // Emit the TypeTable containing all types used by OpenCL builtins.
145  void EmitTypeTable();
146
147  // Emit the SignatureTable. This table contains all the possible signatures.
148  // A signature is stored as a list of indexes of the TypeTable.
149  // The first index references the return type (mandatory), and the followings
150  // reference its arguments.
151  // E.g.:
152  // 15, 2, 15 can represent a function with the signature:
153  // int func(float, int)
154  // The "int" type being at the index 15 in the TypeTable.
155  void EmitSignatureTable();
156
157  // Emit the BuiltinTable table. This table contains all the overloads of
158  // each function, and is a struct OpenCLBuiltinDecl.
159  // E.g.:
160  // // 891 convert_float2_rtn
161  //   { 58, 2, 3, 100, 0 },
162  // This means that the signature of this convert_float2_rtn overload has
163  // 1 argument (+1 for the return type), stored at index 58 in
164  // the SignatureTable.  This prototype requires extension "3" in the
165  // FunctionExtensionTable.  The last two values represent the minimum (1.0)
166  // and maximum (0, meaning no max version) OpenCL version in which this
167  // overload is supported.
168  void EmitBuiltinTable();
169
170  // Emit a StringMatcher function to check whether a function name is an
171  // OpenCL builtin function name.
172  void EmitStringMatcher();
173
174  // Emit a function returning the clang QualType instance associated with
175  // the TableGen Record Type.
176  void EmitQualTypeFinder();
177
178  // Contains a list of the available signatures, without the name of the
179  // function. Each pair consists of a signature and a cumulative index.
180  // E.g.:  <<float, float>, 0>,
181  //        <<float, int, int, 2>>,
182  //        <<float>, 5>,
183  //        ...
184  //        <<double, double>, 35>.
185  std::vector<std::pair<std::vector<Record *>, unsigned>> SignaturesList;
186
187  // Map the name of a builtin function to its prototypes (instances of the
188  // TableGen "Builtin" class).
189  // Each prototype is registered as a pair of:
190  //   <pointer to the "Builtin" instance,
191  //    cumulative index of the associated signature in the SignaturesList>
192  // E.g.:  The function cos: (float cos(float), double cos(double), ...)
193  //        <"cos", <<ptrToPrototype0, 5>,
194  //                 <ptrToPrototype1, 35>,
195  //                 <ptrToPrototype2, 79>>
196  // ptrToPrototype1 has the following signature: <double, double>
197  MapVector<StringRef, std::vector<std::pair<const Record *, unsigned>>>
198      FctOverloadMap;
199
200  // Contains the map of OpenCL types to their index in the TypeTable.
201  MapVector<const Record *, unsigned> TypeMap;
202
203  // List of OpenCL function extensions mapping extension strings to
204  // an index into the FunctionExtensionTable.
205  StringMap<unsigned> FunctionExtensionIndex;
206
207  // List of OpenCL type names in the same order as in enum OpenCLTypeID.
208  // This list does not contain generic types.
209  std::vector<const Record *> TypeList;
210
211  // Same as TypeList, but for generic types only.
212  std::vector<const Record *> GenTypeList;
213
214  // Map an ordered vector of signatures to their original Record instances,
215  // and to a list of function names that share these signatures.
216  //
217  // For example, suppose the "cos" and "sin" functions have only three
218  // signatures, and these signatures are at index Ix in the SignatureTable:
219  //          cos         |         sin         |  Signature    | Index
220  //  float   cos(float)  | float   sin(float)  |  Signature1   | I1
221  //  double  cos(double) | double  sin(double) |  Signature2   | I2
222  //  half    cos(half)   | half    sin(half)   |  Signature3   | I3
223  //
224  // Then we will create a mapping of the vector of signatures:
225  // SignatureListMap[<I1, I2, I3>] = <
226  //                  <"cos", "sin">,
227  //                  <Signature1, Signature2, Signature3>>
228  // The function "tan", having the same signatures, would be mapped to the
229  // same entry (<I1, I2, I3>).
230  MapVector<BuiltinIndexListTy *, BuiltinTableEntries> SignatureListMap;
231};
232} // namespace
233
234void BuiltinNameEmitter::Emit() {
235  emitSourceFileHeader("OpenCL Builtin handling", OS);
236
237  OS << "#include \"llvm/ADT/StringRef.h\"\n";
238  OS << "using namespace clang;\n\n";
239
240  // Emit enums and structs.
241  EmitDeclarations();
242
243  // Parse the Records to populate the internal lists.
244  GetOverloads();
245  GroupBySignature();
246
247  // Emit tables.
248  EmitExtensionTable();
249  EmitTypeTable();
250  EmitSignatureTable();
251  EmitBuiltinTable();
252
253  // Emit functions.
254  EmitStringMatcher();
255  EmitQualTypeFinder();
256}
257
258void BuiltinNameEmitter::ExtractEnumTypes(std::vector<Record *> &Types,
259                                          StringMap<bool> &TypesSeen,
260                                          std::string &Output,
261                                          std::vector<const Record *> &List) {
262  raw_string_ostream SS(Output);
263
264  for (const auto *T : Types) {
265    if (TypesSeen.find(T->getValueAsString("Name")) == TypesSeen.end()) {
266      SS << "  OCLT_" + T->getValueAsString("Name") << ",\n";
267      // Save the type names in the same order as their enum value. Note that
268      // the Record can be a VectorType or something else, only the name is
269      // important.
270      List.push_back(T);
271      TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true));
272    }
273  }
274  SS.flush();
275}
276
277void BuiltinNameEmitter::EmitDeclarations() {
278  // Enum of scalar type names (float, int, ...) and generic type sets.
279  OS << "enum OpenCLTypeID {\n";
280
281  StringMap<bool> TypesSeen;
282  std::string GenTypeEnums;
283  std::string TypeEnums;
284
285  // Extract generic types and non-generic types separately, to keep
286  // gentypes at the end of the enum which simplifies the special handling
287  // for gentypes in SemaLookup.
288  std::vector<Record *> GenTypes =
289      Records.getAllDerivedDefinitions("GenericType");
290  ExtractEnumTypes(GenTypes, TypesSeen, GenTypeEnums, GenTypeList);
291
292  std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
293  ExtractEnumTypes(Types, TypesSeen, TypeEnums, TypeList);
294
295  OS << TypeEnums;
296  OS << GenTypeEnums;
297  OS << "};\n";
298
299  // Structure definitions.
300  OS << R"(
301// Image access qualifier.
302enum OpenCLAccessQual : unsigned char {
303  OCLAQ_None,
304  OCLAQ_ReadOnly,
305  OCLAQ_WriteOnly,
306  OCLAQ_ReadWrite
307};
308
309// Represents a return type or argument type.
310struct OpenCLTypeStruct {
311  // A type (e.g. float, int, ...).
312  const OpenCLTypeID ID;
313  // Vector size (if applicable; 0 for scalars and generic types).
314  const unsigned VectorWidth;
315  // 0 if the type is not a pointer.
316  const bool IsPointer;
317  // 0 if the type is not const.
318  const bool IsConst;
319  // 0 if the type is not volatile.
320  const bool IsVolatile;
321  // Access qualifier.
322  const OpenCLAccessQual AccessQualifier;
323  // Address space of the pointer (if applicable).
324  const LangAS AS;
325};
326
327// One overload of an OpenCL builtin function.
328struct OpenCLBuiltinStruct {
329  // Index of the signature in the OpenCLTypeStruct table.
330  const unsigned SigTableIndex;
331  // Entries between index SigTableIndex and (SigTableIndex + NumTypes - 1) in
332  // the SignatureTable represent the complete signature.  The first type at
333  // index SigTableIndex is the return type.
334  const unsigned NumTypes;
335  // Function attribute __attribute__((pure))
336  const bool IsPure;
337  // Function attribute __attribute__((const))
338  const bool IsConst;
339  // Function attribute __attribute__((convergent))
340  const bool IsConv;
341  // OpenCL extension(s) required for this overload.
342  const unsigned short Extension;
343  // First OpenCL version in which this overload was introduced (e.g. CL20).
344  const unsigned short MinVersion;
345  // First OpenCL version in which this overload was removed (e.g. CL20).
346  const unsigned short MaxVersion;
347};
348
349)";
350}
351
352// Verify that the combination of GenTypes in a signature is supported.
353// To simplify the logic for creating overloads in SemaLookup, only allow
354// a signature to contain different GenTypes if these GenTypes represent
355// the same number of actual scalar or vector types.
356//
357// Exit with a fatal error if an unsupported construct is encountered.
358static void VerifySignature(const std::vector<Record *> &Signature,
359                            const Record *BuiltinRec) {
360  unsigned GenTypeVecSizes = 1;
361  unsigned GenTypeTypes = 1;
362
363  for (const auto *T : Signature) {
364    // Check all GenericType arguments in this signature.
365    if (T->isSubClassOf("GenericType")) {
366      // Check number of vector sizes.
367      unsigned NVecSizes =
368          T->getValueAsDef("VectorList")->getValueAsListOfInts("List").size();
369      if (NVecSizes != GenTypeVecSizes && NVecSizes != 1) {
370        if (GenTypeVecSizes > 1) {
371          // We already saw a gentype with a different number of vector sizes.
372          PrintFatalError(BuiltinRec->getLoc(),
373              "number of vector sizes should be equal or 1 for all gentypes "
374              "in a declaration");
375        }
376        GenTypeVecSizes = NVecSizes;
377      }
378
379      // Check number of data types.
380      unsigned NTypes =
381          T->getValueAsDef("TypeList")->getValueAsListOfDefs("List").size();
382      if (NTypes != GenTypeTypes && NTypes != 1) {
383        if (GenTypeTypes > 1) {
384          // We already saw a gentype with a different number of types.
385          PrintFatalError(BuiltinRec->getLoc(),
386              "number of types should be equal or 1 for all gentypes "
387              "in a declaration");
388        }
389        GenTypeTypes = NTypes;
390      }
391    }
392  }
393}
394
395void BuiltinNameEmitter::GetOverloads() {
396  // Populate the TypeMap.
397  std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
398  unsigned I = 0;
399  for (const auto &T : Types) {
400    TypeMap.insert(std::make_pair(T, I++));
401  }
402
403  // Populate the SignaturesList and the FctOverloadMap.
404  unsigned CumulativeSignIndex = 0;
405  std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin");
406  for (const auto *B : Builtins) {
407    StringRef BName = B->getValueAsString("Name");
408    if (FctOverloadMap.find(BName) == FctOverloadMap.end()) {
409      FctOverloadMap.insert(std::make_pair(
410          BName, std::vector<std::pair<const Record *, unsigned>>{}));
411    }
412
413    auto Signature = B->getValueAsListOfDefs("Signature");
414    // Reuse signatures to avoid unnecessary duplicates.
415    auto it =
416        std::find_if(SignaturesList.begin(), SignaturesList.end(),
417                     [&](const std::pair<std::vector<Record *>, unsigned> &a) {
418                       return a.first == Signature;
419                     });
420    unsigned SignIndex;
421    if (it == SignaturesList.end()) {
422      VerifySignature(Signature, B);
423      SignaturesList.push_back(std::make_pair(Signature, CumulativeSignIndex));
424      SignIndex = CumulativeSignIndex;
425      CumulativeSignIndex += Signature.size();
426    } else {
427      SignIndex = it->second;
428    }
429    FctOverloadMap[BName].push_back(std::make_pair(B, SignIndex));
430  }
431}
432
433void BuiltinNameEmitter::EmitExtensionTable() {
434  OS << "static const char *FunctionExtensionTable[] = {\n";
435  unsigned Index = 0;
436  std::vector<Record *> FuncExtensions =
437      Records.getAllDerivedDefinitions("FunctionExtension");
438
439  for (const auto &FE : FuncExtensions) {
440    // Emit OpenCL extension table entry.
441    OS << "  // " << Index << ": " << FE->getName() << "\n"
442       << "  \"" << FE->getValueAsString("ExtName") << "\",\n";
443
444    // Record index of this extension.
445    FunctionExtensionIndex[FE->getName()] = Index++;
446  }
447  OS << "};\n\n";
448}
449
450void BuiltinNameEmitter::EmitTypeTable() {
451  OS << "static const OpenCLTypeStruct TypeTable[] = {\n";
452  for (const auto &T : TypeMap) {
453    const char *AccessQual =
454        StringSwitch<const char *>(T.first->getValueAsString("AccessQualifier"))
455            .Case("RO", "OCLAQ_ReadOnly")
456            .Case("WO", "OCLAQ_WriteOnly")
457            .Case("RW", "OCLAQ_ReadWrite")
458            .Default("OCLAQ_None");
459
460    OS << "  // " << T.second << "\n"
461       << "  {OCLT_" << T.first->getValueAsString("Name") << ", "
462       << T.first->getValueAsInt("VecWidth") << ", "
463       << T.first->getValueAsBit("IsPointer") << ", "
464       << T.first->getValueAsBit("IsConst") << ", "
465       << T.first->getValueAsBit("IsVolatile") << ", "
466       << AccessQual << ", "
467       << T.first->getValueAsString("AddrSpace") << "},\n";
468  }
469  OS << "};\n\n";
470}
471
472void BuiltinNameEmitter::EmitSignatureTable() {
473  // Store a type (e.g. int, float, int2, ...). The type is stored as an index
474  // of a struct OpenCLType table. Multiple entries following each other form a
475  // signature.
476  OS << "static const unsigned SignatureTable[] = {\n";
477  for (const auto &P : SignaturesList) {
478    OS << "  // " << P.second << "\n  ";
479    for (const Record *R : P.first) {
480      OS << TypeMap.find(R)->second << ", ";
481    }
482    OS << "\n";
483  }
484  OS << "};\n\n";
485}
486
487void BuiltinNameEmitter::EmitBuiltinTable() {
488  unsigned Index = 0;
489
490  OS << "static const OpenCLBuiltinStruct BuiltinTable[] = {\n";
491  for (const auto &SLM : SignatureListMap) {
492
493    OS << "  // " << (Index + 1) << ": ";
494    for (const auto &Name : SLM.second.Names) {
495      OS << Name << ", ";
496    }
497    OS << "\n";
498
499    for (const auto &Overload : SLM.second.Signatures) {
500      StringRef ExtName = Overload.first->getValueAsDef("Extension")->getName();
501      OS << "  { " << Overload.second << ", "
502         << Overload.first->getValueAsListOfDefs("Signature").size() << ", "
503         << (Overload.first->getValueAsBit("IsPure")) << ", "
504         << (Overload.first->getValueAsBit("IsConst")) << ", "
505         << (Overload.first->getValueAsBit("IsConv")) << ", "
506         << FunctionExtensionIndex[ExtName] << ", "
507         << Overload.first->getValueAsDef("MinVersion")->getValueAsInt("ID")
508         << ", "
509         << Overload.first->getValueAsDef("MaxVersion")->getValueAsInt("ID")
510         << " },\n";
511      Index++;
512    }
513  }
514  OS << "};\n\n";
515}
516
517bool BuiltinNameEmitter::CanReuseSignature(
518    BuiltinIndexListTy *Candidate,
519    std::vector<std::pair<const Record *, unsigned>> &SignatureList) {
520  assert(Candidate->size() == SignatureList.size() &&
521         "signature lists should have the same size");
522
523  auto &CandidateSigs =
524      SignatureListMap.find(Candidate)->second.Signatures;
525  for (unsigned Index = 0; Index < Candidate->size(); Index++) {
526    const Record *Rec = SignatureList[Index].first;
527    const Record *Rec2 = CandidateSigs[Index].first;
528    if (Rec->getValueAsBit("IsPure") == Rec2->getValueAsBit("IsPure") &&
529        Rec->getValueAsBit("IsConst") == Rec2->getValueAsBit("IsConst") &&
530        Rec->getValueAsBit("IsConv") == Rec2->getValueAsBit("IsConv") &&
531        Rec->getValueAsDef("MinVersion")->getValueAsInt("ID") ==
532            Rec2->getValueAsDef("MinVersion")->getValueAsInt("ID") &&
533        Rec->getValueAsDef("MaxVersion")->getValueAsInt("ID") ==
534            Rec2->getValueAsDef("MaxVersion")->getValueAsInt("ID") &&
535        Rec->getValueAsDef("Extension")->getName() ==
536            Rec2->getValueAsDef("Extension")->getName()) {
537      return true;
538    }
539  }
540  return false;
541}
542
543void BuiltinNameEmitter::GroupBySignature() {
544  // List of signatures known to be emitted.
545  std::vector<BuiltinIndexListTy *> KnownSignatures;
546
547  for (auto &Fct : FctOverloadMap) {
548    bool FoundReusableSig = false;
549
550    // Gather all signatures for the current function.
551    auto *CurSignatureList = new BuiltinIndexListTy();
552    for (const auto &Signature : Fct.second) {
553      CurSignatureList->push_back(Signature.second);
554    }
555    // Sort the list to facilitate future comparisons.
556    std::sort(CurSignatureList->begin(), CurSignatureList->end());
557
558    // Check if we have already seen another function with the same list of
559    // signatures.  If so, just add the name of the function.
560    for (auto *Candidate : KnownSignatures) {
561      if (Candidate->size() == CurSignatureList->size() &&
562          *Candidate == *CurSignatureList) {
563        if (CanReuseSignature(Candidate, Fct.second)) {
564          SignatureListMap.find(Candidate)->second.Names.push_back(Fct.first);
565          FoundReusableSig = true;
566        }
567      }
568    }
569
570    if (FoundReusableSig) {
571      delete CurSignatureList;
572    } else {
573      // Add a new entry.
574      SignatureListMap[CurSignatureList] = {
575          SmallVector<StringRef, 4>(1, Fct.first), Fct.second};
576      KnownSignatures.push_back(CurSignatureList);
577    }
578  }
579
580  for (auto *I : KnownSignatures) {
581    delete I;
582  }
583}
584
585void BuiltinNameEmitter::EmitStringMatcher() {
586  std::vector<StringMatcher::StringPair> ValidBuiltins;
587  unsigned CumulativeIndex = 1;
588
589  for (const auto &SLM : SignatureListMap) {
590    const auto &Ovl = SLM.second.Signatures;
591
592    // A single signature list may be used by different builtins.  Return the
593    // same <index, length> pair for each of those builtins.
594    for (const auto &FctName : SLM.second.Names) {
595      std::string RetStmt;
596      raw_string_ostream SS(RetStmt);
597      SS << "return std::make_pair(" << CumulativeIndex << ", " << Ovl.size()
598         << ");";
599      SS.flush();
600      ValidBuiltins.push_back(StringMatcher::StringPair(FctName, RetStmt));
601    }
602    CumulativeIndex += Ovl.size();
603  }
604
605  OS << R"(
606// Find out whether a string matches an existing OpenCL builtin function name.
607// Returns: A pair <0, 0> if no name matches.
608//          A pair <Index, Len> indexing the BuiltinTable if the name is
609//          matching an OpenCL builtin function.
610static std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name) {
611
612)";
613
614  StringMatcher("Name", ValidBuiltins, OS).Emit(0, true);
615
616  OS << "  return std::make_pair(0, 0);\n";
617  OS << "} // isOpenCLBuiltin\n";
618}
619
620void BuiltinNameEmitter::EmitQualTypeFinder() {
621  OS << R"(
622
623// Convert an OpenCLTypeStruct type to a list of QualTypes.
624// Generic types represent multiple types and vector sizes, thus a vector
625// is returned. The conversion is done in two steps:
626// Step 1: A switch statement fills a vector with scalar base types for the
627//         Cartesian product of (vector sizes) x (types) for generic types,
628//         or a single scalar type for non generic types.
629// Step 2: Qualifiers and other type properties such as vector size are
630//         applied.
631static void OCL2Qual(ASTContext &Context, const OpenCLTypeStruct &Ty,
632                     llvm::SmallVectorImpl<QualType> &QT) {
633  // Number of scalar types in the GenType.
634  unsigned GenTypeNumTypes;
635  // Pointer to the list of vector sizes for the GenType.
636  llvm::ArrayRef<unsigned> GenVectorSizes;
637)";
638
639  // Generate list of vector sizes for each generic type.
640  for (const auto *VectList : Records.getAllDerivedDefinitions("IntList")) {
641    OS << "  constexpr unsigned List"
642       << VectList->getValueAsString("Name") << "[] = {";
643    for (const auto V : VectList->getValueAsListOfInts("List")) {
644      OS << V << ", ";
645    }
646    OS << "};\n";
647  }
648
649  // Step 1.
650  // Start of switch statement over all types.
651  OS << "\n  switch (Ty.ID) {\n";
652
653  // Switch cases for image types (Image2d, Image3d, ...)
654  std::vector<Record *> ImageTypes =
655      Records.getAllDerivedDefinitions("ImageType");
656
657  // Map an image type name to its 3 access-qualified types (RO, WO, RW).
658  std::map<StringRef, SmallVector<Record *, 3>> ImageTypesMap;
659  for (auto *IT : ImageTypes) {
660    auto Entry = ImageTypesMap.find(IT->getValueAsString("Name"));
661    if (Entry == ImageTypesMap.end()) {
662      SmallVector<Record *, 3> ImageList;
663      ImageList.push_back(IT);
664      ImageTypesMap.insert(
665          std::make_pair(IT->getValueAsString("Name"), ImageList));
666    } else {
667      Entry->second.push_back(IT);
668    }
669  }
670
671  // Emit the cases for the image types.  For an image type name, there are 3
672  // corresponding QualTypes ("RO", "WO", "RW").  The "AccessQualifier" field
673  // tells which one is needed.  Emit a switch statement that puts the
674  // corresponding QualType into "QT".
675  for (const auto &ITE : ImageTypesMap) {
676    OS << "    case OCLT_" << ITE.first.str() << ":\n"
677       << "      switch (Ty.AccessQualifier) {\n"
678       << "        case OCLAQ_None:\n"
679       << "          llvm_unreachable(\"Image without access qualifier\");\n";
680    for (const auto &Image : ITE.second) {
681      OS << StringSwitch<const char *>(
682                Image->getValueAsString("AccessQualifier"))
683                .Case("RO", "        case OCLAQ_ReadOnly:\n")
684                .Case("WO", "        case OCLAQ_WriteOnly:\n")
685                .Case("RW", "        case OCLAQ_ReadWrite:\n")
686         << "          QT.push_back(Context."
687         << Image->getValueAsDef("QTName")->getValueAsString("Name") << ");\n"
688         << "          break;\n";
689    }
690    OS << "      }\n"
691       << "      break;\n";
692  }
693
694  // Switch cases for generic types.
695  for (const auto *GenType : Records.getAllDerivedDefinitions("GenericType")) {
696    OS << "    case OCLT_" << GenType->getValueAsString("Name") << ":\n";
697    OS << "      QT.append({";
698
699    // Build the Cartesian product of (vector sizes) x (types).  Only insert
700    // the plain scalar types for now; other type information such as vector
701    // size and type qualifiers will be added after the switch statement.
702    for (unsigned I = 0; I < GenType->getValueAsDef("VectorList")
703                                 ->getValueAsListOfInts("List")
704                                 .size();
705         I++) {
706      for (const auto *T :
707           GenType->getValueAsDef("TypeList")->getValueAsListOfDefs("List")) {
708        OS << "Context."
709           << T->getValueAsDef("QTName")->getValueAsString("Name") << ", ";
710      }
711    }
712    OS << "});\n";
713    // GenTypeNumTypes is the number of types in the GenType
714    // (e.g. float/double/half).
715    OS << "      GenTypeNumTypes = "
716       << GenType->getValueAsDef("TypeList")->getValueAsListOfDefs("List")
717              .size()
718       << ";\n";
719    // GenVectorSizes is the list of vector sizes for this GenType.
720    // QT contains GenTypeNumTypes * #GenVectorSizes elements.
721    OS << "      GenVectorSizes = List"
722       << GenType->getValueAsDef("VectorList")->getValueAsString("Name")
723       << ";\n";
724    OS << "      break;\n";
725  }
726
727  // Switch cases for non generic, non image types (int, int4, float, ...).
728  // Only insert the plain scalar type; vector information and type qualifiers
729  // are added in step 2.
730  std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
731  StringMap<bool> TypesSeen;
732
733  for (const auto *T : Types) {
734    // Check this is not an image type
735    if (ImageTypesMap.find(T->getValueAsString("Name")) != ImageTypesMap.end())
736      continue;
737    // Check we have not seen this Type
738    if (TypesSeen.find(T->getValueAsString("Name")) != TypesSeen.end())
739      continue;
740    TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true));
741
742    // Check the Type does not have an "abstract" QualType
743    auto QT = T->getValueAsDef("QTName");
744    if (QT->getValueAsBit("IsAbstract") == 1)
745      continue;
746    // Emit the cases for non generic, non image types.
747    OS << "    case OCLT_" << T->getValueAsString("Name") << ":\n";
748    OS << "      QT.push_back(Context." << QT->getValueAsString("Name")
749       << ");\n";
750    OS << "      break;\n";
751  }
752
753  // End of switch statement.
754  OS << "  } // end of switch (Ty.ID)\n\n";
755
756  // Step 2.
757  // Add ExtVector types if this was a generic type, as the switch statement
758  // above only populated the list with scalar types.  This completes the
759  // construction of the Cartesian product of (vector sizes) x (types).
760  OS << "  // Construct the different vector types for each generic type.\n";
761  OS << "  if (Ty.ID >= " << TypeList.size() << ") {";
762  OS << R"(
763    for (unsigned I = 0; I < QT.size(); I++) {
764      // For scalars, size is 1.
765      if (GenVectorSizes[I / GenTypeNumTypes] != 1) {
766        QT[I] = Context.getExtVectorType(QT[I],
767                          GenVectorSizes[I / GenTypeNumTypes]);
768      }
769    }
770  }
771)";
772
773  // Assign the right attributes to the types (e.g. vector size).
774  OS << R"(
775  // Set vector size for non-generic vector types.
776  if (Ty.VectorWidth > 1) {
777    for (unsigned Index = 0; Index < QT.size(); Index++) {
778      QT[Index] = Context.getExtVectorType(QT[Index], Ty.VectorWidth);
779    }
780  }
781
782  if (Ty.IsVolatile != 0) {
783    for (unsigned Index = 0; Index < QT.size(); Index++) {
784      QT[Index] = Context.getVolatileType(QT[Index]);
785    }
786  }
787
788  if (Ty.IsConst != 0) {
789    for (unsigned Index = 0; Index < QT.size(); Index++) {
790      QT[Index] = Context.getConstType(QT[Index]);
791    }
792  }
793
794  // Transform the type to a pointer as the last step, if necessary.
795  // Builtin functions only have pointers on [const|volatile], no
796  // [const|volatile] pointers, so this is ok to do it as a last step.
797  if (Ty.IsPointer != 0) {
798    for (unsigned Index = 0; Index < QT.size(); Index++) {
799      QT[Index] = Context.getAddrSpaceQualType(QT[Index], Ty.AS);
800      QT[Index] = Context.getPointerType(QT[Index]);
801    }
802  }
803)";
804
805  // End of the "OCL2Qual" function.
806  OS << "\n} // OCL2Qual\n";
807}
808
809void clang::EmitClangOpenCLBuiltins(RecordKeeper &Records, raw_ostream &OS) {
810  BuiltinNameEmitter NameChecker(Records, OS);
811  NameChecker.Emit();
812}
813