1 #include <omp-tools.h>
2
3 // From openmp/runtime/test/ompt/callback.h
4 #define register_ompt_callback_t(name, type) \
5 do { \
6 type f_##name = &on_##name; \
7 if (ompt_set_callback(name, (ompt_callback_t)f_##name) == ompt_set_never) \
8 printf("0: Could not register callback '" #name "'\n"); \
9 } while (0)
10
11 #define register_ompt_callback(name) register_ompt_callback_t(name, name##_t)
12
13 // OMPT entry point handles
14 static ompt_set_callback_t ompt_set_callback = 0;
15
16 // Init functions
ompt_initialize(ompt_function_lookup_t lookup,int initial_device_num,ompt_data_t * tool_data)17 int ompt_initialize(ompt_function_lookup_t lookup, int initial_device_num,
18 ompt_data_t *tool_data) {
19 ompt_set_callback = (ompt_set_callback_t)lookup("ompt_set_callback");
20
21 if (!ompt_set_callback)
22 return 0; // failed
23
24 register_ompt_callback(ompt_callback_device_initialize);
25 register_ompt_callback(ompt_callback_device_finalize);
26 register_ompt_callback(ompt_callback_device_load);
27 register_ompt_callback(ompt_callback_target_data_op_emi);
28 register_ompt_callback(ompt_callback_target_data_op);
29 register_ompt_callback(ompt_callback_target);
30 register_ompt_callback(ompt_callback_target_emi);
31 register_ompt_callback(ompt_callback_target_submit);
32
33 return 1; // success
34 }
35
ompt_finalize(ompt_data_t * tool_data)36 void ompt_finalize(ompt_data_t *tool_data) {}
37
38 #ifdef __cplusplus
39 extern "C" {
40 #endif
ompt_start_tool(unsigned int omp_version,const char * runtime_version)41 ompt_start_tool_result_t *ompt_start_tool(unsigned int omp_version,
42 const char *runtime_version) {
43 static ompt_start_tool_result_t ompt_start_tool_result = {&ompt_initialize,
44 &ompt_finalize, 0};
45 return &ompt_start_tool_result;
46 }
47 #ifdef __cplusplus
48 }
49 #endif
50