1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#include <stdio.h>
#include <inttypes.h>
#include <ompt.h>
static ompt_get_task_id_t ompt_get_task_id;
static ompt_get_thread_id_t ompt_get_thread_id;
static ompt_get_parallel_id_t ompt_get_parallel_id;
static void print_ids(int level)
{
printf("%" PRIu64 ": level %d: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), level, ompt_get_parallel_id(level), ompt_get_task_id(level));
}
static void
on_ompt_event_barrier_begin(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id)
{
printf("%" PRIu64 ": ompt_event_barrier_begin: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), parallel_id, task_id);
}
static void
on_ompt_event_barrier_end(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id)
{
printf("%" PRIu64 ": ompt_event_barrier_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), parallel_id, task_id);
}
static void
on_ompt_event_implicit_task_begin(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id)
{
printf("%" PRIu64 ": ompt_event_implicit_task_begin: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), parallel_id, task_id);
}
static void
on_ompt_event_implicit_task_end(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id)
{
printf("%" PRIu64 ": ompt_event_implicit_task_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), parallel_id, task_id);
}
static void
on_ompt_event_loop_begin(
ompt_parallel_id_t parallel_id,
ompt_task_id_t parent_task_id,
void *workshare_function)
{
printf("%" PRIu64 ": ompt_event_loop_begin: parallel_id=%" PRIu64 ", parent_task_id=%" PRIu64 ", workshare_function=%p\n", ompt_get_thread_id(), parallel_id, parent_task_id, workshare_function);
}
static void
on_ompt_event_loop_end(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id)
{
printf("%" PRIu64 ": ompt_event_loop_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 "\n", ompt_get_thread_id(), parallel_id, task_id);
}
static void
on_ompt_event_parallel_begin(
ompt_task_id_t parent_task_id,
ompt_frame_t *parent_task_frame,
ompt_parallel_id_t parallel_id,
uint32_t requested_team_size,
void *parallel_function,
ompt_invoker_t invoker)
{
printf("%" PRIu64 ": ompt_event_parallel_begin: parent_task_id=%" PRIu64 ", parent_task_frame=%p, parallel_id=%" PRIu64 ", requested_team_size=%" PRIu32 ", parallel_function=%p, invoker=%d\n", ompt_get_thread_id(), parent_task_id, parent_task_frame, parallel_id, requested_team_size, parallel_function, invoker);
}
static void
on_ompt_event_parallel_end(
ompt_parallel_id_t parallel_id,
ompt_task_id_t task_id,
ompt_invoker_t invoker)
{
printf("%" PRIu64 ": ompt_event_parallel_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 ", invoker=%d\n", ompt_get_thread_id(), parallel_id, task_id, invoker);
}
void ompt_initialize(
ompt_function_lookup_t lookup,
const char *runtime_version,
unsigned int ompt_version)
{
ompt_set_callback_t ompt_set_callback = (ompt_set_callback_t) lookup("ompt_set_callback");
ompt_get_task_id = (ompt_get_task_id_t) lookup("ompt_get_task_id");
ompt_get_thread_id = (ompt_get_thread_id_t) lookup("ompt_get_thread_id");
ompt_get_parallel_id = (ompt_get_parallel_id_t) lookup("ompt_get_parallel_id");
ompt_set_callback(ompt_event_barrier_begin, (ompt_callback_t) &on_ompt_event_barrier_begin);
ompt_set_callback(ompt_event_barrier_end, (ompt_callback_t) &on_ompt_event_barrier_end);
ompt_set_callback(ompt_event_implicit_task_begin, (ompt_callback_t) &on_ompt_event_implicit_task_begin);
ompt_set_callback(ompt_event_implicit_task_end, (ompt_callback_t) &on_ompt_event_implicit_task_end);
ompt_set_callback(ompt_event_loop_begin, (ompt_callback_t) &on_ompt_event_loop_begin);
ompt_set_callback(ompt_event_loop_end, (ompt_callback_t) &on_ompt_event_loop_end);
ompt_set_callback(ompt_event_parallel_begin, (ompt_callback_t) &on_ompt_event_parallel_begin);
ompt_set_callback(ompt_event_parallel_end, (ompt_callback_t) &on_ompt_event_parallel_end);
}
ompt_initialize_t ompt_tool()
{
return &ompt_initialize;
}
|
