Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
freebsd
GitHub Repository: freebsd/freebsd-src
Path: blob/main/contrib/llvm-project/clang/utils/TableGen/ClangOpenCLBuiltinEmitter.cpp
35230 views
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
// These backends consume the definitions of OpenCL builtin functions in
12
// clang/lib/Sema/OpenCLBuiltins.td and produce builtin handling code for
13
// inclusion in SemaLookup.cpp, or a test file that calls all declared builtins.
14
//
15
//===----------------------------------------------------------------------===//
16
17
#include "TableGenBackends.h"
18
#include "llvm/ADT/MapVector.h"
19
#include "llvm/ADT/STLExtras.h"
20
#include "llvm/ADT/SmallSet.h"
21
#include "llvm/ADT/SmallString.h"
22
#include "llvm/ADT/StringExtras.h"
23
#include "llvm/ADT/StringMap.h"
24
#include "llvm/ADT/StringRef.h"
25
#include "llvm/ADT/StringSwitch.h"
26
#include "llvm/Support/ErrorHandling.h"
27
#include "llvm/Support/raw_ostream.h"
28
#include "llvm/TableGen/Error.h"
29
#include "llvm/TableGen/Record.h"
30
#include "llvm/TableGen/StringMatcher.h"
31
#include "llvm/TableGen/TableGenBackend.h"
32
33
using namespace llvm;
34
35
namespace {
36
37
// A list of signatures that are shared by one or more builtin functions.
38
struct BuiltinTableEntries {
39
SmallVector<StringRef, 4> Names;
40
std::vector<std::pair<const Record *, unsigned>> Signatures;
41
};
42
43
// This tablegen backend emits code for checking whether a function is an
44
// OpenCL builtin function. If so, all overloads of this function are
45
// added to the LookupResult. The generated include file is used by
46
// SemaLookup.cpp
47
//
48
// For a successful lookup of e.g. the "cos" builtin, isOpenCLBuiltin("cos")
49
// returns a pair <Index, Len>.
50
// BuiltinTable[Index] to BuiltinTable[Index + Len] contains the pairs
51
// <SigIndex, SigLen> of the overloads of "cos".
52
// SignatureTable[SigIndex] to SignatureTable[SigIndex + SigLen] contains
53
// one of the signatures of "cos". The SignatureTable entry can be
54
// referenced by other functions, e.g. "sin", to exploit the fact that
55
// many OpenCL builtins share the same signature.
56
//
57
// The file generated by this TableGen emitter contains the following:
58
//
59
// * Structs and enums to represent types and function signatures.
60
//
61
// * const char *FunctionExtensionTable[]
62
// List of space-separated OpenCL extensions. A builtin references an
63
// entry in this table when the builtin requires a particular (set of)
64
// extension(s) to be enabled.
65
//
66
// * OpenCLTypeStruct TypeTable[]
67
// Type information for return types and arguments.
68
//
69
// * unsigned SignatureTable[]
70
// A list of types representing function signatures. Each entry is an index
71
// into the above TypeTable. Multiple entries following each other form a
72
// signature, where the first entry is the return type and subsequent
73
// entries are the argument types.
74
//
75
// * OpenCLBuiltinStruct BuiltinTable[]
76
// Each entry represents one overload of an OpenCL builtin function and
77
// consists of an index into the SignatureTable and the number of arguments.
78
//
79
// * std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name)
80
// Find out whether a string matches an existing OpenCL builtin function
81
// name and return an index into BuiltinTable and the number of overloads.
82
//
83
// * void OCL2Qual(Sema&, OpenCLTypeStruct, std::vector<QualType>&)
84
// Convert an OpenCLTypeStruct type to a list of QualType instances.
85
// One OpenCLTypeStruct can represent multiple types, primarily when using
86
// GenTypes.
87
//
88
class BuiltinNameEmitter {
89
public:
90
BuiltinNameEmitter(RecordKeeper &Records, raw_ostream &OS)
91
: Records(Records), OS(OS) {}
92
93
// Entrypoint to generate the functions and structures for checking
94
// whether a function is an OpenCL builtin function.
95
void Emit();
96
97
private:
98
// A list of indices into the builtin function table.
99
using BuiltinIndexListTy = SmallVector<unsigned, 11>;
100
101
// Contains OpenCL builtin functions and related information, stored as
102
// Record instances. They are coming from the associated TableGen file.
103
RecordKeeper &Records;
104
105
// The output file.
106
raw_ostream &OS;
107
108
// Helper function for BuiltinNameEmitter::EmitDeclarations. Generate enum
109
// definitions in the Output string parameter, and save their Record instances
110
// in the List parameter.
111
// \param Types (in) List containing the Types to extract.
112
// \param TypesSeen (inout) List containing the Types already extracted.
113
// \param Output (out) String containing the enums to emit in the output file.
114
// \param List (out) List containing the extracted Types, except the Types in
115
// TypesSeen.
116
void ExtractEnumTypes(std::vector<Record *> &Types,
117
StringMap<bool> &TypesSeen, std::string &Output,
118
std::vector<const Record *> &List);
119
120
// Emit the enum or struct used in the generated file.
121
// Populate the TypeList at the same time.
122
void EmitDeclarations();
123
124
// Parse the Records generated by TableGen to populate the SignaturesList,
125
// FctOverloadMap and TypeMap.
126
void GetOverloads();
127
128
// Compare two lists of signatures and check that e.g. the OpenCL version,
129
// function attributes, and extension are equal for each signature.
130
// \param Candidate (in) Entry in the SignatureListMap to check.
131
// \param SignatureList (in) List of signatures of the considered function.
132
// \returns true if the two lists of signatures are identical.
133
bool CanReuseSignature(
134
BuiltinIndexListTy *Candidate,
135
std::vector<std::pair<const Record *, unsigned>> &SignatureList);
136
137
// Group functions with the same list of signatures by populating the
138
// SignatureListMap.
139
// Some builtin functions have the same list of signatures, for example the
140
// "sin" and "cos" functions. To save space in the BuiltinTable, the
141
// "isOpenCLBuiltin" function will have the same output for these two
142
// function names.
143
void GroupBySignature();
144
145
// Emit the FunctionExtensionTable that lists all function extensions.
146
void EmitExtensionTable();
147
148
// Emit the TypeTable containing all types used by OpenCL builtins.
149
void EmitTypeTable();
150
151
// Emit the SignatureTable. This table contains all the possible signatures.
152
// A signature is stored as a list of indexes of the TypeTable.
153
// The first index references the return type (mandatory), and the followings
154
// reference its arguments.
155
// E.g.:
156
// 15, 2, 15 can represent a function with the signature:
157
// int func(float, int)
158
// The "int" type being at the index 15 in the TypeTable.
159
void EmitSignatureTable();
160
161
// Emit the BuiltinTable table. This table contains all the overloads of
162
// each function, and is a struct OpenCLBuiltinDecl.
163
// E.g.:
164
// // 891 convert_float2_rtn
165
// { 58, 2, 3, 100, 0 },
166
// This means that the signature of this convert_float2_rtn overload has
167
// 1 argument (+1 for the return type), stored at index 58 in
168
// the SignatureTable. This prototype requires extension "3" in the
169
// FunctionExtensionTable. The last two values represent the minimum (1.0)
170
// and maximum (0, meaning no max version) OpenCL version in which this
171
// overload is supported.
172
void EmitBuiltinTable();
173
174
// Emit a StringMatcher function to check whether a function name is an
175
// OpenCL builtin function name.
176
void EmitStringMatcher();
177
178
// Emit a function returning the clang QualType instance associated with
179
// the TableGen Record Type.
180
void EmitQualTypeFinder();
181
182
// Contains a list of the available signatures, without the name of the
183
// function. Each pair consists of a signature and a cumulative index.
184
// E.g.: <<float, float>, 0>,
185
// <<float, int, int, 2>>,
186
// <<float>, 5>,
187
// ...
188
// <<double, double>, 35>.
189
std::vector<std::pair<std::vector<Record *>, unsigned>> SignaturesList;
190
191
// Map the name of a builtin function to its prototypes (instances of the
192
// TableGen "Builtin" class).
193
// Each prototype is registered as a pair of:
194
// <pointer to the "Builtin" instance,
195
// cumulative index of the associated signature in the SignaturesList>
196
// E.g.: The function cos: (float cos(float), double cos(double), ...)
197
// <"cos", <<ptrToPrototype0, 5>,
198
// <ptrToPrototype1, 35>,
199
// <ptrToPrototype2, 79>>
200
// ptrToPrototype1 has the following signature: <double, double>
201
MapVector<StringRef, std::vector<std::pair<const Record *, unsigned>>>
202
FctOverloadMap;
203
204
// Contains the map of OpenCL types to their index in the TypeTable.
205
MapVector<const Record *, unsigned> TypeMap;
206
207
// List of OpenCL function extensions mapping extension strings to
208
// an index into the FunctionExtensionTable.
209
StringMap<unsigned> FunctionExtensionIndex;
210
211
// List of OpenCL type names in the same order as in enum OpenCLTypeID.
212
// This list does not contain generic types.
213
std::vector<const Record *> TypeList;
214
215
// Same as TypeList, but for generic types only.
216
std::vector<const Record *> GenTypeList;
217
218
// Map an ordered vector of signatures to their original Record instances,
219
// and to a list of function names that share these signatures.
220
//
221
// For example, suppose the "cos" and "sin" functions have only three
222
// signatures, and these signatures are at index Ix in the SignatureTable:
223
// cos | sin | Signature | Index
224
// float cos(float) | float sin(float) | Signature1 | I1
225
// double cos(double) | double sin(double) | Signature2 | I2
226
// half cos(half) | half sin(half) | Signature3 | I3
227
//
228
// Then we will create a mapping of the vector of signatures:
229
// SignatureListMap[<I1, I2, I3>] = <
230
// <"cos", "sin">,
231
// <Signature1, Signature2, Signature3>>
232
// The function "tan", having the same signatures, would be mapped to the
233
// same entry (<I1, I2, I3>).
234
MapVector<BuiltinIndexListTy *, BuiltinTableEntries> SignatureListMap;
235
};
236
237
/// Base class for emitting a file (e.g. header or test) from OpenCLBuiltins.td
238
class OpenCLBuiltinFileEmitterBase {
239
public:
240
OpenCLBuiltinFileEmitterBase(RecordKeeper &Records, raw_ostream &OS)
241
: Records(Records), OS(OS) {}
242
virtual ~OpenCLBuiltinFileEmitterBase() = default;
243
244
// Entrypoint to generate the functions for testing all OpenCL builtin
245
// functions.
246
virtual void emit() = 0;
247
248
protected:
249
struct TypeFlags {
250
TypeFlags() : IsConst(false), IsVolatile(false), IsPointer(false) {}
251
bool IsConst : 1;
252
bool IsVolatile : 1;
253
bool IsPointer : 1;
254
StringRef AddrSpace;
255
};
256
257
// Return a string representation of the given type, such that it can be
258
// used as a type in OpenCL C code.
259
std::string getTypeString(const Record *Type, TypeFlags Flags,
260
int VectorSize) const;
261
262
// Return the type(s) and vector size(s) for the given type. For
263
// non-GenericTypes, the resulting vectors will contain 1 element. For
264
// GenericTypes, the resulting vectors typically contain multiple elements.
265
void getTypeLists(Record *Type, TypeFlags &Flags,
266
std::vector<Record *> &TypeList,
267
std::vector<int64_t> &VectorList) const;
268
269
// Expand the TableGen Records representing a builtin function signature into
270
// one or more function signatures. Return them as a vector of a vector of
271
// strings, with each string containing an OpenCL C type and optional
272
// qualifiers.
273
//
274
// The Records may contain GenericTypes, which expand into multiple
275
// signatures. Repeated occurrences of GenericType in a signature expand to
276
// the same types. For example [char, FGenType, FGenType] expands to:
277
// [char, float, float]
278
// [char, float2, float2]
279
// [char, float3, float3]
280
// ...
281
void
282
expandTypesInSignature(const std::vector<Record *> &Signature,
283
SmallVectorImpl<SmallVector<std::string, 2>> &Types);
284
285
// Emit extension enabling pragmas.
286
void emitExtensionSetup();
287
288
// Emit an #if guard for a Builtin's extension. Return the corresponding
289
// closing #endif, or an empty string if no extension #if guard was emitted.
290
std::string emitExtensionGuard(const Record *Builtin);
291
292
// Emit an #if guard for a Builtin's language version. Return the
293
// corresponding closing #endif, or an empty string if no version #if guard
294
// was emitted.
295
std::string emitVersionGuard(const Record *Builtin);
296
297
// Emit an #if guard for all type extensions required for the given type
298
// strings. Return the corresponding closing #endif, or an empty string
299
// if no extension #if guard was emitted.
300
StringRef
301
emitTypeExtensionGuards(const SmallVectorImpl<std::string> &Signature);
302
303
// Map type strings to type extensions (e.g. "half2" -> "cl_khr_fp16").
304
StringMap<StringRef> TypeExtMap;
305
306
// Contains OpenCL builtin functions and related information, stored as
307
// Record instances. They are coming from the associated TableGen file.
308
RecordKeeper &Records;
309
310
// The output file.
311
raw_ostream &OS;
312
};
313
314
// OpenCL builtin test generator. This class processes the same TableGen input
315
// as BuiltinNameEmitter, but generates a .cl file that contains a call to each
316
// builtin function described in the .td input.
317
class OpenCLBuiltinTestEmitter : public OpenCLBuiltinFileEmitterBase {
318
public:
319
OpenCLBuiltinTestEmitter(RecordKeeper &Records, raw_ostream &OS)
320
: OpenCLBuiltinFileEmitterBase(Records, OS) {}
321
322
// Entrypoint to generate the functions for testing all OpenCL builtin
323
// functions.
324
void emit() override;
325
};
326
327
// OpenCL builtin header generator. This class processes the same TableGen
328
// input as BuiltinNameEmitter, but generates a .h file that contains a
329
// prototype for each builtin function described in the .td input.
330
class OpenCLBuiltinHeaderEmitter : public OpenCLBuiltinFileEmitterBase {
331
public:
332
OpenCLBuiltinHeaderEmitter(RecordKeeper &Records, raw_ostream &OS)
333
: OpenCLBuiltinFileEmitterBase(Records, OS) {}
334
335
// Entrypoint to generate the header.
336
void emit() override;
337
};
338
339
} // namespace
340
341
void BuiltinNameEmitter::Emit() {
342
emitSourceFileHeader("OpenCL Builtin handling", OS, Records);
343
344
OS << "#include \"llvm/ADT/StringRef.h\"\n";
345
OS << "using namespace clang;\n\n";
346
347
// Emit enums and structs.
348
EmitDeclarations();
349
350
// Parse the Records to populate the internal lists.
351
GetOverloads();
352
GroupBySignature();
353
354
// Emit tables.
355
EmitExtensionTable();
356
EmitTypeTable();
357
EmitSignatureTable();
358
EmitBuiltinTable();
359
360
// Emit functions.
361
EmitStringMatcher();
362
EmitQualTypeFinder();
363
}
364
365
void BuiltinNameEmitter::ExtractEnumTypes(std::vector<Record *> &Types,
366
StringMap<bool> &TypesSeen,
367
std::string &Output,
368
std::vector<const Record *> &List) {
369
raw_string_ostream SS(Output);
370
371
for (const auto *T : Types) {
372
if (!TypesSeen.contains(T->getValueAsString("Name"))) {
373
SS << " OCLT_" + T->getValueAsString("Name") << ",\n";
374
// Save the type names in the same order as their enum value. Note that
375
// the Record can be a VectorType or something else, only the name is
376
// important.
377
List.push_back(T);
378
TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true));
379
}
380
}
381
SS.flush();
382
}
383
384
void BuiltinNameEmitter::EmitDeclarations() {
385
// Enum of scalar type names (float, int, ...) and generic type sets.
386
OS << "enum OpenCLTypeID {\n";
387
388
StringMap<bool> TypesSeen;
389
std::string GenTypeEnums;
390
std::string TypeEnums;
391
392
// Extract generic types and non-generic types separately, to keep
393
// gentypes at the end of the enum which simplifies the special handling
394
// for gentypes in SemaLookup.
395
std::vector<Record *> GenTypes =
396
Records.getAllDerivedDefinitions("GenericType");
397
ExtractEnumTypes(GenTypes, TypesSeen, GenTypeEnums, GenTypeList);
398
399
std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
400
ExtractEnumTypes(Types, TypesSeen, TypeEnums, TypeList);
401
402
OS << TypeEnums;
403
OS << GenTypeEnums;
404
OS << "};\n";
405
406
// Structure definitions.
407
OS << R"(
408
// Image access qualifier.
409
enum OpenCLAccessQual : unsigned char {
410
OCLAQ_None,
411
OCLAQ_ReadOnly,
412
OCLAQ_WriteOnly,
413
OCLAQ_ReadWrite
414
};
415
416
// Represents a return type or argument type.
417
struct OpenCLTypeStruct {
418
// A type (e.g. float, int, ...).
419
const OpenCLTypeID ID;
420
// Vector size (if applicable; 0 for scalars and generic types).
421
const unsigned VectorWidth;
422
// 0 if the type is not a pointer.
423
const bool IsPointer : 1;
424
// 0 if the type is not const.
425
const bool IsConst : 1;
426
// 0 if the type is not volatile.
427
const bool IsVolatile : 1;
428
// Access qualifier.
429
const OpenCLAccessQual AccessQualifier;
430
// Address space of the pointer (if applicable).
431
const LangAS AS;
432
};
433
434
// One overload of an OpenCL builtin function.
435
struct OpenCLBuiltinStruct {
436
// Index of the signature in the OpenCLTypeStruct table.
437
const unsigned SigTableIndex;
438
// Entries between index SigTableIndex and (SigTableIndex + NumTypes - 1) in
439
// the SignatureTable represent the complete signature. The first type at
440
// index SigTableIndex is the return type.
441
const unsigned NumTypes;
442
// Function attribute __attribute__((pure))
443
const bool IsPure : 1;
444
// Function attribute __attribute__((const))
445
const bool IsConst : 1;
446
// Function attribute __attribute__((convergent))
447
const bool IsConv : 1;
448
// OpenCL extension(s) required for this overload.
449
const unsigned short Extension;
450
// OpenCL versions in which this overload is available.
451
const unsigned short Versions;
452
};
453
454
)";
455
}
456
457
// Verify that the combination of GenTypes in a signature is supported.
458
// To simplify the logic for creating overloads in SemaLookup, only allow
459
// a signature to contain different GenTypes if these GenTypes represent
460
// the same number of actual scalar or vector types.
461
//
462
// Exit with a fatal error if an unsupported construct is encountered.
463
static void VerifySignature(const std::vector<Record *> &Signature,
464
const Record *BuiltinRec) {
465
unsigned GenTypeVecSizes = 1;
466
unsigned GenTypeTypes = 1;
467
468
for (const auto *T : Signature) {
469
// Check all GenericType arguments in this signature.
470
if (T->isSubClassOf("GenericType")) {
471
// Check number of vector sizes.
472
unsigned NVecSizes =
473
T->getValueAsDef("VectorList")->getValueAsListOfInts("List").size();
474
if (NVecSizes != GenTypeVecSizes && NVecSizes != 1) {
475
if (GenTypeVecSizes > 1) {
476
// We already saw a gentype with a different number of vector sizes.
477
PrintFatalError(BuiltinRec->getLoc(),
478
"number of vector sizes should be equal or 1 for all gentypes "
479
"in a declaration");
480
}
481
GenTypeVecSizes = NVecSizes;
482
}
483
484
// Check number of data types.
485
unsigned NTypes =
486
T->getValueAsDef("TypeList")->getValueAsListOfDefs("List").size();
487
if (NTypes != GenTypeTypes && NTypes != 1) {
488
if (GenTypeTypes > 1) {
489
// We already saw a gentype with a different number of types.
490
PrintFatalError(BuiltinRec->getLoc(),
491
"number of types should be equal or 1 for all gentypes "
492
"in a declaration");
493
}
494
GenTypeTypes = NTypes;
495
}
496
}
497
}
498
}
499
500
void BuiltinNameEmitter::GetOverloads() {
501
// Populate the TypeMap.
502
std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
503
unsigned I = 0;
504
for (const auto &T : Types) {
505
TypeMap.insert(std::make_pair(T, I++));
506
}
507
508
// Populate the SignaturesList and the FctOverloadMap.
509
unsigned CumulativeSignIndex = 0;
510
std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin");
511
for (const auto *B : Builtins) {
512
StringRef BName = B->getValueAsString("Name");
513
if (!FctOverloadMap.contains(BName)) {
514
FctOverloadMap.insert(std::make_pair(
515
BName, std::vector<std::pair<const Record *, unsigned>>{}));
516
}
517
518
auto Signature = B->getValueAsListOfDefs("Signature");
519
// Reuse signatures to avoid unnecessary duplicates.
520
auto it =
521
llvm::find_if(SignaturesList,
522
[&](const std::pair<std::vector<Record *>, unsigned> &a) {
523
return a.first == Signature;
524
});
525
unsigned SignIndex;
526
if (it == SignaturesList.end()) {
527
VerifySignature(Signature, B);
528
SignaturesList.push_back(std::make_pair(Signature, CumulativeSignIndex));
529
SignIndex = CumulativeSignIndex;
530
CumulativeSignIndex += Signature.size();
531
} else {
532
SignIndex = it->second;
533
}
534
FctOverloadMap[BName].push_back(std::make_pair(B, SignIndex));
535
}
536
}
537
538
void BuiltinNameEmitter::EmitExtensionTable() {
539
OS << "static const char *FunctionExtensionTable[] = {\n";
540
unsigned Index = 0;
541
std::vector<Record *> FuncExtensions =
542
Records.getAllDerivedDefinitions("FunctionExtension");
543
544
for (const auto &FE : FuncExtensions) {
545
// Emit OpenCL extension table entry.
546
OS << " // " << Index << ": " << FE->getName() << "\n"
547
<< " \"" << FE->getValueAsString("ExtName") << "\",\n";
548
549
// Record index of this extension.
550
FunctionExtensionIndex[FE->getName()] = Index++;
551
}
552
OS << "};\n\n";
553
}
554
555
void BuiltinNameEmitter::EmitTypeTable() {
556
OS << "static const OpenCLTypeStruct TypeTable[] = {\n";
557
for (const auto &T : TypeMap) {
558
const char *AccessQual =
559
StringSwitch<const char *>(T.first->getValueAsString("AccessQualifier"))
560
.Case("RO", "OCLAQ_ReadOnly")
561
.Case("WO", "OCLAQ_WriteOnly")
562
.Case("RW", "OCLAQ_ReadWrite")
563
.Default("OCLAQ_None");
564
565
OS << " // " << T.second << "\n"
566
<< " {OCLT_" << T.first->getValueAsString("Name") << ", "
567
<< T.first->getValueAsInt("VecWidth") << ", "
568
<< T.first->getValueAsBit("IsPointer") << ", "
569
<< T.first->getValueAsBit("IsConst") << ", "
570
<< T.first->getValueAsBit("IsVolatile") << ", "
571
<< AccessQual << ", "
572
<< T.first->getValueAsString("AddrSpace") << "},\n";
573
}
574
OS << "};\n\n";
575
}
576
577
void BuiltinNameEmitter::EmitSignatureTable() {
578
// Store a type (e.g. int, float, int2, ...). The type is stored as an index
579
// of a struct OpenCLType table. Multiple entries following each other form a
580
// signature.
581
OS << "static const unsigned short SignatureTable[] = {\n";
582
for (const auto &P : SignaturesList) {
583
OS << " // " << P.second << "\n ";
584
for (const Record *R : P.first) {
585
unsigned Entry = TypeMap.find(R)->second;
586
if (Entry > USHRT_MAX) {
587
// Report an error when seeing an entry that is too large for the
588
// current index type (unsigned short). When hitting this, the type
589
// of SignatureTable will need to be changed.
590
PrintFatalError("Entry in SignatureTable exceeds limit.");
591
}
592
OS << Entry << ", ";
593
}
594
OS << "\n";
595
}
596
OS << "};\n\n";
597
}
598
599
// Encode a range MinVersion..MaxVersion into a single bit mask that can be
600
// checked against LangOpts using isOpenCLVersionContainedInMask().
601
// This must be kept in sync with OpenCLVersionID in OpenCLOptions.h.
602
// (Including OpenCLOptions.h here would be a layering violation.)
603
static unsigned short EncodeVersions(unsigned int MinVersion,
604
unsigned int MaxVersion) {
605
unsigned short Encoded = 0;
606
607
// A maximum version of 0 means available in all later versions.
608
if (MaxVersion == 0) {
609
MaxVersion = UINT_MAX;
610
}
611
612
unsigned VersionIDs[] = {100, 110, 120, 200, 300};
613
for (unsigned I = 0; I < std::size(VersionIDs); I++) {
614
if (VersionIDs[I] >= MinVersion && VersionIDs[I] < MaxVersion) {
615
Encoded |= 1 << I;
616
}
617
}
618
619
return Encoded;
620
}
621
622
void BuiltinNameEmitter::EmitBuiltinTable() {
623
unsigned Index = 0;
624
625
OS << "static const OpenCLBuiltinStruct BuiltinTable[] = {\n";
626
for (const auto &SLM : SignatureListMap) {
627
628
OS << " // " << (Index + 1) << ": ";
629
for (const auto &Name : SLM.second.Names) {
630
OS << Name << ", ";
631
}
632
OS << "\n";
633
634
for (const auto &Overload : SLM.second.Signatures) {
635
StringRef ExtName = Overload.first->getValueAsDef("Extension")->getName();
636
unsigned int MinVersion =
637
Overload.first->getValueAsDef("MinVersion")->getValueAsInt("ID");
638
unsigned int MaxVersion =
639
Overload.first->getValueAsDef("MaxVersion")->getValueAsInt("ID");
640
641
OS << " { " << Overload.second << ", "
642
<< Overload.first->getValueAsListOfDefs("Signature").size() << ", "
643
<< (Overload.first->getValueAsBit("IsPure")) << ", "
644
<< (Overload.first->getValueAsBit("IsConst")) << ", "
645
<< (Overload.first->getValueAsBit("IsConv")) << ", "
646
<< FunctionExtensionIndex[ExtName] << ", "
647
<< EncodeVersions(MinVersion, MaxVersion) << " },\n";
648
Index++;
649
}
650
}
651
OS << "};\n\n";
652
}
653
654
bool BuiltinNameEmitter::CanReuseSignature(
655
BuiltinIndexListTy *Candidate,
656
std::vector<std::pair<const Record *, unsigned>> &SignatureList) {
657
assert(Candidate->size() == SignatureList.size() &&
658
"signature lists should have the same size");
659
660
auto &CandidateSigs =
661
SignatureListMap.find(Candidate)->second.Signatures;
662
for (unsigned Index = 0; Index < Candidate->size(); Index++) {
663
const Record *Rec = SignatureList[Index].first;
664
const Record *Rec2 = CandidateSigs[Index].first;
665
if (Rec->getValueAsBit("IsPure") == Rec2->getValueAsBit("IsPure") &&
666
Rec->getValueAsBit("IsConst") == Rec2->getValueAsBit("IsConst") &&
667
Rec->getValueAsBit("IsConv") == Rec2->getValueAsBit("IsConv") &&
668
Rec->getValueAsDef("MinVersion")->getValueAsInt("ID") ==
669
Rec2->getValueAsDef("MinVersion")->getValueAsInt("ID") &&
670
Rec->getValueAsDef("MaxVersion")->getValueAsInt("ID") ==
671
Rec2->getValueAsDef("MaxVersion")->getValueAsInt("ID") &&
672
Rec->getValueAsDef("Extension")->getName() ==
673
Rec2->getValueAsDef("Extension")->getName()) {
674
return true;
675
}
676
}
677
return false;
678
}
679
680
void BuiltinNameEmitter::GroupBySignature() {
681
// List of signatures known to be emitted.
682
std::vector<BuiltinIndexListTy *> KnownSignatures;
683
684
for (auto &Fct : FctOverloadMap) {
685
bool FoundReusableSig = false;
686
687
// Gather all signatures for the current function.
688
auto *CurSignatureList = new BuiltinIndexListTy();
689
for (const auto &Signature : Fct.second) {
690
CurSignatureList->push_back(Signature.second);
691
}
692
// Sort the list to facilitate future comparisons.
693
llvm::sort(*CurSignatureList);
694
695
// Check if we have already seen another function with the same list of
696
// signatures. If so, just add the name of the function.
697
for (auto *Candidate : KnownSignatures) {
698
if (Candidate->size() == CurSignatureList->size() &&
699
*Candidate == *CurSignatureList) {
700
if (CanReuseSignature(Candidate, Fct.second)) {
701
SignatureListMap.find(Candidate)->second.Names.push_back(Fct.first);
702
FoundReusableSig = true;
703
}
704
}
705
}
706
707
if (FoundReusableSig) {
708
delete CurSignatureList;
709
} else {
710
// Add a new entry.
711
SignatureListMap[CurSignatureList] = {
712
SmallVector<StringRef, 4>(1, Fct.first), Fct.second};
713
KnownSignatures.push_back(CurSignatureList);
714
}
715
}
716
717
for (auto *I : KnownSignatures) {
718
delete I;
719
}
720
}
721
722
void BuiltinNameEmitter::EmitStringMatcher() {
723
std::vector<StringMatcher::StringPair> ValidBuiltins;
724
unsigned CumulativeIndex = 1;
725
726
for (const auto &SLM : SignatureListMap) {
727
const auto &Ovl = SLM.second.Signatures;
728
729
// A single signature list may be used by different builtins. Return the
730
// same <index, length> pair for each of those builtins.
731
for (const auto &FctName : SLM.second.Names) {
732
std::string RetStmt;
733
raw_string_ostream SS(RetStmt);
734
SS << "return std::make_pair(" << CumulativeIndex << ", " << Ovl.size()
735
<< ");";
736
SS.flush();
737
ValidBuiltins.push_back(
738
StringMatcher::StringPair(std::string(FctName), RetStmt));
739
}
740
CumulativeIndex += Ovl.size();
741
}
742
743
OS << R"(
744
// Find out whether a string matches an existing OpenCL builtin function name.
745
// Returns: A pair <0, 0> if no name matches.
746
// A pair <Index, Len> indexing the BuiltinTable if the name is
747
// matching an OpenCL builtin function.
748
static std::pair<unsigned, unsigned> isOpenCLBuiltin(llvm::StringRef Name) {
749
750
)";
751
752
StringMatcher("Name", ValidBuiltins, OS).Emit(0, true);
753
754
OS << " return std::make_pair(0, 0);\n";
755
OS << "} // isOpenCLBuiltin\n";
756
}
757
758
// Emit an if-statement with an isMacroDefined call for each extension in
759
// the space-separated list of extensions.
760
static void EmitMacroChecks(raw_ostream &OS, StringRef Extensions) {
761
SmallVector<StringRef, 2> ExtVec;
762
Extensions.split(ExtVec, " ");
763
OS << " if (";
764
for (StringRef Ext : ExtVec) {
765
if (Ext != ExtVec.front())
766
OS << " && ";
767
OS << "S.getPreprocessor().isMacroDefined(\"" << Ext << "\")";
768
}
769
OS << ") {\n ";
770
}
771
772
void BuiltinNameEmitter::EmitQualTypeFinder() {
773
OS << R"(
774
775
static QualType getOpenCLEnumType(Sema &S, llvm::StringRef Name);
776
static QualType getOpenCLTypedefType(Sema &S, llvm::StringRef Name);
777
778
// Convert an OpenCLTypeStruct type to a list of QualTypes.
779
// Generic types represent multiple types and vector sizes, thus a vector
780
// is returned. The conversion is done in two steps:
781
// Step 1: A switch statement fills a vector with scalar base types for the
782
// Cartesian product of (vector sizes) x (types) for generic types,
783
// or a single scalar type for non generic types.
784
// Step 2: Qualifiers and other type properties such as vector size are
785
// applied.
786
static void OCL2Qual(Sema &S, const OpenCLTypeStruct &Ty,
787
llvm::SmallVectorImpl<QualType> &QT) {
788
ASTContext &Context = S.Context;
789
// Number of scalar types in the GenType.
790
unsigned GenTypeNumTypes;
791
// Pointer to the list of vector sizes for the GenType.
792
llvm::ArrayRef<unsigned> GenVectorSizes;
793
)";
794
795
// Generate list of vector sizes for each generic type.
796
for (const auto *VectList : Records.getAllDerivedDefinitions("IntList")) {
797
OS << " constexpr unsigned List"
798
<< VectList->getValueAsString("Name") << "[] = {";
799
for (const auto V : VectList->getValueAsListOfInts("List")) {
800
OS << V << ", ";
801
}
802
OS << "};\n";
803
}
804
805
// Step 1.
806
// Start of switch statement over all types.
807
OS << "\n switch (Ty.ID) {\n";
808
809
// Switch cases for image types (Image2d, Image3d, ...)
810
std::vector<Record *> ImageTypes =
811
Records.getAllDerivedDefinitions("ImageType");
812
813
// Map an image type name to its 3 access-qualified types (RO, WO, RW).
814
StringMap<SmallVector<Record *, 3>> ImageTypesMap;
815
for (auto *IT : ImageTypes) {
816
auto Entry = ImageTypesMap.find(IT->getValueAsString("Name"));
817
if (Entry == ImageTypesMap.end()) {
818
SmallVector<Record *, 3> ImageList;
819
ImageList.push_back(IT);
820
ImageTypesMap.insert(
821
std::make_pair(IT->getValueAsString("Name"), ImageList));
822
} else {
823
Entry->second.push_back(IT);
824
}
825
}
826
827
// Emit the cases for the image types. For an image type name, there are 3
828
// corresponding QualTypes ("RO", "WO", "RW"). The "AccessQualifier" field
829
// tells which one is needed. Emit a switch statement that puts the
830
// corresponding QualType into "QT".
831
for (const auto &ITE : ImageTypesMap) {
832
OS << " case OCLT_" << ITE.getKey() << ":\n"
833
<< " switch (Ty.AccessQualifier) {\n"
834
<< " case OCLAQ_None:\n"
835
<< " llvm_unreachable(\"Image without access qualifier\");\n";
836
for (const auto &Image : ITE.getValue()) {
837
StringRef Exts =
838
Image->getValueAsDef("Extension")->getValueAsString("ExtName");
839
OS << StringSwitch<const char *>(
840
Image->getValueAsString("AccessQualifier"))
841
.Case("RO", " case OCLAQ_ReadOnly:\n")
842
.Case("WO", " case OCLAQ_WriteOnly:\n")
843
.Case("RW", " case OCLAQ_ReadWrite:\n");
844
if (!Exts.empty()) {
845
OS << " ";
846
EmitMacroChecks(OS, Exts);
847
}
848
OS << " QT.push_back("
849
<< Image->getValueAsDef("QTExpr")->getValueAsString("TypeExpr")
850
<< ");\n";
851
if (!Exts.empty()) {
852
OS << " }\n";
853
}
854
OS << " break;\n";
855
}
856
OS << " }\n"
857
<< " break;\n";
858
}
859
860
// Switch cases for generic types.
861
for (const auto *GenType : Records.getAllDerivedDefinitions("GenericType")) {
862
OS << " case OCLT_" << GenType->getValueAsString("Name") << ": {\n";
863
864
// Build the Cartesian product of (vector sizes) x (types). Only insert
865
// the plain scalar types for now; other type information such as vector
866
// size and type qualifiers will be added after the switch statement.
867
std::vector<Record *> BaseTypes =
868
GenType->getValueAsDef("TypeList")->getValueAsListOfDefs("List");
869
870
// Collect all QualTypes for a single vector size into TypeList.
871
OS << " SmallVector<QualType, " << BaseTypes.size() << "> TypeList;\n";
872
for (const auto *T : BaseTypes) {
873
StringRef Exts =
874
T->getValueAsDef("Extension")->getValueAsString("ExtName");
875
if (!Exts.empty()) {
876
EmitMacroChecks(OS, Exts);
877
}
878
OS << " TypeList.push_back("
879
<< T->getValueAsDef("QTExpr")->getValueAsString("TypeExpr") << ");\n";
880
if (!Exts.empty()) {
881
OS << " }\n";
882
}
883
}
884
OS << " GenTypeNumTypes = TypeList.size();\n";
885
886
// Duplicate the TypeList for every vector size.
887
std::vector<int64_t> VectorList =
888
GenType->getValueAsDef("VectorList")->getValueAsListOfInts("List");
889
OS << " QT.reserve(" << VectorList.size() * BaseTypes.size() << ");\n"
890
<< " for (unsigned I = 0; I < " << VectorList.size() << "; I++) {\n"
891
<< " QT.append(TypeList);\n"
892
<< " }\n";
893
894
// GenVectorSizes is the list of vector sizes for this GenType.
895
OS << " GenVectorSizes = List"
896
<< GenType->getValueAsDef("VectorList")->getValueAsString("Name")
897
<< ";\n"
898
<< " break;\n"
899
<< " }\n";
900
}
901
902
// Switch cases for non generic, non image types (int, int4, float, ...).
903
// Only insert the plain scalar type; vector information and type qualifiers
904
// are added in step 2.
905
std::vector<Record *> Types = Records.getAllDerivedDefinitions("Type");
906
StringMap<bool> TypesSeen;
907
908
for (const auto *T : Types) {
909
// Check this is not an image type
910
if (ImageTypesMap.contains(T->getValueAsString("Name")))
911
continue;
912
// Check we have not seen this Type
913
if (TypesSeen.contains(T->getValueAsString("Name")))
914
continue;
915
TypesSeen.insert(std::make_pair(T->getValueAsString("Name"), true));
916
917
// Check the Type does not have an "abstract" QualType
918
auto QT = T->getValueAsDef("QTExpr");
919
if (QT->getValueAsBit("IsAbstract") == 1)
920
continue;
921
// Emit the cases for non generic, non image types.
922
OS << " case OCLT_" << T->getValueAsString("Name") << ":\n";
923
924
StringRef Exts = T->getValueAsDef("Extension")->getValueAsString("ExtName");
925
// If this type depends on an extension, ensure the extension macros are
926
// defined.
927
if (!Exts.empty()) {
928
EmitMacroChecks(OS, Exts);
929
}
930
OS << " QT.push_back(" << QT->getValueAsString("TypeExpr") << ");\n";
931
if (!Exts.empty()) {
932
OS << " }\n";
933
}
934
OS << " break;\n";
935
}
936
937
// End of switch statement.
938
OS << " } // end of switch (Ty.ID)\n\n";
939
940
// Step 2.
941
// Add ExtVector types if this was a generic type, as the switch statement
942
// above only populated the list with scalar types. This completes the
943
// construction of the Cartesian product of (vector sizes) x (types).
944
OS << " // Construct the different vector types for each generic type.\n";
945
OS << " if (Ty.ID >= " << TypeList.size() << ") {";
946
OS << R"(
947
for (unsigned I = 0; I < QT.size(); I++) {
948
// For scalars, size is 1.
949
if (GenVectorSizes[I / GenTypeNumTypes] != 1) {
950
QT[I] = Context.getExtVectorType(QT[I],
951
GenVectorSizes[I / GenTypeNumTypes]);
952
}
953
}
954
}
955
)";
956
957
// Assign the right attributes to the types (e.g. vector size).
958
OS << R"(
959
// Set vector size for non-generic vector types.
960
if (Ty.VectorWidth > 1) {
961
for (unsigned Index = 0; Index < QT.size(); Index++) {
962
QT[Index] = Context.getExtVectorType(QT[Index], Ty.VectorWidth);
963
}
964
}
965
966
if (Ty.IsVolatile != 0) {
967
for (unsigned Index = 0; Index < QT.size(); Index++) {
968
QT[Index] = Context.getVolatileType(QT[Index]);
969
}
970
}
971
972
if (Ty.IsConst != 0) {
973
for (unsigned Index = 0; Index < QT.size(); Index++) {
974
QT[Index] = Context.getConstType(QT[Index]);
975
}
976
}
977
978
// Transform the type to a pointer as the last step, if necessary.
979
// Builtin functions only have pointers on [const|volatile], no
980
// [const|volatile] pointers, so this is ok to do it as a last step.
981
if (Ty.IsPointer != 0) {
982
for (unsigned Index = 0; Index < QT.size(); Index++) {
983
QT[Index] = Context.getAddrSpaceQualType(QT[Index], Ty.AS);
984
QT[Index] = Context.getPointerType(QT[Index]);
985
}
986
}
987
)";
988
989
// End of the "OCL2Qual" function.
990
OS << "\n} // OCL2Qual\n";
991
}
992
993
std::string OpenCLBuiltinFileEmitterBase::getTypeString(const Record *Type,
994
TypeFlags Flags,
995
int VectorSize) const {
996
std::string S;
997
if (Type->getValueAsBit("IsConst") || Flags.IsConst) {
998
S += "const ";
999
}
1000
if (Type->getValueAsBit("IsVolatile") || Flags.IsVolatile) {
1001
S += "volatile ";
1002
}
1003
1004
auto PrintAddrSpace = [&S](StringRef AddrSpace) {
1005
S += StringSwitch<const char *>(AddrSpace)
1006
.Case("clang::LangAS::opencl_private", "__private")
1007
.Case("clang::LangAS::opencl_global", "__global")
1008
.Case("clang::LangAS::opencl_constant", "__constant")
1009
.Case("clang::LangAS::opencl_local", "__local")
1010
.Case("clang::LangAS::opencl_generic", "__generic")
1011
.Default("__private");
1012
S += " ";
1013
};
1014
if (Flags.IsPointer) {
1015
PrintAddrSpace(Flags.AddrSpace);
1016
} else if (Type->getValueAsBit("IsPointer")) {
1017
PrintAddrSpace(Type->getValueAsString("AddrSpace"));
1018
}
1019
1020
StringRef Acc = Type->getValueAsString("AccessQualifier");
1021
if (Acc != "") {
1022
S += StringSwitch<const char *>(Acc)
1023
.Case("RO", "__read_only ")
1024
.Case("WO", "__write_only ")
1025
.Case("RW", "__read_write ");
1026
}
1027
1028
S += Type->getValueAsString("Name").str();
1029
if (VectorSize > 1) {
1030
S += std::to_string(VectorSize);
1031
}
1032
1033
if (Type->getValueAsBit("IsPointer") || Flags.IsPointer) {
1034
S += " *";
1035
}
1036
1037
return S;
1038
}
1039
1040
void OpenCLBuiltinFileEmitterBase::getTypeLists(
1041
Record *Type, TypeFlags &Flags, std::vector<Record *> &TypeList,
1042
std::vector<int64_t> &VectorList) const {
1043
bool isGenType = Type->isSubClassOf("GenericType");
1044
if (isGenType) {
1045
TypeList = Type->getValueAsDef("TypeList")->getValueAsListOfDefs("List");
1046
VectorList =
1047
Type->getValueAsDef("VectorList")->getValueAsListOfInts("List");
1048
return;
1049
}
1050
1051
if (Type->isSubClassOf("PointerType") || Type->isSubClassOf("ConstType") ||
1052
Type->isSubClassOf("VolatileType")) {
1053
StringRef SubTypeName = Type->getValueAsString("Name");
1054
Record *PossibleGenType = Records.getDef(SubTypeName);
1055
if (PossibleGenType && PossibleGenType->isSubClassOf("GenericType")) {
1056
// When PointerType, ConstType, or VolatileType is applied to a
1057
// GenericType, the flags need to be taken from the subtype, not from the
1058
// GenericType.
1059
Flags.IsPointer = Type->getValueAsBit("IsPointer");
1060
Flags.IsConst = Type->getValueAsBit("IsConst");
1061
Flags.IsVolatile = Type->getValueAsBit("IsVolatile");
1062
Flags.AddrSpace = Type->getValueAsString("AddrSpace");
1063
getTypeLists(PossibleGenType, Flags, TypeList, VectorList);
1064
return;
1065
}
1066
}
1067
1068
// Not a GenericType, so just insert the single type.
1069
TypeList.push_back(Type);
1070
VectorList.push_back(Type->getValueAsInt("VecWidth"));
1071
}
1072
1073
void OpenCLBuiltinFileEmitterBase::expandTypesInSignature(
1074
const std::vector<Record *> &Signature,
1075
SmallVectorImpl<SmallVector<std::string, 2>> &Types) {
1076
// Find out if there are any GenTypes in this signature, and if so, calculate
1077
// into how many signatures they will expand.
1078
unsigned NumSignatures = 1;
1079
SmallVector<SmallVector<std::string, 4>, 4> ExpandedGenTypes;
1080
for (const auto &Arg : Signature) {
1081
SmallVector<std::string, 4> ExpandedArg;
1082
std::vector<Record *> TypeList;
1083
std::vector<int64_t> VectorList;
1084
TypeFlags Flags;
1085
1086
getTypeLists(Arg, Flags, TypeList, VectorList);
1087
1088
// Insert the Cartesian product of the types and vector sizes.
1089
for (const auto &Vector : VectorList) {
1090
for (const auto &Type : TypeList) {
1091
std::string FullType = getTypeString(Type, Flags, Vector);
1092
ExpandedArg.push_back(FullType);
1093
1094
// If the type requires an extension, add a TypeExtMap entry mapping
1095
// the full type name to the extension.
1096
StringRef Ext =
1097
Type->getValueAsDef("Extension")->getValueAsString("ExtName");
1098
if (!Ext.empty() && !TypeExtMap.contains(FullType)) {
1099
TypeExtMap.insert({FullType, Ext});
1100
}
1101
}
1102
}
1103
NumSignatures = std::max<unsigned>(NumSignatures, ExpandedArg.size());
1104
ExpandedGenTypes.push_back(ExpandedArg);
1105
}
1106
1107
// Now the total number of signatures is known. Populate the return list with
1108
// all signatures.
1109
for (unsigned I = 0; I < NumSignatures; I++) {
1110
SmallVector<std::string, 2> Args;
1111
1112
// Process a single signature.
1113
for (unsigned ArgNum = 0; ArgNum < Signature.size(); ArgNum++) {
1114
// For differently-sized GenTypes in a parameter list, the smaller
1115
// GenTypes just repeat, so index modulo the number of expanded types.
1116
size_t TypeIndex = I % ExpandedGenTypes[ArgNum].size();
1117
Args.push_back(ExpandedGenTypes[ArgNum][TypeIndex]);
1118
}
1119
Types.push_back(Args);
1120
}
1121
}
1122
1123
void OpenCLBuiltinFileEmitterBase::emitExtensionSetup() {
1124
OS << R"(
1125
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
1126
#pragma OPENCL EXTENSION cl_khr_fp64 : enable
1127
#pragma OPENCL EXTENSION cl_khr_int64_base_atomics : enable
1128
#pragma OPENCL EXTENSION cl_khr_int64_extended_atomics : enable
1129
#pragma OPENCL EXTENSION cl_khr_gl_msaa_sharing : enable
1130
#pragma OPENCL EXTENSION cl_khr_mipmap_image_writes : enable
1131
#pragma OPENCL EXTENSION cl_khr_3d_image_writes : enable
1132
1133
)";
1134
}
1135
1136
std::string
1137
OpenCLBuiltinFileEmitterBase::emitExtensionGuard(const Record *Builtin) {
1138
StringRef Extensions =
1139
Builtin->getValueAsDef("Extension")->getValueAsString("ExtName");
1140
if (Extensions.empty())
1141
return "";
1142
1143
OS << "#if";
1144
1145
SmallVector<StringRef, 2> ExtVec;
1146
Extensions.split(ExtVec, " ");
1147
bool isFirst = true;
1148
for (StringRef Ext : ExtVec) {
1149
if (!isFirst) {
1150
OS << " &&";
1151
}
1152
OS << " defined(" << Ext << ")";
1153
isFirst = false;
1154
}
1155
OS << "\n";
1156
1157
return "#endif // Extension\n";
1158
}
1159
1160
std::string
1161
OpenCLBuiltinFileEmitterBase::emitVersionGuard(const Record *Builtin) {
1162
std::string OptionalEndif;
1163
auto PrintOpenCLVersion = [this](int Version) {
1164
OS << "CL_VERSION_" << (Version / 100) << "_" << ((Version % 100) / 10);
1165
};
1166
int MinVersion = Builtin->getValueAsDef("MinVersion")->getValueAsInt("ID");
1167
if (MinVersion != 100) {
1168
// OpenCL 1.0 is the default minimum version.
1169
OS << "#if __OPENCL_C_VERSION__ >= ";
1170
PrintOpenCLVersion(MinVersion);
1171
OS << "\n";
1172
OptionalEndif = "#endif // MinVersion\n" + OptionalEndif;
1173
}
1174
int MaxVersion = Builtin->getValueAsDef("MaxVersion")->getValueAsInt("ID");
1175
if (MaxVersion) {
1176
OS << "#if __OPENCL_C_VERSION__ < ";
1177
PrintOpenCLVersion(MaxVersion);
1178
OS << "\n";
1179
OptionalEndif = "#endif // MaxVersion\n" + OptionalEndif;
1180
}
1181
return OptionalEndif;
1182
}
1183
1184
StringRef OpenCLBuiltinFileEmitterBase::emitTypeExtensionGuards(
1185
const SmallVectorImpl<std::string> &Signature) {
1186
SmallSet<StringRef, 2> ExtSet;
1187
1188
// Iterate over all types to gather the set of required TypeExtensions.
1189
for (const auto &Ty : Signature) {
1190
StringRef TypeExt = TypeExtMap.lookup(Ty);
1191
if (!TypeExt.empty()) {
1192
// The TypeExtensions are space-separated in the .td file.
1193
SmallVector<StringRef, 2> ExtVec;
1194
TypeExt.split(ExtVec, " ");
1195
for (const auto Ext : ExtVec) {
1196
ExtSet.insert(Ext);
1197
}
1198
}
1199
}
1200
1201
// Emit the #if only when at least one extension is required.
1202
if (ExtSet.empty())
1203
return "";
1204
1205
OS << "#if ";
1206
bool isFirst = true;
1207
for (const auto Ext : ExtSet) {
1208
if (!isFirst)
1209
OS << " && ";
1210
OS << "defined(" << Ext << ")";
1211
isFirst = false;
1212
}
1213
OS << "\n";
1214
return "#endif // TypeExtension\n";
1215
}
1216
1217
void OpenCLBuiltinTestEmitter::emit() {
1218
emitSourceFileHeader("OpenCL Builtin exhaustive testing", OS, Records);
1219
1220
emitExtensionSetup();
1221
1222
// Ensure each test has a unique name by numbering them.
1223
unsigned TestID = 0;
1224
1225
// Iterate over all builtins.
1226
std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin");
1227
for (const auto *B : Builtins) {
1228
StringRef Name = B->getValueAsString("Name");
1229
1230
SmallVector<SmallVector<std::string, 2>, 4> FTypes;
1231
expandTypesInSignature(B->getValueAsListOfDefs("Signature"), FTypes);
1232
1233
OS << "// Test " << Name << "\n";
1234
1235
std::string OptionalExtensionEndif = emitExtensionGuard(B);
1236
std::string OptionalVersionEndif = emitVersionGuard(B);
1237
1238
for (const auto &Signature : FTypes) {
1239
StringRef OptionalTypeExtEndif = emitTypeExtensionGuards(Signature);
1240
1241
// Emit function declaration.
1242
OS << Signature[0] << " test" << TestID++ << "_" << Name << "(";
1243
if (Signature.size() > 1) {
1244
for (unsigned I = 1; I < Signature.size(); I++) {
1245
if (I != 1)
1246
OS << ", ";
1247
OS << Signature[I] << " arg" << I;
1248
}
1249
}
1250
OS << ") {\n";
1251
1252
// Emit function body.
1253
OS << " ";
1254
if (Signature[0] != "void") {
1255
OS << "return ";
1256
}
1257
OS << Name << "(";
1258
for (unsigned I = 1; I < Signature.size(); I++) {
1259
if (I != 1)
1260
OS << ", ";
1261
OS << "arg" << I;
1262
}
1263
OS << ");\n";
1264
1265
// End of function body.
1266
OS << "}\n";
1267
OS << OptionalTypeExtEndif;
1268
}
1269
1270
OS << OptionalVersionEndif;
1271
OS << OptionalExtensionEndif;
1272
}
1273
}
1274
1275
void OpenCLBuiltinHeaderEmitter::emit() {
1276
emitSourceFileHeader("OpenCL Builtin declarations", OS, Records);
1277
1278
emitExtensionSetup();
1279
1280
OS << R"(
1281
#define __ovld __attribute__((overloadable))
1282
#define __conv __attribute__((convergent))
1283
#define __purefn __attribute__((pure))
1284
#define __cnfn __attribute__((const))
1285
1286
)";
1287
1288
// Iterate over all builtins; sort to follow order of definition in .td file.
1289
std::vector<Record *> Builtins = Records.getAllDerivedDefinitions("Builtin");
1290
llvm::sort(Builtins, LessRecord());
1291
1292
for (const auto *B : Builtins) {
1293
StringRef Name = B->getValueAsString("Name");
1294
1295
std::string OptionalExtensionEndif = emitExtensionGuard(B);
1296
std::string OptionalVersionEndif = emitVersionGuard(B);
1297
1298
SmallVector<SmallVector<std::string, 2>, 4> FTypes;
1299
expandTypesInSignature(B->getValueAsListOfDefs("Signature"), FTypes);
1300
1301
for (const auto &Signature : FTypes) {
1302
StringRef OptionalTypeExtEndif = emitTypeExtensionGuards(Signature);
1303
1304
// Emit function declaration.
1305
OS << Signature[0] << " __ovld ";
1306
if (B->getValueAsBit("IsConst"))
1307
OS << "__cnfn ";
1308
if (B->getValueAsBit("IsPure"))
1309
OS << "__purefn ";
1310
if (B->getValueAsBit("IsConv"))
1311
OS << "__conv ";
1312
1313
OS << Name << "(";
1314
if (Signature.size() > 1) {
1315
for (unsigned I = 1; I < Signature.size(); I++) {
1316
if (I != 1)
1317
OS << ", ";
1318
OS << Signature[I];
1319
}
1320
}
1321
OS << ");\n";
1322
1323
OS << OptionalTypeExtEndif;
1324
}
1325
1326
OS << OptionalVersionEndif;
1327
OS << OptionalExtensionEndif;
1328
}
1329
1330
OS << "\n// Disable any extensions we may have enabled previously.\n"
1331
"#pragma OPENCL EXTENSION all : disable\n";
1332
}
1333
1334
void clang::EmitClangOpenCLBuiltins(RecordKeeper &Records, raw_ostream &OS) {
1335
BuiltinNameEmitter NameChecker(Records, OS);
1336
NameChecker.Emit();
1337
}
1338
1339
void clang::EmitClangOpenCLBuiltinHeader(RecordKeeper &Records,
1340
raw_ostream &OS) {
1341
OpenCLBuiltinHeaderEmitter HeaderFileGenerator(Records, OS);
1342
HeaderFileGenerator.emit();
1343
}
1344
1345
void clang::EmitClangOpenCLBuiltinTests(RecordKeeper &Records,
1346
raw_ostream &OS) {
1347
OpenCLBuiltinTestEmitter TestFileGenerator(Records, OS);
1348
TestFileGenerator.emit();
1349
}
1350
1351