forked from OSchip/llvm-project
Missed to add new test case in previous commit
llvm-svn: 322179
This commit is contained in:
parent
14b512e20c
commit
1014a6b6c6
|
@ -0,0 +1,150 @@
|
|||
// RUN: %libomp-compile-and-run | %sort-threads | FileCheck %s
|
||||
// REQUIRES: ompt
|
||||
|
||||
// This test checks that values stored in task_data in a barrier_begin event
|
||||
// are still present in the corresponding barrier_end event.
|
||||
// Therefore, callback implementations different from the ones in callback.h are neccessary.
|
||||
// This is a test for an issue reported in
|
||||
// https://github.com/OpenMPToolsInterface/LLVM-openmp/issues/39
|
||||
|
||||
#define _BSD_SOURCE
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
#include <inttypes.h>
|
||||
#include <omp.h>
|
||||
#include <ompt.h>
|
||||
|
||||
static const char* ompt_thread_type_t_values[] = {
|
||||
NULL,
|
||||
"ompt_thread_initial",
|
||||
"ompt_thread_worker",
|
||||
"ompt_thread_other"
|
||||
};
|
||||
|
||||
static ompt_get_unique_id_t ompt_get_unique_id;
|
||||
static ompt_get_thread_data_t ompt_get_thread_data;
|
||||
|
||||
int main()
|
||||
{
|
||||
#pragma omp parallel num_threads(4)
|
||||
{
|
||||
#pragma omp master
|
||||
{
|
||||
sleep(1);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// Check if libomp supports the callbacks for this test.
|
||||
// CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_sync_region'
|
||||
// CHECK-NOT: {{^}}0: Could not register callback 'ompt_callback_sync_region_wait'
|
||||
|
||||
// CHECK: 0: NULL_POINTER=[[NULL:.*$]]
|
||||
|
||||
// master thread implicit barrier at parallel end
|
||||
// CHECK: {{^}}[[MASTER_ID:[0-9]+]]: ompt_event_barrier_begin: parallel_id=0, task_id=[[TASK_ID:[0-9]+]], codeptr_ra={{0x[0-f]*}}
|
||||
// CHECK: {{^}}[[MASTER_ID]]: ompt_event_wait_barrier_begin: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra={{0x[0-f]*}}
|
||||
// CHECK: {{^}}[[MASTER_ID]]: ompt_event_wait_barrier_end: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra={{0x[0-f]*}}
|
||||
// CHECK: {{^}}[[MASTER_ID]]: ompt_event_barrier_end: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra={{0x[0-f]*}}
|
||||
|
||||
|
||||
// worker thread implicit barrier at parallel end
|
||||
// CHECK: {{^}}[[THREAD_ID:[0-9]+]]: ompt_event_barrier_begin: parallel_id=0, task_id=[[TASK_ID:[0-9]+]], codeptr_ra=[[NULL]]
|
||||
// CHECK: {{^}}[[THREAD_ID]]: ompt_event_wait_barrier_begin: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra=[[NULL]]
|
||||
// CHECK: {{^}}[[THREAD_ID]]: ompt_event_wait_barrier_end: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra=[[NULL]]
|
||||
// CHECK: {{^}}[[THREAD_ID]]: ompt_event_barrier_end: parallel_id=0, task_id=[[TASK_ID]], codeptr_ra=[[NULL]]
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void
|
||||
on_ompt_callback_thread_begin(
|
||||
ompt_thread_type_t thread_type,
|
||||
ompt_data_t *thread_data)
|
||||
{
|
||||
if(thread_data->ptr)
|
||||
printf("%s\n", "0: thread_data initially not null");
|
||||
thread_data->value = ompt_get_unique_id();
|
||||
printf("%" PRIu64 ": ompt_event_thread_begin: thread_type=%s=%d, thread_id=%" PRIu64 "\n", ompt_get_thread_data()->value, ompt_thread_type_t_values[thread_type], thread_type, thread_data->value);
|
||||
}
|
||||
|
||||
static void
|
||||
on_ompt_callback_sync_region(
|
||||
ompt_sync_region_kind_t kind,
|
||||
ompt_scope_endpoint_t endpoint,
|
||||
ompt_data_t *parallel_data,
|
||||
ompt_data_t *task_data,
|
||||
const void *codeptr_ra)
|
||||
{
|
||||
switch(endpoint)
|
||||
{
|
||||
case ompt_scope_begin:
|
||||
task_data->value = ompt_get_unique_id();
|
||||
if(kind == ompt_sync_region_barrier)
|
||||
printf("%" PRIu64 ": ompt_event_barrier_begin: parallel_id=%" PRIu64 ", task_id=%" PRIu64 ", codeptr_ra=%p\n", ompt_get_thread_data()->value, parallel_data->value, task_data->value, codeptr_ra);
|
||||
break;
|
||||
case ompt_scope_end:
|
||||
if(kind == ompt_sync_region_barrier)
|
||||
printf("%" PRIu64 ": ompt_event_barrier_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 ", codeptr_ra=%p\n", ompt_get_thread_data()->value, (parallel_data)?parallel_data->value:0, task_data->value, codeptr_ra);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
on_ompt_callback_sync_region_wait(
|
||||
ompt_sync_region_kind_t kind,
|
||||
ompt_scope_endpoint_t endpoint,
|
||||
ompt_data_t *parallel_data,
|
||||
ompt_data_t *task_data,
|
||||
const void *codeptr_ra)
|
||||
{
|
||||
switch(endpoint)
|
||||
{
|
||||
case ompt_scope_begin:
|
||||
if(kind == ompt_sync_region_barrier)
|
||||
printf("%" PRIu64 ": ompt_event_wait_barrier_begin: parallel_id=%" PRIu64 ", task_id=%" PRIu64 ", codeptr_ra=%p\n", ompt_get_thread_data()->value, parallel_data->value, task_data->value, codeptr_ra);
|
||||
break;
|
||||
case ompt_scope_end:
|
||||
if(kind == ompt_sync_region_barrier)
|
||||
printf("%" PRIu64 ": ompt_event_wait_barrier_end: parallel_id=%" PRIu64 ", task_id=%" PRIu64 ", codeptr_ra=%p\n", ompt_get_thread_data()->value, (parallel_data)?parallel_data->value:0, task_data->value, codeptr_ra);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
#define register_callback_t(name, type) \
|
||||
do{ \
|
||||
type f_##name = &on_##name; \
|
||||
if (ompt_set_callback(name, (ompt_callback_t)f_##name) == \
|
||||
ompt_set_never) \
|
||||
printf("0: Could not register callback '" #name "'\n"); \
|
||||
}while(0)
|
||||
|
||||
#define register_callback(name) register_callback_t(name, name##_t)
|
||||
|
||||
int ompt_initialize(
|
||||
ompt_function_lookup_t lookup,
|
||||
ompt_data_t *tool_data)
|
||||
{
|
||||
ompt_set_callback_t ompt_set_callback;
|
||||
ompt_set_callback = (ompt_set_callback_t) lookup("ompt_set_callback");
|
||||
ompt_get_unique_id = (ompt_get_unique_id_t) lookup("ompt_get_unique_id");
|
||||
ompt_get_thread_data = (ompt_get_thread_data_t) lookup("ompt_get_thread_data");
|
||||
register_callback(ompt_callback_sync_region);
|
||||
register_callback_t(ompt_callback_sync_region_wait, ompt_callback_sync_region_t);
|
||||
register_callback(ompt_callback_thread_begin);
|
||||
printf("0: NULL_POINTER=%p\n", (void*)NULL);
|
||||
return 1; //success
|
||||
}
|
||||
|
||||
void ompt_finalize(ompt_data_t *tool_data)
|
||||
{
|
||||
printf("0: ompt_event_runtime_shutdown\n");
|
||||
}
|
||||
|
||||
ompt_start_tool_result_t* ompt_start_tool(
|
||||
unsigned int omp_version,
|
||||
const char *runtime_version)
|
||||
{
|
||||
static ompt_start_tool_result_t ompt_start_tool_result = {&ompt_initialize,&ompt_finalize, 0};
|
||||
return &ompt_start_tool_result;
|
||||
}
|
Loading…
Reference in New Issue