| //===--- OpenMPKinds.cpp - Token Kinds Support ----------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| /// \file |
| /// This file implements the OpenMP enum and support functions. |
| /// |
| //===----------------------------------------------------------------------===// |
| |
| #include "clang/Basic/OpenMPKinds.h" |
| #include "clang/Basic/IdentifierTable.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/ADT/StringSwitch.h" |
| #include "llvm/Support/ErrorHandling.h" |
| #include <cassert> |
| |
| using namespace clang; |
| using namespace llvm::omp; |
| |
| unsigned clang::getOpenMPSimpleClauseType(OpenMPClauseKind Kind, StringRef Str, |
| unsigned OpenMPVersion) { |
| switch (Kind) { |
| case OMPC_default: |
| return llvm::StringSwitch<unsigned>(Str) |
| #define OMP_DEFAULT_KIND(Enum, Name) .Case(Name, unsigned(Enum)) |
| #include "llvm/Frontend/OpenMP/OMPKinds.def" |
| .Default(unsigned(llvm::omp::OMP_DEFAULT_unknown)); |
| case OMPC_proc_bind: |
| return llvm::StringSwitch<unsigned>(Str) |
| #define OMP_PROC_BIND_KIND(Enum, Name, Value) .Case(Name, Value) |
| #include "llvm/Frontend/OpenMP/OMPKinds.def" |
| .Default(unsigned(llvm::omp::OMP_PROC_BIND_unknown)); |
| case OMPC_schedule: |
| return llvm::StringSwitch<unsigned>(Str) |
| #define OPENMP_SCHEDULE_KIND(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_##Name)) |
| #define OPENMP_SCHEDULE_MODIFIER(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_SCHEDULE_MODIFIER_##Name)) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_SCHEDULE_unknown); |
| case OMPC_depend: |
| return llvm::StringSwitch<OpenMPDependClauseKind>(Str) |
| #define OPENMP_DEPEND_KIND(Name) .Case(#Name, OMPC_DEPEND_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DEPEND_unknown); |
| case OMPC_linear: |
| return llvm::StringSwitch<OpenMPLinearClauseKind>(Str) |
| #define OPENMP_LINEAR_KIND(Name) .Case(#Name, OMPC_LINEAR_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_LINEAR_unknown); |
| case OMPC_map: { |
| unsigned Type = llvm::StringSwitch<unsigned>(Str) |
| #define OPENMP_MAP_KIND(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_MAP_##Name)) |
| #define OPENMP_MAP_MODIFIER_KIND(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_MAP_MODIFIER_##Name)) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_MAP_unknown); |
| if (OpenMPVersion < 51 && Type == OMPC_MAP_MODIFIER_present) |
| return OMPC_MAP_MODIFIER_unknown; |
| return Type; |
| } |
| case OMPC_to: |
| case OMPC_from: { |
| unsigned Type = llvm::StringSwitch<unsigned>(Str) |
| #define OPENMP_MOTION_MODIFIER_KIND(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_MOTION_MODIFIER_##Name)) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_MOTION_MODIFIER_unknown); |
| if (OpenMPVersion < 51 && Type == OMPC_MOTION_MODIFIER_present) |
| return OMPC_MOTION_MODIFIER_unknown; |
| return Type; |
| } |
| case OMPC_dist_schedule: |
| return llvm::StringSwitch<OpenMPDistScheduleClauseKind>(Str) |
| #define OPENMP_DIST_SCHEDULE_KIND(Name) .Case(#Name, OMPC_DIST_SCHEDULE_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DIST_SCHEDULE_unknown); |
| case OMPC_defaultmap: |
| return llvm::StringSwitch<unsigned>(Str) |
| #define OPENMP_DEFAULTMAP_KIND(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_##Name)) |
| #define OPENMP_DEFAULTMAP_MODIFIER(Name) \ |
| .Case(#Name, static_cast<unsigned>(OMPC_DEFAULTMAP_MODIFIER_##Name)) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DEFAULTMAP_unknown); |
| case OMPC_atomic_default_mem_order: |
| return llvm::StringSwitch<OpenMPAtomicDefaultMemOrderClauseKind>(Str) |
| #define OPENMP_ATOMIC_DEFAULT_MEM_ORDER_KIND(Name) \ |
| .Case(#Name, OMPC_ATOMIC_DEFAULT_MEM_ORDER_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown); |
| case OMPC_device_type: |
| return llvm::StringSwitch<OpenMPDeviceType>(Str) |
| #define OPENMP_DEVICE_TYPE_KIND(Name) .Case(#Name, OMPC_DEVICE_TYPE_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DEVICE_TYPE_unknown); |
| case OMPC_lastprivate: |
| return llvm::StringSwitch<OpenMPLastprivateModifier>(Str) |
| #define OPENMP_LASTPRIVATE_KIND(Name) .Case(#Name, OMPC_LASTPRIVATE_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_LASTPRIVATE_unknown); |
| case OMPC_order: |
| return llvm::StringSwitch<OpenMPOrderClauseKind>(Str) |
| #define OPENMP_ORDER_KIND(Name) .Case(#Name, OMPC_ORDER_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_ORDER_unknown); |
| case OMPC_update: |
| return llvm::StringSwitch<OpenMPDependClauseKind>(Str) |
| #define OPENMP_DEPEND_KIND(Name) .Case(#Name, OMPC_DEPEND_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DEPEND_unknown); |
| case OMPC_device: |
| return llvm::StringSwitch<OpenMPDeviceClauseModifier>(Str) |
| #define OPENMP_DEVICE_MODIFIER(Name) .Case(#Name, OMPC_DEVICE_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_DEVICE_unknown); |
| case OMPC_reduction: |
| return llvm::StringSwitch<OpenMPReductionClauseModifier>(Str) |
| #define OPENMP_REDUCTION_MODIFIER(Name) .Case(#Name, OMPC_REDUCTION_##Name) |
| #include "clang/Basic/OpenMPKinds.def" |
| .Default(OMPC_REDUCTION_unknown); |
| case OMPC_unknown: |
| case OMPC_threadprivate: |
| case OMPC_if: |
| case OMPC_final: |
| case OMPC_num_threads: |
| case OMPC_safelen: |
| case OMPC_simdlen: |
| case OMPC_allocator: |
| case OMPC_allocate: |
| case OMPC_collapse: |
| case OMPC_private: |
| case OMPC_firstprivate: |
| case OMPC_shared: |
| case OMPC_task_reduction: |
| case OMPC_in_reduction: |
| case OMPC_aligned: |
| case OMPC_copyin: |
| case OMPC_copyprivate: |
| case OMPC_ordered: |
| case OMPC_nowait: |
| case OMPC_untied: |
| case OMPC_mergeable: |
| case OMPC_flush: |
| case OMPC_depobj: |
| case OMPC_read: |
| case OMPC_write: |
| case OMPC_capture: |
| case OMPC_seq_cst: |
| case OMPC_acq_rel: |
| case OMPC_acquire: |
| case OMPC_release: |
| case OMPC_relaxed: |
| case OMPC_threads: |
| case OMPC_simd: |
| case OMPC_num_teams: |
| case OMPC_thread_limit: |
| case OMPC_priority: |
| case OMPC_grainsize: |
| case OMPC_nogroup: |
| case OMPC_num_tasks: |
| case OMPC_hint: |
| case OMPC_uniform: |
| case OMPC_use_device_ptr: |
| case OMPC_use_device_addr: |
| case OMPC_is_device_ptr: |
| case OMPC_unified_address: |
| case OMPC_unified_shared_memory: |
| case OMPC_reverse_offload: |
| case OMPC_dynamic_allocators: |
| case OMPC_match: |
| case OMPC_nontemporal: |
| case OMPC_destroy: |
| case OMPC_detach: |
| case OMPC_inclusive: |
| case OMPC_exclusive: |
| case OMPC_uses_allocators: |
| case OMPC_affinity: |
| break; |
| default: |
| break; |
| } |
| llvm_unreachable("Invalid OpenMP simple clause kind"); |
| } |
| |
| const char *clang::getOpenMPSimpleClauseTypeName(OpenMPClauseKind Kind, |
| unsigned Type) { |
| switch (Kind) { |
| case OMPC_default: |
| switch (llvm::omp::DefaultKind(Type)) { |
| #define OMP_DEFAULT_KIND(Enum, Name) \ |
| case Enum: \ |
| return Name; |
| #include "llvm/Frontend/OpenMP/OMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'default' clause type"); |
| case OMPC_proc_bind: |
| switch (Type) { |
| #define OMP_PROC_BIND_KIND(Enum, Name, Value) \ |
| case Value: \ |
| return Name; |
| #include "llvm/Frontend/OpenMP/OMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'proc_bind' clause type"); |
| case OMPC_schedule: |
| switch (Type) { |
| case OMPC_SCHEDULE_unknown: |
| case OMPC_SCHEDULE_MODIFIER_last: |
| return "unknown"; |
| #define OPENMP_SCHEDULE_KIND(Name) \ |
| case OMPC_SCHEDULE_##Name: \ |
| return #Name; |
| #define OPENMP_SCHEDULE_MODIFIER(Name) \ |
| case OMPC_SCHEDULE_MODIFIER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'schedule' clause type"); |
| case OMPC_depend: |
| switch (Type) { |
| case OMPC_DEPEND_unknown: |
| return "unknown"; |
| #define OPENMP_DEPEND_KIND(Name) \ |
| case OMPC_DEPEND_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'depend' clause type"); |
| case OMPC_linear: |
| switch (Type) { |
| case OMPC_LINEAR_unknown: |
| return "unknown"; |
| #define OPENMP_LINEAR_KIND(Name) \ |
| case OMPC_LINEAR_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'linear' clause type"); |
| case OMPC_map: |
| switch (Type) { |
| case OMPC_MAP_unknown: |
| case OMPC_MAP_MODIFIER_last: |
| return "unknown"; |
| #define OPENMP_MAP_KIND(Name) \ |
| case OMPC_MAP_##Name: \ |
| return #Name; |
| #define OPENMP_MAP_MODIFIER_KIND(Name) \ |
| case OMPC_MAP_MODIFIER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| default: |
| break; |
| } |
| llvm_unreachable("Invalid OpenMP 'map' clause type"); |
| case OMPC_to: |
| case OMPC_from: |
| switch (Type) { |
| case OMPC_MOTION_MODIFIER_unknown: |
| return "unknown"; |
| #define OPENMP_MOTION_MODIFIER_KIND(Name) \ |
| case OMPC_MOTION_MODIFIER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| default: |
| break; |
| } |
| llvm_unreachable("Invalid OpenMP 'to' or 'from' clause type"); |
| case OMPC_dist_schedule: |
| switch (Type) { |
| case OMPC_DIST_SCHEDULE_unknown: |
| return "unknown"; |
| #define OPENMP_DIST_SCHEDULE_KIND(Name) \ |
| case OMPC_DIST_SCHEDULE_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'dist_schedule' clause type"); |
| case OMPC_defaultmap: |
| switch (Type) { |
| case OMPC_DEFAULTMAP_unknown: |
| case OMPC_DEFAULTMAP_MODIFIER_last: |
| return "unknown"; |
| #define OPENMP_DEFAULTMAP_KIND(Name) \ |
| case OMPC_DEFAULTMAP_##Name: \ |
| return #Name; |
| #define OPENMP_DEFAULTMAP_MODIFIER(Name) \ |
| case OMPC_DEFAULTMAP_MODIFIER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'schedule' clause type"); |
| case OMPC_atomic_default_mem_order: |
| switch (Type) { |
| case OMPC_ATOMIC_DEFAULT_MEM_ORDER_unknown: |
| return "unknown"; |
| #define OPENMP_ATOMIC_DEFAULT_MEM_ORDER_KIND(Name) \ |
| case OMPC_ATOMIC_DEFAULT_MEM_ORDER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'atomic_default_mem_order' clause type"); |
| case OMPC_device_type: |
| switch (Type) { |
| case OMPC_DEVICE_TYPE_unknown: |
| return "unknown"; |
| #define OPENMP_DEVICE_TYPE_KIND(Name) \ |
| case OMPC_DEVICE_TYPE_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'device_type' clause type"); |
| case OMPC_lastprivate: |
| switch (Type) { |
| case OMPC_LASTPRIVATE_unknown: |
| return "unknown"; |
| #define OPENMP_LASTPRIVATE_KIND(Name) \ |
| case OMPC_LASTPRIVATE_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'lastprivate' clause type"); |
| case OMPC_order: |
| switch (Type) { |
| case OMPC_ORDER_unknown: |
| return "unknown"; |
| #define OPENMP_ORDER_KIND(Name) \ |
| case OMPC_ORDER_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'order' clause type"); |
| case OMPC_update: |
| switch (Type) { |
| case OMPC_DEPEND_unknown: |
| return "unknown"; |
| #define OPENMP_DEPEND_KIND(Name) \ |
| case OMPC_DEPEND_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'depend' clause type"); |
| case OMPC_device: |
| switch (Type) { |
| case OMPC_DEVICE_unknown: |
| return "unknown"; |
| #define OPENMP_DEVICE_MODIFIER(Name) \ |
| case OMPC_DEVICE_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'device' clause modifier"); |
| case OMPC_reduction: |
| switch (Type) { |
| case OMPC_REDUCTION_unknown: |
| return "unknown"; |
| #define OPENMP_REDUCTION_MODIFIER(Name) \ |
| case OMPC_REDUCTION_##Name: \ |
| return #Name; |
| #include "clang/Basic/OpenMPKinds.def" |
| } |
| llvm_unreachable("Invalid OpenMP 'reduction' clause modifier"); |
| case OMPC_unknown: |
| case OMPC_threadprivate: |
| case OMPC_if: |
| case OMPC_final: |
| case OMPC_num_threads: |
| case OMPC_safelen: |
| case OMPC_simdlen: |
| case OMPC_allocator: |
| case OMPC_allocate: |
| case OMPC_collapse: |
| case OMPC_private: |
| case OMPC_firstprivate: |
| case OMPC_shared: |
| case OMPC_task_reduction: |
| case OMPC_in_reduction: |
| case OMPC_aligned: |
| case OMPC_copyin: |
| case OMPC_copyprivate: |
| case OMPC_ordered: |
| case OMPC_nowait: |
| case OMPC_untied: |
| case OMPC_mergeable: |
| case OMPC_flush: |
| case OMPC_depobj: |
| case OMPC_read: |
| case OMPC_write: |
| case OMPC_capture: |
| case OMPC_seq_cst: |
| case OMPC_acq_rel: |
| case OMPC_acquire: |
| case OMPC_release: |
| case OMPC_relaxed: |
| case OMPC_threads: |
| case OMPC_simd: |
| case OMPC_num_teams: |
| case OMPC_thread_limit: |
| case OMPC_priority: |
| case OMPC_grainsize: |
| case OMPC_nogroup: |
| case OMPC_num_tasks: |
| case OMPC_hint: |
| case OMPC_uniform: |
| case OMPC_use_device_ptr: |
| case OMPC_use_device_addr: |
| case OMPC_is_device_ptr: |
| case OMPC_unified_address: |
| case OMPC_unified_shared_memory: |
| case OMPC_reverse_offload: |
| case OMPC_dynamic_allocators: |
| case OMPC_match: |
| case OMPC_nontemporal: |
| case OMPC_destroy: |
| case OMPC_detach: |
| case OMPC_inclusive: |
| case OMPC_exclusive: |
| case OMPC_uses_allocators: |
| case OMPC_affinity: |
| break; |
| default: |
| break; |
| } |
| llvm_unreachable("Invalid OpenMP simple clause kind"); |
| } |
| |
| bool clang::isOpenMPLoopDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_simd || DKind == OMPD_for || DKind == OMPD_for_simd || |
| DKind == OMPD_parallel_for || DKind == OMPD_parallel_for_simd || |
| DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd || |
| DKind == OMPD_master_taskloop || DKind == OMPD_master_taskloop_simd || |
| DKind == OMPD_parallel_master_taskloop || |
| DKind == OMPD_parallel_master_taskloop_simd || |
| DKind == OMPD_distribute || DKind == OMPD_target_parallel_for || |
| DKind == OMPD_distribute_parallel_for || |
| DKind == OMPD_distribute_parallel_for_simd || |
| DKind == OMPD_distribute_simd || |
| DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd || |
| DKind == OMPD_teams_distribute || |
| DKind == OMPD_teams_distribute_simd || |
| DKind == OMPD_teams_distribute_parallel_for_simd || |
| DKind == OMPD_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute || |
| DKind == OMPD_target_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_simd; |
| } |
| |
| bool clang::isOpenMPWorksharingDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_for || DKind == OMPD_for_simd || |
| DKind == OMPD_sections || DKind == OMPD_section || |
| DKind == OMPD_single || DKind == OMPD_parallel_for || |
| DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections || |
| DKind == OMPD_target_parallel_for || |
| DKind == OMPD_distribute_parallel_for || |
| DKind == OMPD_distribute_parallel_for_simd || |
| DKind == OMPD_target_parallel_for_simd || |
| DKind == OMPD_teams_distribute_parallel_for_simd || |
| DKind == OMPD_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd; |
| } |
| |
| bool clang::isOpenMPTaskLoopDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_taskloop || DKind == OMPD_taskloop_simd || |
| DKind == OMPD_master_taskloop || DKind == OMPD_master_taskloop_simd || |
| DKind == OMPD_parallel_master_taskloop || |
| DKind == OMPD_parallel_master_taskloop_simd; |
| } |
| |
| bool clang::isOpenMPParallelDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_parallel || DKind == OMPD_parallel_for || |
| DKind == OMPD_parallel_for_simd || DKind == OMPD_parallel_sections || |
| DKind == OMPD_target_parallel || DKind == OMPD_target_parallel_for || |
| DKind == OMPD_distribute_parallel_for || |
| DKind == OMPD_distribute_parallel_for_simd || |
| DKind == OMPD_target_parallel_for_simd || |
| DKind == OMPD_teams_distribute_parallel_for || |
| DKind == OMPD_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd || |
| DKind == OMPD_parallel_master || |
| DKind == OMPD_parallel_master_taskloop || |
| DKind == OMPD_parallel_master_taskloop_simd; |
| } |
| |
| bool clang::isOpenMPTargetExecutionDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_target || DKind == OMPD_target_parallel || |
| DKind == OMPD_target_parallel_for || |
| DKind == OMPD_target_parallel_for_simd || DKind == OMPD_target_simd || |
| DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute || |
| DKind == OMPD_target_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_simd; |
| } |
| |
| bool clang::isOpenMPTargetDataManagementDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_target_data || DKind == OMPD_target_enter_data || |
| DKind == OMPD_target_exit_data || DKind == OMPD_target_update; |
| } |
| |
| bool clang::isOpenMPNestingTeamsDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_teams || DKind == OMPD_teams_distribute || |
| DKind == OMPD_teams_distribute_simd || |
| DKind == OMPD_teams_distribute_parallel_for_simd || |
| DKind == OMPD_teams_distribute_parallel_for; |
| } |
| |
| bool clang::isOpenMPTeamsDirective(OpenMPDirectiveKind DKind) { |
| return isOpenMPNestingTeamsDirective(DKind) || |
| DKind == OMPD_target_teams || DKind == OMPD_target_teams_distribute || |
| DKind == OMPD_target_teams_distribute_parallel_for || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_simd; |
| } |
| |
| bool clang::isOpenMPSimdDirective(OpenMPDirectiveKind DKind) { |
| return DKind == OMPD_simd || DKind == OMPD_for_simd || |
| DKind == OMPD_parallel_for_simd || DKind == OMPD_taskloop_simd || |
| DKind == OMPD_master_taskloop_simd || |
| DKind == OMPD_parallel_master_taskloop_simd || |
| DKind == OMPD_distribute_parallel_for_simd || |
| DKind == OMPD_distribute_simd || DKind == OMPD_target_simd || |
| DKind == OMPD_teams_distribute_simd || |
| DKind == OMPD_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_parallel_for_simd || |
| DKind == OMPD_target_teams_distribute_simd || |
| DKind == OMPD_target_parallel_for_simd; |
| } |
| |
| bool clang::isOpenMPNestingDistributeDirective(OpenMPDirectiveKind Kind) { |
| return Kind == OMPD_distribute || Kind == OMPD_distribute_parallel_for || |
| Kind == OMPD_distribute_parallel_for_simd || |
| Kind == OMPD_distribute_simd; |
| // TODO add next directives. |
| } |
| |
| bool clang::isOpenMPDistributeDirective(OpenMPDirectiveKind Kind) { |
| return isOpenMPNestingDistributeDirective(Kind) || |
| Kind == OMPD_teams_distribute || Kind == OMPD_teams_distribute_simd || |
| Kind == OMPD_teams_distribute_parallel_for_simd || |
| Kind == OMPD_teams_distribute_parallel_for || |
| Kind == OMPD_target_teams_distribute || |
| Kind == OMPD_target_teams_distribute_parallel_for || |
| Kind == OMPD_target_teams_distribute_parallel_for_simd || |
| Kind == OMPD_target_teams_distribute_simd; |
| } |
| |
| bool clang::isOpenMPPrivate(OpenMPClauseKind Kind) { |
| return Kind == OMPC_private || Kind == OMPC_firstprivate || |
| Kind == OMPC_lastprivate || Kind == OMPC_linear || |
| Kind == OMPC_reduction || Kind == OMPC_task_reduction || |
| Kind == OMPC_in_reduction; // TODO add next clauses like 'reduction'. |
| } |
| |
| bool clang::isOpenMPThreadPrivate(OpenMPClauseKind Kind) { |
| return Kind == OMPC_threadprivate || Kind == OMPC_copyin; |
| } |
| |
| bool clang::isOpenMPTaskingDirective(OpenMPDirectiveKind Kind) { |
| return Kind == OMPD_task || isOpenMPTaskLoopDirective(Kind); |
| } |
| |
| bool clang::isOpenMPLoopBoundSharingDirective(OpenMPDirectiveKind Kind) { |
| return Kind == OMPD_distribute_parallel_for || |
| Kind == OMPD_distribute_parallel_for_simd || |
| Kind == OMPD_teams_distribute_parallel_for_simd || |
| Kind == OMPD_teams_distribute_parallel_for || |
| Kind == OMPD_target_teams_distribute_parallel_for || |
| Kind == OMPD_target_teams_distribute_parallel_for_simd; |
| } |
| |
| void clang::getOpenMPCaptureRegions( |
| SmallVectorImpl<OpenMPDirectiveKind> &CaptureRegions, |
| OpenMPDirectiveKind DKind) { |
| assert(unsigned(DKind) < llvm::omp::Directive_enumSize); |
| switch (DKind) { |
| case OMPD_parallel: |
| case OMPD_parallel_for: |
| case OMPD_parallel_for_simd: |
| case OMPD_parallel_master: |
| case OMPD_parallel_sections: |
| case OMPD_distribute_parallel_for: |
| case OMPD_distribute_parallel_for_simd: |
| CaptureRegions.push_back(OMPD_parallel); |
| break; |
| case OMPD_target_teams: |
| case OMPD_target_teams_distribute: |
| case OMPD_target_teams_distribute_simd: |
| CaptureRegions.push_back(OMPD_task); |
| CaptureRegions.push_back(OMPD_target); |
| CaptureRegions.push_back(OMPD_teams); |
| break; |
| case OMPD_teams: |
| case OMPD_teams_distribute: |
| case OMPD_teams_distribute_simd: |
| CaptureRegions.push_back(OMPD_teams); |
| break; |
| case OMPD_target: |
| case OMPD_target_simd: |
| CaptureRegions.push_back(OMPD_task); |
| CaptureRegions.push_back(OMPD_target); |
| break; |
| case OMPD_teams_distribute_parallel_for: |
| case OMPD_teams_distribute_parallel_for_simd: |
| CaptureRegions.push_back(OMPD_teams); |
| CaptureRegions.push_back(OMPD_parallel); |
| break; |
| case OMPD_target_parallel: |
| case OMPD_target_parallel_for: |
| case OMPD_target_parallel_for_simd: |
| CaptureRegions.push_back(OMPD_task); |
| CaptureRegions.push_back(OMPD_target); |
| CaptureRegions.push_back(OMPD_parallel); |
| break; |
| case OMPD_task: |
| case OMPD_target_enter_data: |
| case OMPD_target_exit_data: |
| case OMPD_target_update: |
| CaptureRegions.push_back(OMPD_task); |
| break; |
| case OMPD_taskloop: |
| case OMPD_taskloop_simd: |
| case OMPD_master_taskloop: |
| case OMPD_master_taskloop_simd: |
| CaptureRegions.push_back(OMPD_taskloop); |
| break; |
| case OMPD_parallel_master_taskloop: |
| case OMPD_parallel_master_taskloop_simd: |
| CaptureRegions.push_back(OMPD_parallel); |
| CaptureRegions.push_back(OMPD_taskloop); |
| break; |
| case OMPD_target_teams_distribute_parallel_for: |
| case OMPD_target_teams_distribute_parallel_for_simd: |
| CaptureRegions.push_back(OMPD_task); |
| CaptureRegions.push_back(OMPD_target); |
| CaptureRegions.push_back(OMPD_teams); |
| CaptureRegions.push_back(OMPD_parallel); |
| break; |
| case OMPD_simd: |
| case OMPD_for: |
| case OMPD_for_simd: |
| case OMPD_sections: |
| case OMPD_section: |
| case OMPD_single: |
| case OMPD_master: |
| case OMPD_critical: |
| case OMPD_taskgroup: |
| case OMPD_distribute: |
| case OMPD_ordered: |
| case OMPD_atomic: |
| case OMPD_target_data: |
| case OMPD_distribute_simd: |
| CaptureRegions.push_back(OMPD_unknown); |
| break; |
| case OMPD_threadprivate: |
| case OMPD_allocate: |
| case OMPD_taskyield: |
| case OMPD_barrier: |
| case OMPD_taskwait: |
| case OMPD_cancellation_point: |
| case OMPD_cancel: |
| case OMPD_flush: |
| case OMPD_depobj: |
| case OMPD_scan: |
| case OMPD_declare_reduction: |
| case OMPD_declare_mapper: |
| case OMPD_declare_simd: |
| case OMPD_declare_target: |
| case OMPD_end_declare_target: |
| case OMPD_requires: |
| case OMPD_declare_variant: |
| case OMPD_begin_declare_variant: |
| case OMPD_end_declare_variant: |
| llvm_unreachable("OpenMP Directive is not allowed"); |
| case OMPD_unknown: |
| default: |
| llvm_unreachable("Unknown OpenMP directive"); |
| } |
| } |