xref: /llvm-project/openmp/runtime/test/ompt/tasks/explicit_task_thread_num.c (revision d6f00654fbc686f5413f9c8d1bac2c4050fdf350)
1 // RUN: %libomp-compile-and-run | FileCheck %s
2 // REQUIRES: ompt
3 
4 #include "callback.h"
5 #include <omp.h>
6 
7 
8 __attribute__ ((noinline)) // workaround for bug in icc
print_task_info_at(int ancestor_level,int id)9 void print_task_info_at(int ancestor_level, int id)
10 {
11 #pragma omp critical
12   {
13     int task_type;
14     char buffer[2048];
15     ompt_data_t *parallel_data;
16     ompt_data_t *task_data;
17     int thread_num;
18     ompt_get_task_info(ancestor_level, &task_type, &task_data, NULL,
19                        &parallel_data, &thread_num);
20     format_task_type(task_type, buffer);
21     printf("%" PRIu64 ": ancestor_level=%d id=%d task_type=%s=%d "
22                       "parallel_id=%" PRIu64 " task_id=%" PRIu64
23                       " thread_num=%d\n",
24         ompt_get_thread_data()->value, ancestor_level, id, buffer,
25         task_type, parallel_data->value, task_data->value, thread_num);
26   }
27 };
28 
main()29 int main()
30 {
31 
32 #pragma omp parallel num_threads(2)
33   {
34 
35     if (omp_get_thread_num() == 1) {
36       // To assert that task is executed by the worker thread,
37       // if(0) is used in order to ensure that the task is immediately
38       // executed after its creation.
39 #pragma omp task if(0)
40       {
41         // thread_num should be equal to 1 for both explicit and implicit task
42         print_task_info_at(0, 1);
43         print_task_info_at(1, 0);
44       };
45     }
46   }
47 
48   // Check if libomp supports the callbacks for this test.
49   // CHECK-NOT: {{^}}0: Could not register callback 'ompt_event_parallel_begin'
50   // CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_task_create'
51   // CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_implicit_task'
52 
53   // CHECK: {{^}}0: NULL_POINTER=[[NULL:.*$]]
54   // CHECK: {{^}}[[MASTER_ID:[0-9]+]]: ompt_event_initial_task_begin
55 
56   // parallel region used only to determine worker thread id
57   // CHECK: {{^}}[[MASTER_ID]]: ompt_event_parallel_begin
58   // CHECK: {{^}}[[WID:[0-9]+]]: ompt_event_implicit_task{{.*}}thread_num=1
59 
60   // thread_num must be equal to 1 for both explicit and the implicit tasks
61   // CHECK: {{^}}[[WID]]: ancestor_level=0 id=1 task_type=ompt_task_explicit
62   // CHECK-SAME: thread_num=1
63   // CHECK: {{^}}[[WID]]: ancestor_level=1 id=0 task_type=ompt_task_implicit
64   // CHECK-SAME: thread_num=1
65 
66   return 0;
67 }
68