1 /* 2 * ompt-internal.h - header of OMPT internal data structures 3 */ 4 5 //===----------------------------------------------------------------------===// 6 // 7 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 8 // See https://llvm.org/LICENSE.txt for license information. 9 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef __OMPT_INTERNAL_H__ 14 #define __OMPT_INTERNAL_H__ 15 16 #include "ompt-event-specific.h" 17 #include "omp-tools.h" 18 19 #define OMPT_VERSION 1 20 21 #define _OMP_EXTERN extern "C" 22 23 #define OMPT_INVOKER(x) \ 24 ((x == fork_context_gnu) ? ompt_parallel_invoker_program \ 25 : ompt_parallel_invoker_runtime) 26 27 #define ompt_callback(e) e##_callback 28 29 typedef struct ompt_callbacks_internal_s { 30 #define ompt_event_macro(event, callback, eventid) \ 31 callback ompt_callback(event); 32 33 FOREACH_OMPT_EVENT(ompt_event_macro) 34 35 #undef ompt_event_macro 36 } ompt_callbacks_internal_t; 37 38 typedef struct ompt_callbacks_active_s { 39 unsigned int enabled : 1; 40 #define ompt_event_macro(event, callback, eventid) unsigned int event : 1; 41 42 FOREACH_OMPT_EVENT(ompt_event_macro) 43 44 #undef ompt_event_macro 45 } ompt_callbacks_active_t; 46 47 #define TASK_TYPE_DETAILS_FORMAT(info) \ 48 ((info->td_flags.task_serial || info->td_flags.tasking_ser) \ 49 ? ompt_task_undeferred \ 50 : 0x0) | \ 51 ((!(info->td_flags.tiedness)) ? ompt_task_untied : 0x0) | \ 52 (info->td_flags.final ? ompt_task_final : 0x0) | \ 53 (info->td_flags.merged_if0 ? ompt_task_mergeable : 0x0) 54 55 typedef struct { 56 ompt_frame_t frame; 57 ompt_data_t task_data; 58 struct kmp_taskdata *scheduling_parent; 59 int thread_num; 60 ompt_dispatch_chunk_t dispatch_chunk; 61 } ompt_task_info_t; 62 63 typedef struct { 64 ompt_data_t parallel_data; 65 void *master_return_address; 66 } ompt_team_info_t; 67 68 typedef struct ompt_lw_taskteam_s { 69 ompt_team_info_t ompt_team_info; 70 ompt_task_info_t ompt_task_info; 71 int heap; 72 struct ompt_lw_taskteam_s *parent; 73 } ompt_lw_taskteam_t; 74 75 typedef struct { 76 ompt_data_t thread_data; 77 ompt_data_t task_data; /* stored here from implicit barrier-begin until 78 implicit-task-end */ 79 void *return_address; /* stored here on entry of runtime */ 80 ompt_state_t state; 81 ompt_wait_id_t wait_id; 82 int ompt_task_yielded; 83 int parallel_flags; // information for the last parallel region invoked 84 void *idle_frame; 85 } ompt_thread_info_t; 86 87 extern ompt_callbacks_internal_t ompt_callbacks; 88 89 #if OMPT_SUPPORT && OMPT_OPTIONAL 90 #if USE_FAST_MEMORY 91 #define KMP_OMPT_DEPS_ALLOC __kmp_fast_allocate 92 #define KMP_OMPT_DEPS_FREE __kmp_fast_free 93 #else 94 #define KMP_OMPT_DEPS_ALLOC __kmp_thread_malloc 95 #define KMP_OMPT_DEPS_FREE __kmp_thread_free 96 #endif 97 #endif /* OMPT_SUPPORT && OMPT_OPTIONAL */ 98 99 #ifdef __cplusplus 100 extern "C" { 101 #endif 102 103 void ompt_pre_init(void); 104 void ompt_post_init(void); 105 void ompt_fini(void); 106 107 #define OMPT_GET_RETURN_ADDRESS(level) __builtin_return_address(level) 108 #define OMPT_GET_FRAME_ADDRESS(level) __builtin_frame_address(level) 109 110 int __kmp_control_tool(uint64_t command, uint64_t modifier, void *arg); 111 112 extern ompt_callbacks_active_t ompt_enabled; 113 114 #if KMP_OS_WINDOWS 115 #define UNLIKELY(x) (x) 116 #define OMPT_NOINLINE __declspec(noinline) 117 #else 118 #define UNLIKELY(x) __builtin_expect(!!(x), 0) 119 #define OMPT_NOINLINE __attribute__((noinline)) 120 #endif 121 122 #ifdef __cplusplus 123 } 124 #endif 125 126 #endif 127