1 //===--- OpenMPKinds.cpp - Token Kinds Support ----------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 /// \file
9 /// This file implements the OpenMP enum and support functions.
10 ///
11 //===----------------------------------------------------------------------===//
12 
13 #include "clang/Basic/OpenMPKinds.h"
14 #include "clang/Basic/IdentifierTable.h"
15 #include "llvm/ADT/StringRef.h"
16 #include "llvm/ADT/StringSwitch.h"
17 #include "llvm/Support/ErrorHandling.h"
18 #include <cassert>
19 
20 using namespace clang;
21 using namespace llvm::omp;
22 
23 unsigned clang::getOpenMPSimpleClauseType(OpenMPClauseKind Kind, StringRef Str,
24                                           const LangOptions &LangOpts) {
25   switch (Kind) {
26   case OMPC_default:
27     return llvm::StringSwitch<unsigned>(Str)
28 #define OMP_DEFAULT_KIND(Enum, Name) .Case(Name, unsigned(Enum))
29 #include "llvm/Frontend/OpenMP/OMPKinds.def"
30         .Default(unsigned(llvm::omp::OMP_DEFAULT_unknown));
31   case OMPC_proc_bind:
32     return llvm::StringSwitch<unsigned>(Str)
33 #define OMP_PROC_BIND_KIND(Enum, Name, Value) .Case(Name, Value)
34 #include "llvm/Frontend/OpenMP/OMPKinds.def"
35         .Default(unsigned(llvm::omp::OMP_PROC_BIND_unknown));
36   case OMPC_schedule:
37     return llvm::StringSwitch<unsigned>(Str)
38 #define OPENMP_SCHEDULE_KIND(Name)                                             \
39   .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_##Name))
40 #define OPENMP_SCHEDULE_MODIFIER(Name)                                         \
41   .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_MODIFIER_##Name))
42 #include "clang/Basic/OpenMPKinds.def"
43         .Default(OMPC_SCHEDULE_unknown);
44   case OMPC_depend:
45     return llvm::StringSwitch<OpenMPDependClauseKind>(Str)
46 #define OPENMP_DEPEND_KIND(Name) .Case(#Name, OMPC_DEPEND_##Name)
47 #include "clang/Basic/OpenMPKinds.def"
48         .Default(OMPC_DEPEND_unknown);
49   case OMPC_linear:
50     return llvm::StringSwitch<OpenMPLinearClauseKind>(Str)
51 #define OPENMP_LINEAR_KIND(Name) .Case(#Name, OMPC_LINEAR_##Name)
52 #include "clang/Basic/OpenMPKinds.def"
53         .Default(OMPC_LINEAR_unknown);
54   case OMPC_map: {
55     unsigned Type = llvm::StringSwitch<unsigned>(Str)
56 #define OPENMP_MAP_KIND(Name)                                                  \
57   .Case(#Name, static_cast<unsigned>(OMPC_MAP_##Name))
58 #define OPENMP_MAP_MODIFIER_KIND(Name)                                         \
59   .Case(#Name, static_cast<unsigned>(OMPC_MAP_MODIFIER_##Name))
60 #include "clang/Basic/OpenMPKinds.def"
61         .Default(OMPC_MAP_unknown);
62     if (LangOpts.OpenMP < 51 && Type == OMPC_MAP_MODIFIER_present)
63       return OMPC_MAP_MODIFIER_unknown;
64     if (!LangOpts.OpenMPExtensions && Type == OMPC_MAP_MODIFIER_ompx_hold)
65       return OMPC_MAP_MODIFIER_unknown;
66     return Type;
67   }
68   case OMPC_to:
69   case OMPC_from: {
70     unsigned Type = llvm::StringSwitch<unsigned>(Str)
71 #define OPENMP_MOTION_MODIFIER_KIND(Name)                                      \
72   .Case(#Name, static_cast<unsigned>(OMPC_MOTION_MODIFIER_##Name))
73 #include "clang/Basic/OpenMPKinds.def"
74         .Default(OMPC_MOTION_MODIFIER_unknown);
75     if (LangOpts.OpenMP < 51 && Type == OMPC_MOTION_MODIFIER_present)
76       return OMPC_MOTION_MODIFIER_unknown;
77     return Type;
78   }
79   case OMPC_dist_schedule:
80     return llvm::StringSwitch<OpenMPDistScheduleClauseKind>(Str)
81 #define OPENMP_DIST_SCHEDULE_KIND(Name) .Case(#Name, OMPC_DIST_SCHEDULE_##Name)
82 #include "clang/Basic/OpenMPKinds.def"
83         .Default(OMPC_DIST_SCHEDULE_unknown);
84   case OMPC_defaultmap:
85     return llvm::StringSwitch<unsigned>(Str)
86 #define OPENMP_DEFAULTMAP_KIND(Name)                                           \
87   .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_##Name))
88 #define OPENMP_DEFAULTMAP_MODIFIER(Name)                                       \
89   .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_MODIFIER_##Name))
90 #include "clang/Basic/OpenMPKinds.def"
91         .Default(OMPC_DEFAULTMAP_unknown);
92   case OMPC_atomic_default_mem_order:
93      return llvm::StringSwitch<OpenMPAtomicDefaultMemOrderClauseKind>(Str)
94 #define OPENMP_ATOMIC_DEFAULT_MEM_ORDER_KIND(Name)       \
95   .Case(#Name, OMPC_ATOMIC_DEFAULT_MEM_ORDER_##Name)
96 #include "clang/Basic/OpenMPKinds.def"
97         .Default(OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown);
98   case OMPC_device_type:
99     return llvm::StringSwitch<OpenMPDeviceType>(Str)
100 #define OPENMP_DEVICE_TYPE_KIND(Name) .Case(#Name, OMPC_DEVICE_TYPE_##Name)
101 #include "clang/Basic/OpenMPKinds.def"
102         .Default(OMPC_DEVICE_TYPE_unknown);
103   case OMPC_lastprivate:
104     return llvm::StringSwitch<OpenMPLastprivateModifier>(Str)
105 #define OPENMP_LASTPRIVATE_KIND(Name) .Case(#Name, OMPC_LASTPRIVATE_##Name)
106 #include "clang/Basic/OpenMPKinds.def"
107         .Default(OMPC_LASTPRIVATE_unknown);
108   case OMPC_order:
109     return llvm::StringSwitch<OpenMPOrderClauseKind>(Str)
110 #define OPENMP_ORDER_KIND(Name) .Case(#Name, OMPC_ORDER_##Name)
111 #include "clang/Basic/OpenMPKinds.def"
112         .Default(OMPC_ORDER_unknown);
113   case OMPC_update:
114     return llvm::StringSwitch<OpenMPDependClauseKind>(Str)
115 #define OPENMP_DEPEND_KIND(Name) .Case(#Name, OMPC_DEPEND_##Name)
116 #include "clang/Basic/OpenMPKinds.def"
117         .Default(OMPC_DEPEND_unknown);
118   case OMPC_device:
119     return llvm::StringSwitch<OpenMPDeviceClauseModifier>(Str)
120 #define OPENMP_DEVICE_MODIFIER(Name) .Case(#Name, OMPC_DEVICE_##Name)
121 #include "clang/Basic/OpenMPKinds.def"
122         .Default(OMPC_DEVICE_unknown);
123   case OMPC_reduction:
124     return llvm::StringSwitch<OpenMPReductionClauseModifier>(Str)
125 #define OPENMP_REDUCTION_MODIFIER(Name) .Case(#Name, OMPC_REDUCTION_##Name)
126 #include "clang/Basic/OpenMPKinds.def"
127         .Default(OMPC_REDUCTION_unknown);
128   case OMPC_unknown:
129   case OMPC_threadprivate:
130   case OMPC_if:
131   case OMPC_final:
132   case OMPC_num_threads:
133   case OMPC_safelen:
134   case OMPC_simdlen:
135   case OMPC_sizes:
136   case OMPC_allocator:
137   case OMPC_allocate:
138   case OMPC_collapse:
139   case OMPC_private:
140   case OMPC_firstprivate:
141   case OMPC_shared:
142   case OMPC_task_reduction:
143   case OMPC_in_reduction:
144   case OMPC_aligned:
145   case OMPC_copyin:
146   case OMPC_copyprivate:
147   case OMPC_ordered:
148   case OMPC_nowait:
149   case OMPC_untied:
150   case OMPC_mergeable:
151   case OMPC_flush:
152   case OMPC_depobj:
153   case OMPC_read:
154   case OMPC_write:
155   case OMPC_capture:
156   case OMPC_seq_cst:
157   case OMPC_acq_rel:
158   case OMPC_acquire:
159   case OMPC_release:
160   case OMPC_relaxed:
161   case OMPC_threads:
162   case OMPC_simd:
163   case OMPC_num_teams:
164   case OMPC_thread_limit:
165   case OMPC_priority:
166   case OMPC_grainsize:
167   case OMPC_nogroup:
168   case OMPC_num_tasks:
169   case OMPC_hint:
170   case OMPC_uniform:
171   case OMPC_use_device_ptr:
172   case OMPC_use_device_addr:
173   case OMPC_is_device_ptr:
174   case OMPC_unified_address:
175   case OMPC_unified_shared_memory:
176   case OMPC_reverse_offload:
177   case OMPC_dynamic_allocators:
178   case OMPC_match:
179   case OMPC_nontemporal:
180   case OMPC_destroy:
181   case OMPC_novariants:
182   case OMPC_nocontext:
183   case OMPC_detach:
184   case OMPC_inclusive:
185   case OMPC_exclusive:
186   case OMPC_uses_allocators:
187   case OMPC_affinity:
188     break;
189   default:
190     break;
191   }
192   llvm_unreachable("Invalid OpenMP simple clause kind");
193 }
194 
195 const char *clang::getOpenMPSimpleClauseTypeName(OpenMPClauseKind Kind,
196                                                  unsigned Type) {
197   switch (Kind) {
198   case OMPC_default:
199     switch (llvm::omp::DefaultKind(Type)) {
200 #define OMP_DEFAULT_KIND(Enum, Name)                                           \
201   case Enum:                                                                   \
202     return Name;
203 #include "llvm/Frontend/OpenMP/OMPKinds.def"
204     }
205     llvm_unreachable("Invalid OpenMP 'default' clause type");
206   case OMPC_proc_bind:
207     switch (Type) {
208 #define OMP_PROC_BIND_KIND(Enum, Name, Value)                                  \
209   case Value:                                                                  \
210     return Name;
211 #include "llvm/Frontend/OpenMP/OMPKinds.def"
212     }
213     llvm_unreachable("Invalid OpenMP 'proc_bind' clause type");
214   case OMPC_schedule:
215     switch (Type) {
216     case OMPC_SCHEDULE_unknown:
217     case OMPC_SCHEDULE_MODIFIER_last:
218       return "unknown";
219 #define OPENMP_SCHEDULE_KIND(Name)                                             \
220     case OMPC_SCHEDULE_##Name:                                                 \
221       return #Name;
222 #define OPENMP_SCHEDULE_MODIFIER(Name)                                         \
223     case OMPC_SCHEDULE_MODIFIER_##Name:                                        \
224       return #Name;
225 #include "clang/Basic/OpenMPKinds.def"
226     }
227     llvm_unreachable("Invalid OpenMP 'schedule' clause type");
228   case OMPC_depend:
229     switch (Type) {
230     case OMPC_DEPEND_unknown:
231       return "unknown";
232 #define OPENMP_DEPEND_KIND(Name)                                             \
233   case OMPC_DEPEND_##Name:                                                   \
234     return #Name;
235 #include "clang/Basic/OpenMPKinds.def"
236     }
237     llvm_unreachable("Invalid OpenMP 'depend' clause type");
238   case OMPC_linear:
239     switch (Type) {
240     case OMPC_LINEAR_unknown:
241       return "unknown";
242 #define OPENMP_LINEAR_KIND(Name)                                             \
243   case OMPC_LINEAR_##Name:                                                   \
244     return #Name;
245 #include "clang/Basic/OpenMPKinds.def"
246     }
247     llvm_unreachable("Invalid OpenMP 'linear' clause type");
248   case OMPC_map:
249     switch (Type) {
250     case OMPC_MAP_unknown:
251     case OMPC_MAP_MODIFIER_last:
252       return "unknown";
253 #define OPENMP_MAP_KIND(Name)                                                \
254   case OMPC_MAP_##Name:                                                      \
255     return #Name;
256 #define OPENMP_MAP_MODIFIER_KIND(Name)                                       \
257   case OMPC_MAP_MODIFIER_##Name:                                             \
258     return #Name;
259 #include "clang/Basic/OpenMPKinds.def"
260     default:
261       break;
262     }
263     llvm_unreachable("Invalid OpenMP 'map' clause type");
264   case OMPC_to:
265   case OMPC_from:
266     switch (Type) {
267     case OMPC_MOTION_MODIFIER_unknown:
268       return "unknown";
269 #define OPENMP_MOTION_MODIFIER_KIND(Name)                                      \
270   case OMPC_MOTION_MODIFIER_##Name:                                            \
271     return #Name;
272 #include "clang/Basic/OpenMPKinds.def"
273     default:
274       break;
275     }
276     llvm_unreachable("Invalid OpenMP 'to' or 'from' clause type");
277   case OMPC_dist_schedule:
278     switch (Type) {
279     case OMPC_DIST_SCHEDULE_unknown:
280       return "unknown";
281 #define OPENMP_DIST_SCHEDULE_KIND(Name)                                      \
282   case OMPC_DIST_SCHEDULE_##Name:                                            \
283     return #Name;
284 #include "clang/Basic/OpenMPKinds.def"
285     }
286     llvm_unreachable("Invalid OpenMP 'dist_schedule' clause type");
287   case OMPC_defaultmap:
288     switch (Type) {
289     case OMPC_DEFAULTMAP_unknown:
290     case OMPC_DEFAULTMAP_MODIFIER_last:
291       return "unknown";
292 #define OPENMP_DEFAULTMAP_KIND(Name)                                         \
293     case OMPC_DEFAULTMAP_##Name:                                             \
294       return #Name;
295 #define OPENMP_DEFAULTMAP_MODIFIER(Name)                                     \
296     case OMPC_DEFAULTMAP_MODIFIER_##Name:                                    \
297       return #Name;
298 #include "clang/Basic/OpenMPKinds.def"
299     }
300     llvm_unreachable("Invalid OpenMP 'schedule' clause type");
301   case OMPC_atomic_default_mem_order:
302     switch (Type) {
303     case OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown:
304       return "unknown";
305 #define OPENMP_ATOMIC_DEFAULT_MEM_ORDER_KIND(Name)                           \
306     case OMPC_ATOMIC_DEFAULT_MEM_ORDER_##Name:                               \
307       return #Name;
308 #include "clang/Basic/OpenMPKinds.def"
309 }
310     llvm_unreachable("Invalid OpenMP 'atomic_default_mem_order' clause type");
311   case OMPC_device_type:
312     switch (Type) {
313     case OMPC_DEVICE_TYPE_unknown:
314       return "unknown";
315 #define OPENMP_DEVICE_TYPE_KIND(Name)                                          \
316     case OMPC_DEVICE_TYPE_##Name:                                              \
317       return #Name;
318 #include "clang/Basic/OpenMPKinds.def"
319     }
320     llvm_unreachable("Invalid OpenMP 'device_type' clause type");
321   case OMPC_lastprivate:
322     switch (Type) {
323     case OMPC_LASTPRIVATE_unknown:
324       return "unknown";
325 #define OPENMP_LASTPRIVATE_KIND(Name)                                          \
326     case OMPC_LASTPRIVATE_##Name:                                              \
327       return #Name;
328 #include "clang/Basic/OpenMPKinds.def"
329     }
330     llvm_unreachable("Invalid OpenMP 'lastprivate' clause type");
331   case OMPC_order:
332     switch (Type) {
333     case OMPC_ORDER_unknown:
334       return "unknown";
335 #define OPENMP_ORDER_KIND(Name)                                                \
336     case OMPC_ORDER_##Name:                                                    \
337       return #Name;
338 #include "clang/Basic/OpenMPKinds.def"
339     }
340     llvm_unreachable("Invalid OpenMP 'order' clause type");
341   case OMPC_update:
342     switch (Type) {
343     case OMPC_DEPEND_unknown:
344       return "unknown";
345 #define OPENMP_DEPEND_KIND(Name)                                               \
346   case OMPC_DEPEND_##Name:                                                     \
347     return #Name;
348 #include "clang/Basic/OpenMPKinds.def"
349     }
350     llvm_unreachable("Invalid OpenMP 'depend' clause type");
351   case OMPC_device:
352     switch (Type) {
353     case OMPC_DEVICE_unknown:
354       return "unknown";
355 #define OPENMP_DEVICE_MODIFIER(Name)                                           \
356   case OMPC_DEVICE_##Name:                                                     \
357     return #Name;
358 #include "clang/Basic/OpenMPKinds.def"
359     }
360     llvm_unreachable("Invalid OpenMP 'device' clause modifier");
361   case OMPC_reduction:
362     switch (Type) {
363     case OMPC_REDUCTION_unknown:
364       return "unknown";
365 #define OPENMP_REDUCTION_MODIFIER(Name)                                        \
366   case OMPC_REDUCTION_##Name:                                                  \
367     return #Name;
368 #include "clang/Basic/OpenMPKinds.def"
369     }
370     llvm_unreachable("Invalid OpenMP 'reduction' clause modifier");
371   case OMPC_unknown:
372   case OMPC_threadprivate:
373   case OMPC_if:
374   case OMPC_final:
375   case OMPC_num_threads:
376   case OMPC_safelen:
377   case OMPC_simdlen:
378   case OMPC_sizes:
379   case OMPC_allocator:
380   case OMPC_allocate:
381   case OMPC_collapse:
382   case OMPC_private:
383   case OMPC_firstprivate:
384   case OMPC_shared:
385   case OMPC_task_reduction:
386   case OMPC_in_reduction:
387   case OMPC_aligned:
388   case OMPC_copyin:
389   case OMPC_copyprivate:
390   case OMPC_ordered:
391   case OMPC_nowait:
392   case OMPC_untied:
393   case OMPC_mergeable:
394   case OMPC_flush:
395   case OMPC_depobj:
396   case OMPC_read:
397   case OMPC_write:
398   case OMPC_capture:
399   case OMPC_seq_cst:
400   case OMPC_acq_rel:
401   case OMPC_acquire:
402   case OMPC_release:
403   case OMPC_relaxed:
404   case OMPC_threads:
405   case OMPC_simd:
406   case OMPC_num_teams:
407   case OMPC_thread_limit:
408   case OMPC_priority:
409   case OMPC_grainsize:
410   case OMPC_nogroup:
411   case OMPC_num_tasks:
412   case OMPC_hint:
413   case OMPC_uniform:
414   case OMPC_use_device_ptr:
415   case OMPC_use_device_addr:
416   case OMPC_is_device_ptr:
417   case OMPC_unified_address:
418   case OMPC_unified_shared_memory:
419   case OMPC_reverse_offload:
420   case OMPC_dynamic_allocators:
421   case OMPC_match:
422   case OMPC_nontemporal:
423   case OMPC_destroy:
424   case OMPC_detach:
425   case OMPC_novariants:
426   case OMPC_nocontext:
427   case OMPC_inclusive:
428   case OMPC_exclusive:
429   case OMPC_uses_allocators:
430   case OMPC_affinity:
431     break;
432   default:
433     break;
434   }
435   llvm_unreachable("Invalid OpenMP simple clause kind");
436 }
437 
438 bool clang::isOpenMPLoopDirective(OpenMPDirectiveKind DKind) {
439   return DKind == OMPD_simd || DKind == OMPD_for || DKind == OMPD_for_simd ||
440          DKind == OMPD_parallel_for || DKind == OMPD_parallel_for_simd ||
441          DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd ||
442          DKind == OMPD_master_taskloop || DKind == OMPD_master_taskloop_simd ||
443          DKind == OMPD_parallel_master_taskloop ||
444          DKind == OMPD_parallel_master_taskloop_simd ||
445          DKind == OMPD_distribute || DKind == OMPD_target_parallel_for ||
446          DKind == OMPD_distribute_parallel_for ||
447          DKind == OMPD_distribute_parallel_for_simd ||
448          DKind == OMPD_distribute_simd ||
449          DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd ||
450          DKind == OMPD_teams_distribute ||
451          DKind == OMPD_teams_distribute_simd ||
452          DKind == OMPD_teams_distribute_parallel_for_simd ||
453          DKind == OMPD_teams_distribute_parallel_for ||
454          DKind == OMPD_target_teams_distribute ||
455          DKind == OMPD_target_teams_distribute_parallel_for ||
456          DKind == OMPD_target_teams_distribute_parallel_for_simd ||
457          DKind == OMPD_target_teams_distribute_simd || DKind == OMPD_tile ||
458          DKind == OMPD_unroll;
459 }
460 
461 bool clang::isOpenMPWorksharingDirective(OpenMPDirectiveKind DKind) {
462   return DKind == OMPD_for || DKind == OMPD_for_simd ||
463          DKind == OMPD_sections || DKind == OMPD_section ||
464          DKind == OMPD_single || DKind == OMPD_parallel_for ||
465          DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections ||
466          DKind == OMPD_target_parallel_for ||
467          DKind == OMPD_distribute_parallel_for ||
468          DKind == OMPD_distribute_parallel_for_simd ||
469          DKind == OMPD_target_parallel_for_simd ||
470          DKind == OMPD_teams_distribute_parallel_for_simd ||
471          DKind == OMPD_teams_distribute_parallel_for ||
472          DKind == OMPD_target_teams_distribute_parallel_for ||
473          DKind == OMPD_target_teams_distribute_parallel_for_simd;
474 }
475 
476 bool clang::isOpenMPTaskLoopDirective(OpenMPDirectiveKind DKind) {
477   return DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd ||
478          DKind == OMPD_master_taskloop || DKind == OMPD_master_taskloop_simd ||
479          DKind == OMPD_parallel_master_taskloop ||
480          DKind == OMPD_parallel_master_taskloop_simd;
481 }
482 
483 bool clang::isOpenMPParallelDirective(OpenMPDirectiveKind DKind) {
484   return DKind == OMPD_parallel || DKind == OMPD_parallel_for ||
485          DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections ||
486          DKind == OMPD_target_parallel || DKind == OMPD_target_parallel_for ||
487          DKind == OMPD_distribute_parallel_for ||
488          DKind == OMPD_distribute_parallel_for_simd ||
489          DKind == OMPD_target_parallel_for_simd ||
490          DKind == OMPD_teams_distribute_parallel_for ||
491          DKind == OMPD_teams_distribute_parallel_for_simd ||
492          DKind == OMPD_target_teams_distribute_parallel_for ||
493          DKind == OMPD_target_teams_distribute_parallel_for_simd ||
494          DKind == OMPD_parallel_master ||
495          DKind == OMPD_parallel_master_taskloop ||
496          DKind == OMPD_parallel_master_taskloop_simd;
497 }
498 
499 bool clang::isOpenMPTargetExecutionDirective(OpenMPDirectiveKind DKind) {
500   return DKind == OMPD_target || DKind == OMPD_target_parallel ||
501          DKind == OMPD_target_parallel_for ||
502          DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd ||
503          DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute ||
504          DKind == OMPD_target_teams_distribute_parallel_for ||
505          DKind == OMPD_target_teams_distribute_parallel_for_simd ||
506          DKind == OMPD_target_teams_distribute_simd;
507 }
508 
509 bool clang::isOpenMPTargetDataManagementDirective(OpenMPDirectiveKind DKind) {
510   return DKind == OMPD_target_data || DKind == OMPD_target_enter_data ||
511          DKind == OMPD_target_exit_data || DKind == OMPD_target_update;
512 }
513 
514 bool clang::isOpenMPNestingTeamsDirective(OpenMPDirectiveKind DKind) {
515   return DKind == OMPD_teams || DKind == OMPD_teams_distribute ||
516          DKind == OMPD_teams_distribute_simd ||
517          DKind == OMPD_teams_distribute_parallel_for_simd ||
518          DKind == OMPD_teams_distribute_parallel_for;
519 }
520 
521 bool clang::isOpenMPTeamsDirective(OpenMPDirectiveKind DKind) {
522   return isOpenMPNestingTeamsDirective(DKind) ||
523          DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute ||
524          DKind == OMPD_target_teams_distribute_parallel_for ||
525          DKind == OMPD_target_teams_distribute_parallel_for_simd ||
526          DKind == OMPD_target_teams_distribute_simd;
527 }
528 
529 bool clang::isOpenMPSimdDirective(OpenMPDirectiveKind DKind) {
530   return DKind == OMPD_simd || DKind == OMPD_for_simd ||
531          DKind == OMPD_parallel_for_simd || DKind == OMPD_taskloop_simd ||
532          DKind == OMPD_master_taskloop_simd ||
533          DKind == OMPD_parallel_master_taskloop_simd ||
534          DKind == OMPD_distribute_parallel_for_simd ||
535          DKind == OMPD_distribute_simd || DKind == OMPD_target_simd ||
536          DKind == OMPD_teams_distribute_simd ||
537          DKind == OMPD_teams_distribute_parallel_for_simd ||
538          DKind == OMPD_target_teams_distribute_parallel_for_simd ||
539          DKind == OMPD_target_teams_distribute_simd ||
540          DKind == OMPD_target_parallel_for_simd;
541 }
542 
543 bool clang::isOpenMPNestingDistributeDirective(OpenMPDirectiveKind Kind) {
544   return Kind == OMPD_distribute || Kind == OMPD_distribute_parallel_for ||
545          Kind == OMPD_distribute_parallel_for_simd ||
546          Kind == OMPD_distribute_simd;
547   // TODO add next directives.
548 }
549 
550 bool clang::isOpenMPDistributeDirective(OpenMPDirectiveKind Kind) {
551   return isOpenMPNestingDistributeDirective(Kind) ||
552          Kind == OMPD_teams_distribute || Kind == OMPD_teams_distribute_simd ||
553          Kind == OMPD_teams_distribute_parallel_for_simd ||
554          Kind == OMPD_teams_distribute_parallel_for ||
555          Kind == OMPD_target_teams_distribute ||
556          Kind == OMPD_target_teams_distribute_parallel_for ||
557          Kind == OMPD_target_teams_distribute_parallel_for_simd ||
558          Kind == OMPD_target_teams_distribute_simd;
559 }
560 
561 bool clang::isOpenMPPrivate(OpenMPClauseKind Kind) {
562   return Kind == OMPC_private || Kind == OMPC_firstprivate ||
563          Kind == OMPC_lastprivate || Kind == OMPC_linear ||
564          Kind == OMPC_reduction || Kind == OMPC_task_reduction ||
565          Kind == OMPC_in_reduction; // TODO add next clauses like 'reduction'.
566 }
567 
568 bool clang::isOpenMPThreadPrivate(OpenMPClauseKind Kind) {
569   return Kind == OMPC_threadprivate || Kind == OMPC_copyin;
570 }
571 
572 bool clang::isOpenMPTaskingDirective(OpenMPDirectiveKind Kind) {
573   return Kind == OMPD_task || isOpenMPTaskLoopDirective(Kind);
574 }
575 
576 bool clang::isOpenMPLoopBoundSharingDirective(OpenMPDirectiveKind Kind) {
577   return Kind == OMPD_distribute_parallel_for ||
578          Kind == OMPD_distribute_parallel_for_simd ||
579          Kind == OMPD_teams_distribute_parallel_for_simd ||
580          Kind == OMPD_teams_distribute_parallel_for ||
581          Kind == OMPD_target_teams_distribute_parallel_for ||
582          Kind == OMPD_target_teams_distribute_parallel_for_simd;
583 }
584 
585 bool clang::isOpenMPLoopTransformationDirective(OpenMPDirectiveKind DKind) {
586   return DKind == OMPD_tile || DKind == OMPD_unroll;
587 }
588 
589 void clang::getOpenMPCaptureRegions(
590     SmallVectorImpl<OpenMPDirectiveKind> &CaptureRegions,
591     OpenMPDirectiveKind DKind) {
592   assert(unsigned(DKind) < llvm::omp::Directive_enumSize);
593   switch (DKind) {
594   case OMPD_parallel:
595   case OMPD_parallel_for:
596   case OMPD_parallel_for_simd:
597   case OMPD_parallel_master:
598   case OMPD_parallel_sections:
599   case OMPD_distribute_parallel_for:
600   case OMPD_distribute_parallel_for_simd:
601     CaptureRegions.push_back(OMPD_parallel);
602     break;
603   case OMPD_target_teams:
604   case OMPD_target_teams_distribute:
605   case OMPD_target_teams_distribute_simd:
606     CaptureRegions.push_back(OMPD_task);
607     CaptureRegions.push_back(OMPD_target);
608     CaptureRegions.push_back(OMPD_teams);
609     break;
610   case OMPD_teams:
611   case OMPD_teams_distribute:
612   case OMPD_teams_distribute_simd:
613     CaptureRegions.push_back(OMPD_teams);
614     break;
615   case OMPD_target:
616   case OMPD_target_simd:
617     CaptureRegions.push_back(OMPD_task);
618     CaptureRegions.push_back(OMPD_target);
619     break;
620   case OMPD_teams_distribute_parallel_for:
621   case OMPD_teams_distribute_parallel_for_simd:
622     CaptureRegions.push_back(OMPD_teams);
623     CaptureRegions.push_back(OMPD_parallel);
624     break;
625   case OMPD_target_parallel:
626   case OMPD_target_parallel_for:
627   case OMPD_target_parallel_for_simd:
628     CaptureRegions.push_back(OMPD_task);
629     CaptureRegions.push_back(OMPD_target);
630     CaptureRegions.push_back(OMPD_parallel);
631     break;
632   case OMPD_task:
633   case OMPD_target_enter_data:
634   case OMPD_target_exit_data:
635   case OMPD_target_update:
636     CaptureRegions.push_back(OMPD_task);
637     break;
638   case OMPD_taskloop:
639   case OMPD_taskloop_simd:
640   case OMPD_master_taskloop:
641   case OMPD_master_taskloop_simd:
642     CaptureRegions.push_back(OMPD_taskloop);
643     break;
644   case OMPD_parallel_master_taskloop:
645   case OMPD_parallel_master_taskloop_simd:
646     CaptureRegions.push_back(OMPD_parallel);
647     CaptureRegions.push_back(OMPD_taskloop);
648     break;
649   case OMPD_target_teams_distribute_parallel_for:
650   case OMPD_target_teams_distribute_parallel_for_simd:
651     CaptureRegions.push_back(OMPD_task);
652     CaptureRegions.push_back(OMPD_target);
653     CaptureRegions.push_back(OMPD_teams);
654     CaptureRegions.push_back(OMPD_parallel);
655     break;
656   case OMPD_simd:
657   case OMPD_for:
658   case OMPD_for_simd:
659   case OMPD_sections:
660   case OMPD_section:
661   case OMPD_single:
662   case OMPD_master:
663   case OMPD_critical:
664   case OMPD_taskgroup:
665   case OMPD_distribute:
666   case OMPD_ordered:
667   case OMPD_atomic:
668   case OMPD_target_data:
669   case OMPD_distribute_simd:
670   case OMPD_dispatch:
671     CaptureRegions.push_back(OMPD_unknown);
672     break;
673   case OMPD_tile:
674   case OMPD_unroll:
675     // loop transformations do not introduce captures.
676     break;
677   case OMPD_threadprivate:
678   case OMPD_allocate:
679   case OMPD_taskyield:
680   case OMPD_barrier:
681   case OMPD_taskwait:
682   case OMPD_cancellation_point:
683   case OMPD_cancel:
684   case OMPD_flush:
685   case OMPD_depobj:
686   case OMPD_scan:
687   case OMPD_declare_reduction:
688   case OMPD_declare_mapper:
689   case OMPD_declare_simd:
690   case OMPD_declare_target:
691   case OMPD_end_declare_target:
692   case OMPD_requires:
693   case OMPD_declare_variant:
694   case OMPD_begin_declare_variant:
695   case OMPD_end_declare_variant:
696     llvm_unreachable("OpenMP Directive is not allowed");
697   case OMPD_unknown:
698   default:
699     llvm_unreachable("Unknown OpenMP directive");
700   }
701 }
702