diff options
Diffstat (limited to 'final/runtime/src/ompt-specific.h')
-rw-r--r-- | final/runtime/src/ompt-specific.h | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/final/runtime/src/ompt-specific.h b/final/runtime/src/ompt-specific.h new file mode 100644 index 0000000..317580f --- /dev/null +++ b/final/runtime/src/ompt-specific.h @@ -0,0 +1,104 @@ +/* + * ompt-specific.h - header of OMPT internal functions implementation + */ + +//===----------------------------------------------------------------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is dual licensed under the MIT and the University of Illinois Open +// Source Licenses. See LICENSE.txt for details. +// +//===----------------------------------------------------------------------===// + +#ifndef OMPT_SPECIFIC_H +#define OMPT_SPECIFIC_H + +#include "kmp.h" + +/***************************************************************************** + * forward declarations + ****************************************************************************/ + +void __ompt_team_assign_id(kmp_team_t *team, ompt_data_t ompt_pid); +void __ompt_thread_assign_wait_id(void *variable); + +void __ompt_lw_taskteam_init(ompt_lw_taskteam_t *lwt, kmp_info_t *thr, + int gtid, ompt_data_t *ompt_pid, void *codeptr); + +void __ompt_lw_taskteam_link(ompt_lw_taskteam_t *lwt, kmp_info_t *thr, + int on_heap); + +void __ompt_lw_taskteam_unlink(kmp_info_t *thr); + +ompt_team_info_t *__ompt_get_teaminfo(int depth, int *size); + +ompt_task_info_t *__ompt_get_task_info_object(int depth); + +int __ompt_get_parallel_info_internal(int ancestor_level, + ompt_data_t **parallel_data, + int *team_size); + +int __ompt_get_task_info_internal(int ancestor_level, int *type, + ompt_data_t **task_data, + ompt_frame_t **task_frame, + ompt_data_t **parallel_data, int *thread_num); + +ompt_data_t *__ompt_get_thread_data_internal(); + +/* + * Unused currently +static uint64_t __ompt_get_get_unique_id_internal(); +*/ + +/***************************************************************************** + * macros + ****************************************************************************/ + +#define OMPT_CUR_TASK_INFO(thr) (&(thr->th.th_current_task->ompt_task_info)) +#define OMPT_CUR_TASK_DATA(thr) \ + (&(thr->th.th_current_task->ompt_task_info.task_data)) +#define OMPT_CUR_TEAM_INFO(thr) (&(thr->th.th_team->t.ompt_team_info)) +#define OMPT_CUR_TEAM_DATA(thr) \ + (&(thr->th.th_team->t.ompt_team_info.parallel_data)) + +#define OMPT_HAVE_WEAK_ATTRIBUTE KMP_HAVE_WEAK_ATTRIBUTE +#define OMPT_HAVE_PSAPI KMP_HAVE_PSAPI +#define OMPT_STR_MATCH(haystack, needle) __kmp_str_match(haystack, 0, needle) + +inline void *__ompt_load_return_address(int gtid) { + kmp_info_t *thr = __kmp_threads[gtid]; + void *return_address = thr->th.ompt_thread_info.return_address; + thr->th.ompt_thread_info.return_address = NULL; + return return_address; +} + +#define OMPT_STORE_RETURN_ADDRESS(gtid) \ + if (ompt_enabled.enabled && gtid >= 0 && __kmp_threads[gtid] && \ + !__kmp_threads[gtid]->th.ompt_thread_info.return_address) \ + __kmp_threads[gtid]->th.ompt_thread_info.return_address = \ + __builtin_return_address(0) +#define OMPT_LOAD_RETURN_ADDRESS(gtid) __ompt_load_return_address(gtid) + +//****************************************************************************** +// inline functions +//****************************************************************************** + +inline kmp_info_t *ompt_get_thread_gtid(int gtid) { + return (gtid >= 0) ? __kmp_thread_from_gtid(gtid) : NULL; +} + +inline kmp_info_t *ompt_get_thread() { + int gtid = __kmp_get_gtid(); + return ompt_get_thread_gtid(gtid); +} + +inline void ompt_set_thread_state(kmp_info_t *thread, ompt_state_t state) { + thread->th.ompt_thread_info.state = state; +} + +inline const char *ompt_get_runtime_version() { + return &__kmp_version_lib_ver[KMP_VERSION_MAGIC_LEN]; +} + +#endif |