openmp/runtime/test/ompt/synchronization/master.c

   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(-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]]