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