| 1 | // RUN: %libomp-compile-and-run | FileCheck %s |
|---|
| 2 | // REQUIRES: ompt |
|---|
| 3 | // GCC generates code that does not call the runtime for the master construct |
|---|
| 4 | // XFAIL: gcc |
|---|
| 5 | |
|---|
| 6 | #define USE_PRIVATE_TOOL 1 |
|---|
| 7 | #include "callback.h" |
|---|
| 8 | #include <omp.h> |
|---|
| 9 | |
|---|
| 10 | int main() { |
|---|
| 11 | int x = 0; |
|---|
| 12 | #pragma omp parallel num_threads(2) |
|---|
| 13 | { |
|---|
| 14 | #pragma omp master |
|---|
| 15 | { |
|---|
| 16 | print_fuzzy_address(1); |
|---|
| 17 | x++; |
|---|
| 18 | } |
|---|
| 19 | print_current_address(2); |
|---|
| 20 | } |
|---|
| 21 | |
|---|
| 22 | printf("%" PRIu64 ": x=%d\n" , ompt_get_thread_data()->value, x); |
|---|
| 23 | |
|---|
| 24 | return 0; |
|---|
| 25 | } |
|---|
| 26 | |
|---|
| 27 | static void on_ompt_callback_master(ompt_scope_endpoint_t endpoint, |
|---|
| 28 | ompt_data_t *parallel_data, |
|---|
| 29 | ompt_data_t *task_data, |
|---|
| 30 | const void *codeptr_ra) { |
|---|
| 31 | switch (endpoint) { |
|---|
| 32 | case ompt_scope_begin: |
|---|
| 33 | printf("%" PRIu64 ":" _TOOL_PREFIX |
|---|
| 34 | " ompt_event_master_begin: codeptr_ra=%p\n" , |
|---|
| 35 | ompt_get_thread_data()->value, codeptr_ra); |
|---|
| 36 | break; |
|---|
| 37 | case ompt_scope_end: |
|---|
| 38 | printf("%" PRIu64 ":" _TOOL_PREFIX |
|---|
| 39 | " ompt_event_master_end: codeptr_ra=%p\n" , |
|---|
| 40 | ompt_get_thread_data()->value, codeptr_ra); |
|---|
| 41 | break; |
|---|
| 42 | case ompt_scope_beginend: |
|---|
| 43 | printf("ompt_scope_beginend should never be passed to %s\n" , __func__); |
|---|
| 44 | exit(status: -1); |
|---|
| 45 | } |
|---|
| 46 | } |
|---|
| 47 | |
|---|
| 48 | static void on_ompt_callback_thread_begin(ompt_thread_t thread_type, |
|---|
| 49 | ompt_data_t *thread_data) { |
|---|
| 50 | if (thread_data->ptr) |
|---|
| 51 | printf("%s\n" , "0: thread_data initially not null" ); |
|---|
| 52 | thread_data->value = ompt_get_unique_id(); |
|---|
| 53 | printf("%" PRIu64 ":" _TOOL_PREFIX |
|---|
| 54 | " ompt_event_thread_begin: thread_type=%s=%d, thread_id=%" PRIu64 "\n" , |
|---|
| 55 | ompt_get_thread_data()->value, ompt_thread_t_values[thread_type], |
|---|
| 56 | thread_type, thread_data->value); |
|---|
| 57 | } |
|---|
| 58 | |
|---|
| 59 | int ompt_initialize(ompt_function_lookup_t lookup, int initial_device_num, |
|---|
| 60 | ompt_data_t *tool_data) { |
|---|
| 61 | ompt_set_callback = (ompt_set_callback_t)lookup("ompt_set_callback" ); |
|---|
| 62 | ompt_get_unique_id = (ompt_get_unique_id_t)lookup("ompt_get_unique_id" ); |
|---|
| 63 | ompt_get_thread_data = (ompt_get_thread_data_t)lookup("ompt_get_thread_data" ); |
|---|
| 64 | |
|---|
| 65 | register_ompt_callback(ompt_callback_master); |
|---|
| 66 | printf("0: NULL_POINTER=%p\n" , (void *)NULL); |
|---|
| 67 | return 1; // success |
|---|
| 68 | } |
|---|
| 69 | |
|---|
| 70 | void ompt_finalize(ompt_data_t *tool_data) {} |
|---|
| 71 | |
|---|
| 72 | ompt_start_tool_result_t *ompt_start_tool(unsigned int omp_version, |
|---|
| 73 | const char *runtime_version) { |
|---|
| 74 | static ompt_start_tool_result_t ompt_start_tool_result = {&ompt_initialize, |
|---|
| 75 | &ompt_finalize, 0}; |
|---|
| 76 | return &ompt_start_tool_result; |
|---|
| 77 | } |
|---|
| 78 | |
|---|
| 79 | // Check if libomp supports the callbacks for this test. |
|---|
| 80 | // CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_master' |
|---|
| 81 | |
|---|
| 82 | // CHECK: 0: NULL_POINTER=[[NULL:.*$]] |
|---|
| 83 | |
|---|
| 84 | // CHECK: {{^}}[[MASTER_ID:[0-9]+]]: ompt_event_master_begin: |
|---|
| 85 | // CHECK-SAME: codeptr_ra=[[RETURN_ADDRESS:(0x)?[0-f]+]]{{[0-f][0-f]}} |
|---|
| 86 | // CHECK: {{^}}[[MASTER_ID]]: fuzzy_address={{.*}}[[RETURN_ADDRESS]] |
|---|
| 87 | // CHECK: {{^}}[[MASTER_ID]]: ompt_event_master_end: |
|---|
| 88 | // CHECK-SAME: codeptr_ra=[[RETURN_ADDRESS_END:(0x)?[0-f]+]] |
|---|
| 89 | // CHECK: {{^}}[[MASTER_ID]]: current_address={{.*}}[[RETURN_ADDRESS_END]] |
|---|
| 90 | |
|---|
