154 lines · c
1// clang-format off2// RUN: %libomp-compile-and-run | FileCheck %s3// REQUIRES: ompt4// clang-format on5 6#define _BSD_SOURCE7#define _DEFAULT_SOURCE8 9#include <stdio.h>10#include <inttypes.h>11#include <omp.h>12#include <omp-tools.h>13 14static ompt_set_callback_t ompt_set_callback;15static ompt_get_callback_t ompt_get_callback;16static ompt_get_state_t ompt_get_state;17static ompt_get_task_info_t ompt_get_task_info;18static ompt_get_thread_data_t ompt_get_thread_data;19static ompt_get_parallel_info_t ompt_get_parallel_info;20static ompt_get_unique_id_t ompt_get_unique_id;21static ompt_get_num_procs_t ompt_get_num_procs;22static ompt_get_num_places_t ompt_get_num_places;23static ompt_get_place_proc_ids_t ompt_get_place_proc_ids;24static ompt_get_place_num_t ompt_get_place_num;25static ompt_get_partition_place_nums_t ompt_get_partition_place_nums;26static ompt_get_proc_id_t ompt_get_proc_id;27static ompt_enumerate_states_t ompt_enumerate_states;28static ompt_enumerate_mutex_impls_t ompt_enumerate_mutex_impls;29 30int main() {31 // Call OpenMP API function to force initialization of OMPT.32 // (omp_get_thread_num() does not work because it just returns 0 if the33 // runtime isn't initialized yet...)34 omp_get_num_threads();35 36 ompt_data_t *tdata = ompt_get_thread_data();37 uint64_t tvalue = tdata ? tdata->value : 0;38 39 printf("%" PRIu64 ": ompt_get_num_places()=%d\n", tvalue,40 ompt_get_num_places());41 42 printf("%" PRIu64 ": ompt_get_place_proc_ids()=%d\n", tvalue,43 ompt_get_place_proc_ids(0, 0, NULL));44 45 printf("%" PRIu64 ": ompt_get_place_num()=%d\n", tvalue,46 ompt_get_place_num());47 48 printf("%" PRIu64 ": ompt_get_partition_place_nums()=%d\n", tvalue,49 ompt_get_partition_place_nums(0, NULL));50 51 printf("%" PRIu64 ": ompt_get_proc_id()=%d\n", tvalue, ompt_get_proc_id());52 53 printf("%" PRIu64 ": ompt_get_num_procs()=%d\n", tvalue,54 ompt_get_num_procs());55 56 ompt_callback_t callback;57 printf("%" PRIu64 ": ompt_get_callback()=%d\n", tvalue,58 ompt_get_callback(ompt_callback_thread_begin, &callback));59 60 printf("%" PRIu64 ": ompt_get_state()=%d\n", tvalue, ompt_get_state(NULL));61 62 int state = ompt_state_undefined;63 const char *state_name;64 printf("%" PRIu64 ": ompt_enumerate_states()=%d\n", tvalue,65 ompt_enumerate_states(state, &state, &state_name));66 67 int impl = ompt_mutex_impl_none;68 const char *impl_name;69 printf("%" PRIu64 ": ompt_enumerate_mutex_impls()=%d\n", tvalue,70 ompt_enumerate_mutex_impls(impl, &impl, &impl_name));71 72 printf("%" PRIu64 ": ompt_get_thread_data()=%p\n", tvalue,73 ompt_get_thread_data());74 75 printf("%" PRIu64 ": ompt_get_parallel_info()=%d\n", tvalue,76 ompt_get_parallel_info(0, NULL, NULL));77 78 printf("%" PRIu64 ": ompt_get_task_info()=%d\n", tvalue,79 ompt_get_task_info(0, NULL, NULL, NULL, NULL, NULL));80 81 // clang-format off82 // Check if libomp supports the callbacks for this test.83 84 // CHECK: 0: NULL_POINTER=[[NULL:.*$]]85 86 // CHECK: {{^}}[[MASTER_ID:[0-9]+]]: ompt_get_num_places()={{[0-9]+}}87 88 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_place_proc_ids()={{[0-9]+}}89 90 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_place_num()=-191 92 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_partition_place_nums()=093 94 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_proc_id()=-195 96 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_num_procs()={{[0-9]+}}97 98 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_callback()=099 100 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_state()=0101 102 // CHECK: {{^}}[[MASTER_ID]]: ompt_enumerate_states()=1103 104 // CHECK: {{^}}[[MASTER_ID]]: ompt_enumerate_mutex_impls()=1105 106 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_thread_data()=[[NULL]]107 108 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_parallel_info()=0109 110 // CHECK: {{^}}[[MASTER_ID]]: ompt_get_task_info()=0111 // clang-format on112 113 return 0;114}115 116int ompt_initialize(ompt_function_lookup_t lookup, int initial_device_num,117 ompt_data_t *tool_data) {118 ompt_set_callback = (ompt_set_callback_t)lookup("ompt_set_callback");119 ompt_get_callback = (ompt_get_callback_t)lookup("ompt_get_callback");120 ompt_get_state = (ompt_get_state_t)lookup("ompt_get_state");121 ompt_get_task_info = (ompt_get_task_info_t)lookup("ompt_get_task_info");122 ompt_get_thread_data = (ompt_get_thread_data_t)lookup("ompt_get_thread_data");123 ompt_get_parallel_info =124 (ompt_get_parallel_info_t)lookup("ompt_get_parallel_info");125 ompt_get_unique_id = (ompt_get_unique_id_t)lookup("ompt_get_unique_id");126 127 ompt_get_num_procs = (ompt_get_num_procs_t)lookup("ompt_get_num_procs");128 ompt_get_num_places = (ompt_get_num_places_t)lookup("ompt_get_num_places");129 ompt_get_place_proc_ids =130 (ompt_get_place_proc_ids_t)lookup("ompt_get_place_proc_ids");131 ompt_get_place_num = (ompt_get_place_num_t)lookup("ompt_get_place_num");132 ompt_get_partition_place_nums =133 (ompt_get_partition_place_nums_t)lookup("ompt_get_partition_place_nums");134 ompt_get_proc_id = (ompt_get_proc_id_t)lookup("ompt_get_proc_id");135 ompt_enumerate_states =136 (ompt_enumerate_states_t)lookup("ompt_enumerate_states");137 ompt_enumerate_mutex_impls =138 (ompt_enumerate_mutex_impls_t)lookup("ompt_enumerate_mutex_impls");139 140 printf("0: NULL_POINTER=%p\n", (void *)NULL);141 return 0; // no success -> OMPT not enabled142}143 144void ompt_finalize(ompt_data_t *tool_data) {145 printf("0: ompt_event_runtime_shutdown\n");146}147 148ompt_start_tool_result_t *ompt_start_tool(unsigned int omp_version,149 const char *runtime_version) {150 static ompt_start_tool_result_t ompt_start_tool_result = {&ompt_initialize,151 &ompt_finalize, 0};152 return &ompt_start_tool_result;153}154