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