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 } 500 501 bool clang::isOpenMPWorksharingDirective(OpenMPDirectiveKind DKind) { 502 return DKind == OMPD_for || DKind == OMPD_for_simd || 503 DKind == OMPD_sections || DKind == OMPD_section || 504 DKind == OMPD_single || DKind == OMPD_parallel_for || 505 DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections || 506 DKind == OMPD_target_parallel_for || 507 DKind == OMPD_distribute_parallel_for || 508 DKind == OMPD_distribute_parallel_for_simd || 509 DKind == OMPD_target_parallel_for_simd || 510 DKind == OMPD_teams_distribute_parallel_for_simd || 511 DKind == OMPD_teams_distribute_parallel_for || 512 DKind == OMPD_target_teams_distribute_parallel_for || 513 DKind == OMPD_target_teams_distribute_parallel_for_simd; 514 } 515 516 bool clang::isOpenMPTaskLoopDirective(OpenMPDirectiveKind DKind) { 517 return DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd || 518 DKind == OMPD_master_taskloop || DKind == OMPD_master_taskloop_simd || 519 DKind == OMPD_parallel_master_taskloop || 520 DKind == OMPD_parallel_master_taskloop_simd; 521 } 522 523 bool clang::isOpenMPParallelDirective(OpenMPDirectiveKind DKind) { 524 return DKind == OMPD_parallel || DKind == OMPD_parallel_for || 525 DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections || 526 DKind == OMPD_target_parallel || DKind == OMPD_target_parallel_for || 527 DKind == OMPD_distribute_parallel_for || 528 DKind == OMPD_distribute_parallel_for_simd || 529 DKind == OMPD_target_parallel_for_simd || 530 DKind == OMPD_teams_distribute_parallel_for || 531 DKind == OMPD_teams_distribute_parallel_for_simd || 532 DKind == OMPD_target_teams_distribute_parallel_for || 533 DKind == OMPD_target_teams_distribute_parallel_for_simd || 534 DKind == OMPD_parallel_master || 535 DKind == OMPD_parallel_master_taskloop || 536 DKind == OMPD_parallel_master_taskloop_simd; 537 } 538 539 bool clang::isOpenMPTargetExecutionDirective(OpenMPDirectiveKind DKind) { 540 return DKind == OMPD_target || DKind == OMPD_target_parallel || 541 DKind == OMPD_target_parallel_for || 542 DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd || 543 DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute || 544 DKind == OMPD_target_teams_distribute_parallel_for || 545 DKind == OMPD_target_teams_distribute_parallel_for_simd || 546 DKind == OMPD_target_teams_distribute_simd; 547 } 548 549 bool clang::isOpenMPTargetDataManagementDirective(OpenMPDirectiveKind DKind) { 550 return DKind == OMPD_target_data || DKind == OMPD_target_enter_data || 551 DKind == OMPD_target_exit_data || DKind == OMPD_target_update; 552 } 553 554 bool clang::isOpenMPNestingTeamsDirective(OpenMPDirectiveKind DKind) { 555 return DKind == OMPD_teams || DKind == OMPD_teams_distribute || 556 DKind == OMPD_teams_distribute_simd || 557 DKind == OMPD_teams_distribute_parallel_for_simd || 558 DKind == OMPD_teams_distribute_parallel_for; 559 } 560 561 bool clang::isOpenMPTeamsDirective(OpenMPDirectiveKind DKind) { 562 return isOpenMPNestingTeamsDirective(DKind) || 563 DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute || 564 DKind == OMPD_target_teams_distribute_parallel_for || 565 DKind == OMPD_target_teams_distribute_parallel_for_simd || 566 DKind == OMPD_target_teams_distribute_simd; 567 } 568 569 bool clang::isOpenMPSimdDirective(OpenMPDirectiveKind DKind) { 570 return DKind == OMPD_simd || DKind == OMPD_for_simd || 571 DKind == OMPD_parallel_for_simd || DKind == OMPD_taskloop_simd || 572 DKind == OMPD_master_taskloop_simd || 573 DKind == OMPD_parallel_master_taskloop_simd || 574 DKind == OMPD_distribute_parallel_for_simd || 575 DKind == OMPD_distribute_simd || DKind == OMPD_target_simd || 576 DKind == OMPD_teams_distribute_simd || 577 DKind == OMPD_teams_distribute_parallel_for_simd || 578 DKind == OMPD_target_teams_distribute_parallel_for_simd || 579 DKind == OMPD_target_teams_distribute_simd || 580 DKind == OMPD_target_parallel_for_simd; 581 } 582 583 bool clang::isOpenMPNestingDistributeDirective(OpenMPDirectiveKind Kind) { 584 return Kind == OMPD_distribute || Kind == OMPD_distribute_parallel_for || 585 Kind == OMPD_distribute_parallel_for_simd || 586 Kind == OMPD_distribute_simd; 587 // TODO add next directives. 588 } 589 590 bool clang::isOpenMPDistributeDirective(OpenMPDirectiveKind Kind) { 591 return isOpenMPNestingDistributeDirective(Kind) || 592 Kind == OMPD_teams_distribute || Kind == OMPD_teams_distribute_simd || 593 Kind == OMPD_teams_distribute_parallel_for_simd || 594 Kind == OMPD_teams_distribute_parallel_for || 595 Kind == OMPD_target_teams_distribute || 596 Kind == OMPD_target_teams_distribute_parallel_for || 597 Kind == OMPD_target_teams_distribute_parallel_for_simd || 598 Kind == OMPD_target_teams_distribute_simd; 599 } 600 601 bool clang::isOpenMPGenericLoopDirective(OpenMPDirectiveKind Kind) { 602 return Kind == OMPD_loop; 603 } 604 605 bool clang::isOpenMPPrivate(OpenMPClauseKind Kind) { 606 return Kind == OMPC_private || Kind == OMPC_firstprivate || 607 Kind == OMPC_lastprivate || Kind == OMPC_linear || 608 Kind == OMPC_reduction || Kind == OMPC_task_reduction || 609 Kind == OMPC_in_reduction; // TODO add next clauses like 'reduction'. 610 } 611 612 bool clang::isOpenMPThreadPrivate(OpenMPClauseKind Kind) { 613 return Kind == OMPC_threadprivate || Kind == OMPC_copyin; 614 } 615 616 bool clang::isOpenMPTaskingDirective(OpenMPDirectiveKind Kind) { 617 return Kind == OMPD_task || isOpenMPTaskLoopDirective(Kind); 618 } 619 620 bool clang::isOpenMPLoopBoundSharingDirective(OpenMPDirectiveKind Kind) { 621 return Kind == OMPD_distribute_parallel_for || 622 Kind == OMPD_distribute_parallel_for_simd || 623 Kind == OMPD_teams_distribute_parallel_for_simd || 624 Kind == OMPD_teams_distribute_parallel_for || 625 Kind == OMPD_target_teams_distribute_parallel_for || 626 Kind == OMPD_target_teams_distribute_parallel_for_simd; 627 } 628 629 bool clang::isOpenMPLoopTransformationDirective(OpenMPDirectiveKind DKind) { 630 return DKind == OMPD_tile || DKind == OMPD_unroll; 631 } 632 633 void clang::getOpenMPCaptureRegions( 634 SmallVectorImpl<OpenMPDirectiveKind> &CaptureRegions, 635 OpenMPDirectiveKind DKind) { 636 assert(unsigned(DKind) < llvm::omp::Directive_enumSize); 637 switch (DKind) { 638 case OMPD_metadirective: 639 CaptureRegions.push_back(OMPD_metadirective); 640 break; 641 case OMPD_parallel: 642 case OMPD_parallel_for: 643 case OMPD_parallel_for_simd: 644 case OMPD_parallel_master: 645 case OMPD_parallel_sections: 646 case OMPD_distribute_parallel_for: 647 case OMPD_distribute_parallel_for_simd: 648 CaptureRegions.push_back(OMPD_parallel); 649 break; 650 case OMPD_target_teams: 651 case OMPD_target_teams_distribute: 652 case OMPD_target_teams_distribute_simd: 653 CaptureRegions.push_back(OMPD_task); 654 CaptureRegions.push_back(OMPD_target); 655 CaptureRegions.push_back(OMPD_teams); 656 break; 657 case OMPD_teams: 658 case OMPD_teams_distribute: 659 case OMPD_teams_distribute_simd: 660 CaptureRegions.push_back(OMPD_teams); 661 break; 662 case OMPD_target: 663 case OMPD_target_simd: 664 CaptureRegions.push_back(OMPD_task); 665 CaptureRegions.push_back(OMPD_target); 666 break; 667 case OMPD_teams_distribute_parallel_for: 668 case OMPD_teams_distribute_parallel_for_simd: 669 CaptureRegions.push_back(OMPD_teams); 670 CaptureRegions.push_back(OMPD_parallel); 671 break; 672 case OMPD_target_parallel: 673 case OMPD_target_parallel_for: 674 case OMPD_target_parallel_for_simd: 675 CaptureRegions.push_back(OMPD_task); 676 CaptureRegions.push_back(OMPD_target); 677 CaptureRegions.push_back(OMPD_parallel); 678 break; 679 case OMPD_task: 680 case OMPD_target_enter_data: 681 case OMPD_target_exit_data: 682 case OMPD_target_update: 683 CaptureRegions.push_back(OMPD_task); 684 break; 685 case OMPD_taskloop: 686 case OMPD_taskloop_simd: 687 case OMPD_master_taskloop: 688 case OMPD_master_taskloop_simd: 689 CaptureRegions.push_back(OMPD_taskloop); 690 break; 691 case OMPD_parallel_master_taskloop: 692 case OMPD_parallel_master_taskloop_simd: 693 CaptureRegions.push_back(OMPD_parallel); 694 CaptureRegions.push_back(OMPD_taskloop); 695 break; 696 case OMPD_target_teams_distribute_parallel_for: 697 case OMPD_target_teams_distribute_parallel_for_simd: 698 CaptureRegions.push_back(OMPD_task); 699 CaptureRegions.push_back(OMPD_target); 700 CaptureRegions.push_back(OMPD_teams); 701 CaptureRegions.push_back(OMPD_parallel); 702 break; 703 case OMPD_loop: 704 // TODO: 'loop' may require different capture regions depending on the bind 705 // clause or the parent directive when there is no bind clause. Use 706 // OMPD_unknown for now. 707 case OMPD_simd: 708 case OMPD_for: 709 case OMPD_for_simd: 710 case OMPD_sections: 711 case OMPD_section: 712 case OMPD_single: 713 case OMPD_master: 714 case OMPD_critical: 715 case OMPD_taskgroup: 716 case OMPD_distribute: 717 case OMPD_ordered: 718 case OMPD_atomic: 719 case OMPD_target_data: 720 case OMPD_distribute_simd: 721 case OMPD_dispatch: 722 CaptureRegions.push_back(OMPD_unknown); 723 break; 724 case OMPD_tile: 725 case OMPD_unroll: 726 // loop transformations do not introduce captures. 727 break; 728 case OMPD_threadprivate: 729 case OMPD_allocate: 730 case OMPD_taskyield: 731 case OMPD_barrier: 732 case OMPD_taskwait: 733 case OMPD_cancellation_point: 734 case OMPD_cancel: 735 case OMPD_flush: 736 case OMPD_depobj: 737 case OMPD_scan: 738 case OMPD_declare_reduction: 739 case OMPD_declare_mapper: 740 case OMPD_declare_simd: 741 case OMPD_declare_target: 742 case OMPD_end_declare_target: 743 case OMPD_requires: 744 case OMPD_declare_variant: 745 case OMPD_begin_declare_variant: 746 case OMPD_end_declare_variant: 747 llvm_unreachable("OpenMP Directive is not allowed"); 748 case OMPD_unknown: 749 default: 750 llvm_unreachable("Unknown OpenMP directive"); 751 } 752 } 753