1#ifndef LLVM_OpenACC_INC
2#define LLVM_OpenACC_INC
3
4#include "llvm/ADT/ArrayRef.h"
5#include "llvm/ADT/BitmaskEnum.h"
6#include "llvm/ADT/Sequence.h"
7#include "llvm/ADT/STLExtras.h"
8#include "llvm/ADT/StringRef.h"
9#include "llvm/Frontend/Directive/Spelling.h"
10#include "llvm/Support/Compiler.h"
11#include <cstddef>
12#include <utility>
13
14namespace llvm {
15
16namespace acc {
17
18LLVM_ENABLE_BITMASK_ENUMS_IN_NAMESPACE();
19
20enum class Association {
21 Block,
22 First_ = Block,
23 Declaration,
24 Delimited,
25 LoopNest,
26 LoopSeq,
27 None,
28 Separating,
29 Last_ = Separating,
30};
31
32static constexpr std::size_t Association_enumSize = 7;
33
34enum class Category {
35 Declarative,
36 First_ = Declarative,
37 Executable,
38 Informational,
39 Meta,
40 Subsidiary,
41 Utility,
42 Last_ = Utility,
43};
44
45static constexpr std::size_t Category_enumSize = 6;
46
47enum class SourceLanguage : uint32_t {
48 C = 1U,
49 Fortran = 2U,
50 LLVM_MARK_AS_BITMASK_ENUM(/*LargestValue=*/Fortran)
51};
52
53static constexpr std::size_t SourceLanguage_enumSize = 2;
54
55enum class Directive {
56 ACCD_atomic,
57 First_ = ACCD_atomic,
58 ACCD_cache,
59 ACCD_data,
60 ACCD_declare,
61 ACCD_enter_data,
62 ACCD_exit_data,
63 ACCD_host_data,
64 ACCD_init,
65 ACCD_kernels,
66 ACCD_kernels_loop,
67 ACCD_loop,
68 ACCD_parallel,
69 ACCD_parallel_loop,
70 ACCD_routine,
71 ACCD_serial,
72 ACCD_serial_loop,
73 ACCD_set,
74 ACCD_shutdown,
75 ACCD_unknown,
76 ACCD_update,
77 ACCD_wait,
78 Last_ = ACCD_wait,
79};
80
81static constexpr std::size_t Directive_enumSize = 21;
82
83constexpr auto ACCD_atomic = Directive::ACCD_atomic;
84constexpr auto ACCD_cache = Directive::ACCD_cache;
85constexpr auto ACCD_data = Directive::ACCD_data;
86constexpr auto ACCD_declare = Directive::ACCD_declare;
87constexpr auto ACCD_enter_data = Directive::ACCD_enter_data;
88constexpr auto ACCD_exit_data = Directive::ACCD_exit_data;
89constexpr auto ACCD_host_data = Directive::ACCD_host_data;
90constexpr auto ACCD_init = Directive::ACCD_init;
91constexpr auto ACCD_kernels = Directive::ACCD_kernels;
92constexpr auto ACCD_kernels_loop = Directive::ACCD_kernels_loop;
93constexpr auto ACCD_loop = Directive::ACCD_loop;
94constexpr auto ACCD_parallel = Directive::ACCD_parallel;
95constexpr auto ACCD_parallel_loop = Directive::ACCD_parallel_loop;
96constexpr auto ACCD_routine = Directive::ACCD_routine;
97constexpr auto ACCD_serial = Directive::ACCD_serial;
98constexpr auto ACCD_serial_loop = Directive::ACCD_serial_loop;
99constexpr auto ACCD_set = Directive::ACCD_set;
100constexpr auto ACCD_shutdown = Directive::ACCD_shutdown;
101constexpr auto ACCD_unknown = Directive::ACCD_unknown;
102constexpr auto ACCD_update = Directive::ACCD_update;
103constexpr auto ACCD_wait = Directive::ACCD_wait;
104
105enum class Clause {
106 ACCC_async,
107 First_ = ACCC_async,
108 ACCC_attach,
109 ACCC_auto,
110 ACCC_bind,
111 ACCC_capture,
112 ACCC_collapse,
113 ACCC_copy,
114 ACCC_copyin,
115 ACCC_copyout,
116 ACCC_create,
117 ACCC_default,
118 ACCC_default_async,
119 ACCC_delete,
120 ACCC_detach,
121 ACCC_device,
122 ACCC_device_num,
123 ACCC_deviceptr,
124 ACCC_device_resident,
125 ACCC_device_type,
126 ACCC_finalize,
127 ACCC_firstprivate,
128 ACCC_gang,
129 ACCC_host,
130 ACCC_if,
131 ACCC_if_present,
132 ACCC_independent,
133 ACCC_link,
134 ACCC_no_create,
135 ACCC_nohost,
136 ACCC_num_gangs,
137 ACCC_num_workers,
138 ACCC_present,
139 ACCC_private,
140 ACCC_read,
141 ACCC_reduction,
142 ACCC_self,
143 ACCC_seq,
144 ACCC_shortloop,
145 ACCC_tile,
146 ACCC_unknown,
147 ACCC_use_device,
148 ACCC_vector,
149 ACCC_vector_length,
150 ACCC_wait,
151 ACCC_worker,
152 ACCC_write,
153 Last_ = ACCC_write,
154};
155
156static constexpr std::size_t Clause_enumSize = 46;
157
158constexpr auto ACCC_async = Clause::ACCC_async;
159constexpr auto ACCC_attach = Clause::ACCC_attach;
160constexpr auto ACCC_auto = Clause::ACCC_auto;
161constexpr auto ACCC_bind = Clause::ACCC_bind;
162constexpr auto ACCC_capture = Clause::ACCC_capture;
163constexpr auto ACCC_collapse = Clause::ACCC_collapse;
164constexpr auto ACCC_copy = Clause::ACCC_copy;
165constexpr auto ACCC_copyin = Clause::ACCC_copyin;
166constexpr auto ACCC_copyout = Clause::ACCC_copyout;
167constexpr auto ACCC_create = Clause::ACCC_create;
168constexpr auto ACCC_default = Clause::ACCC_default;
169constexpr auto ACCC_default_async = Clause::ACCC_default_async;
170constexpr auto ACCC_delete = Clause::ACCC_delete;
171constexpr auto ACCC_detach = Clause::ACCC_detach;
172constexpr auto ACCC_device = Clause::ACCC_device;
173constexpr auto ACCC_device_num = Clause::ACCC_device_num;
174constexpr auto ACCC_deviceptr = Clause::ACCC_deviceptr;
175constexpr auto ACCC_device_resident = Clause::ACCC_device_resident;
176constexpr auto ACCC_device_type = Clause::ACCC_device_type;
177constexpr auto ACCC_finalize = Clause::ACCC_finalize;
178constexpr auto ACCC_firstprivate = Clause::ACCC_firstprivate;
179constexpr auto ACCC_gang = Clause::ACCC_gang;
180constexpr auto ACCC_host = Clause::ACCC_host;
181constexpr auto ACCC_if = Clause::ACCC_if;
182constexpr auto ACCC_if_present = Clause::ACCC_if_present;
183constexpr auto ACCC_independent = Clause::ACCC_independent;
184constexpr auto ACCC_link = Clause::ACCC_link;
185constexpr auto ACCC_no_create = Clause::ACCC_no_create;
186constexpr auto ACCC_nohost = Clause::ACCC_nohost;
187constexpr auto ACCC_num_gangs = Clause::ACCC_num_gangs;
188constexpr auto ACCC_num_workers = Clause::ACCC_num_workers;
189constexpr auto ACCC_present = Clause::ACCC_present;
190constexpr auto ACCC_private = Clause::ACCC_private;
191constexpr auto ACCC_read = Clause::ACCC_read;
192constexpr auto ACCC_reduction = Clause::ACCC_reduction;
193constexpr auto ACCC_self = Clause::ACCC_self;
194constexpr auto ACCC_seq = Clause::ACCC_seq;
195constexpr auto ACCC_shortloop = Clause::ACCC_shortloop;
196constexpr auto ACCC_tile = Clause::ACCC_tile;
197constexpr auto ACCC_unknown = Clause::ACCC_unknown;
198constexpr auto ACCC_use_device = Clause::ACCC_use_device;
199constexpr auto ACCC_vector = Clause::ACCC_vector;
200constexpr auto ACCC_vector_length = Clause::ACCC_vector_length;
201constexpr auto ACCC_wait = Clause::ACCC_wait;
202constexpr auto ACCC_worker = Clause::ACCC_worker;
203constexpr auto ACCC_write = Clause::ACCC_write;
204
205enum class DefaultValue {
206 ACC_Default_present=0,
207 ACC_Default_none=1,
208};
209
210constexpr auto ACC_Default_present = DefaultValue::ACC_Default_present;
211constexpr auto ACC_Default_none = DefaultValue::ACC_Default_none;
212
213// Constexpr functions.
214
215constexpr Association getDirectiveAssociation(Directive Dir) {
216 switch (Dir) {
217 case ACCD_atomic:
218 return Association::Block;
219 case ACCD_cache:
220 return Association::None;
221 case ACCD_data:
222 return Association::Block;
223 case ACCD_declare:
224 return Association::None;
225 case ACCD_enter_data:
226 return Association::None;
227 case ACCD_exit_data:
228 return Association::None;
229 case ACCD_host_data:
230 return Association::Block;
231 case ACCD_init:
232 return Association::None;
233 case ACCD_kernels:
234 return Association::Block;
235 case ACCD_kernels_loop:
236 return Association::LoopNest;
237 case ACCD_loop:
238 return Association::LoopNest;
239 case ACCD_parallel:
240 return Association::Block;
241 case ACCD_parallel_loop:
242 return Association::LoopNest;
243 case ACCD_routine:
244 return Association::Declaration;
245 case ACCD_serial:
246 return Association::Block;
247 case ACCD_serial_loop:
248 return Association::LoopNest;
249 case ACCD_set:
250 return Association::None;
251 case ACCD_shutdown:
252 return Association::None;
253 case ACCD_unknown:
254 return Association::None;
255 case ACCD_update:
256 return Association::None;
257 case ACCD_wait:
258 return Association::None;
259 } // switch (Dir)
260#if !defined(__clang__) && defined(__GNUC__) && __GNUC__ < 9
261 abort();
262#else
263 llvm_unreachable("Unexpected directive");
264#endif
265}
266
267constexpr Category getDirectiveCategory(Directive Dir) {
268 switch (Dir) {
269 case ACCD_atomic:
270 return Category::Executable;
271 case ACCD_cache:
272 return Category::Executable;
273 case ACCD_data:
274 return Category::Executable;
275 case ACCD_declare:
276 return Category::Declarative;
277 case ACCD_enter_data:
278 return Category::Executable;
279 case ACCD_exit_data:
280 return Category::Executable;
281 case ACCD_host_data:
282 return Category::Executable;
283 case ACCD_init:
284 return Category::Executable;
285 case ACCD_kernels:
286 return Category::Executable;
287 case ACCD_kernels_loop:
288 return Category::Executable;
289 case ACCD_loop:
290 return Category::Executable;
291 case ACCD_parallel:
292 return Category::Executable;
293 case ACCD_parallel_loop:
294 return Category::Executable;
295 case ACCD_routine:
296 return Category::Declarative;
297 case ACCD_serial:
298 return Category::Executable;
299 case ACCD_serial_loop:
300 return Category::Executable;
301 case ACCD_set:
302 return Category::Executable;
303 case ACCD_shutdown:
304 return Category::Executable;
305 case ACCD_unknown:
306 return Category::Utility;
307 case ACCD_update:
308 return Category::Executable;
309 case ACCD_wait:
310 return Category::Executable;
311 } // switch (Dir)
312#if !defined(__clang__) && defined(__GNUC__) && __GNUC__ < 9
313 abort();
314#else
315 llvm_unreachable("Unexpected directive");
316#endif
317}
318
319constexpr SourceLanguage getDirectiveLanguages(Directive D) {
320 switch (D) {
321 case ACCD_atomic:
322 return SourceLanguage::C | SourceLanguage::Fortran;
323 case ACCD_cache:
324 return SourceLanguage::C | SourceLanguage::Fortran;
325 case ACCD_data:
326 return SourceLanguage::C | SourceLanguage::Fortran;
327 case ACCD_declare:
328 return SourceLanguage::C | SourceLanguage::Fortran;
329 case ACCD_enter_data:
330 return SourceLanguage::C | SourceLanguage::Fortran;
331 case ACCD_exit_data:
332 return SourceLanguage::C | SourceLanguage::Fortran;
333 case ACCD_host_data:
334 return SourceLanguage::C | SourceLanguage::Fortran;
335 case ACCD_init:
336 return SourceLanguage::C | SourceLanguage::Fortran;
337 case ACCD_kernels:
338 return SourceLanguage::C | SourceLanguage::Fortran;
339 case ACCD_kernels_loop:
340 return SourceLanguage::C | SourceLanguage::Fortran;
341 case ACCD_loop:
342 return SourceLanguage::C | SourceLanguage::Fortran;
343 case ACCD_parallel:
344 return SourceLanguage::C | SourceLanguage::Fortran;
345 case ACCD_parallel_loop:
346 return SourceLanguage::C | SourceLanguage::Fortran;
347 case ACCD_routine:
348 return SourceLanguage::C | SourceLanguage::Fortran;
349 case ACCD_serial:
350 return SourceLanguage::C | SourceLanguage::Fortran;
351 case ACCD_serial_loop:
352 return SourceLanguage::C | SourceLanguage::Fortran;
353 case ACCD_set:
354 return SourceLanguage::C | SourceLanguage::Fortran;
355 case ACCD_shutdown:
356 return SourceLanguage::C | SourceLanguage::Fortran;
357 case ACCD_unknown:
358 return SourceLanguage::C | SourceLanguage::Fortran;
359 case ACCD_update:
360 return SourceLanguage::C | SourceLanguage::Fortran;
361 case ACCD_wait:
362 return SourceLanguage::C | SourceLanguage::Fortran;
363 } // switch(D)
364#if !defined(__clang__) && defined(__GNUC__) && __GNUC__ < 9
365 abort();
366#else
367 llvm_unreachable("Unexpected directive");
368#endif
369}
370
371// Enumeration helper functions
372LLVM_ABI std::pair<Directive, directive::VersionRange> getOpenACCDirectiveKindAndVersions(StringRef Str);
373inline Directive getOpenACCDirectiveKind(StringRef Str) {
374 return getOpenACCDirectiveKindAndVersions(Str).first;
375}
376
377LLVM_ABI StringRef getOpenACCDirectiveName(Directive D, unsigned Ver = 0);
378
379LLVM_ABI std::pair<Clause, directive::VersionRange> getOpenACCClauseKindAndVersions(StringRef Str);
380
381inline Clause getOpenACCClauseKind(StringRef Str) {
382 return getOpenACCClauseKindAndVersions(Str).first;
383}
384
385LLVM_ABI StringRef getOpenACCClauseName(Clause C, unsigned Ver = 0);
386
387/// Return true if \p C is a valid clause for \p D in version \p Version.
388LLVM_ABI bool isAllowedClauseForDirective(Directive D, Clause C, unsigned Version);
389
390constexpr std::size_t getMaxLeafCount() { return 2; }
391LLVM_ABI DefaultValue getDefaultValue(StringRef Str);
392LLVM_ABI StringRef getOpenACCDefaultValueName(DefaultValue x);
393
394} // namespace acc
395
396template <> struct enum_iteration_traits<acc::Association> {
397 static constexpr bool is_iterable = true;
398};
399
400template <> struct enum_iteration_traits<acc::Category> {
401 static constexpr bool is_iterable = true;
402};
403
404template <> struct enum_iteration_traits<acc::Directive> {
405 static constexpr bool is_iterable = true;
406};
407
408template <> struct enum_iteration_traits<acc::Clause> {
409 static constexpr bool is_iterable = true;
410};
411
412} // namespace llvm
413
414#endif // LLVM_OpenACC_INC
415
416