70 lines · c
1// clang-format off2// RUN: %libomp-compile-and-run | FileCheck %s3// REQUIRES: ompt4// clang-format on5 6#include "callback.h"7#include <omp.h>8 9__attribute__ ((noinline)) // workaround for bug in icc10void print_task_info_at(int ancestor_level, int id)11{12#pragma omp critical13 {14 int task_type;15char buffer[2048];16ompt_data_t *parallel_data;17ompt_data_t *task_data;18int thread_num;19ompt_get_task_info(ancestor_level, &task_type, &task_data, NULL, ¶llel_data,20 &thread_num);21format_task_type(task_type, buffer);22printf("%" PRIu64 ": ancestor_level=%d id=%d task_type=%s=%d "23 "parallel_id=%" PRIu64 " task_id=%" PRIu64 " thread_num=%d\n",24 ompt_get_thread_data()->value, ancestor_level, id, buffer, task_type,25 parallel_data->value, task_data->value, thread_num);26}27}28;29 30int main() {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 immediately38 // executed after its creation.39#pragma omp task if (0)40 {41 // thread_num should be equal to 1 for both explicit and implicit task42 print_task_info_at(0, 1);43 print_task_info_at(1, 0);44 };45 }46 }47 48 // clang-format off49 // Check if libomp supports the callbacks for this test.50 // CHECK-NOT: {{^}}0: Could not register callback 'ompt_event_parallel_begin'51 // CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_task_create'52 // CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_implicit_task'53 54 // CHECK: {{^}}0: NULL_POINTER=[[NULL:.*$]]55 // CHECK: {{^}}[[MASTER_ID:[0-9]+]]: ompt_event_initial_task_begin56 57 // parallel region used only to determine worker thread id58 // CHECK: {{^}}[[MASTER_ID]]: ompt_event_parallel_begin59 // CHECK: {{^}}[[WID:[0-9]+]]: ompt_event_implicit_task{{.*}}thread_num=160 61 // thread_num must be equal to 1 for both explicit and the implicit tasks62 // CHECK: {{^}}[[WID]]: ancestor_level=0 id=1 task_type=ompt_task_explicit63 // CHECK-SAME: thread_num=164 // CHECK: {{^}}[[WID]]: ancestor_level=1 id=0 task_type=ompt_task_implicit65 // CHECK-SAME: thread_num=166 // clang-format on67 68 return 0;69}70