76 lines
2.6 KiB
C++
76 lines
2.6 KiB
C++
|
//===---------- OmptCallback.cpp - Generic OMPT callbacks --------- C++ -*-===//
|
||
|
|
//
|
||
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
//
|
||
|
|
// OMPT support for PluginInterface
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#ifdef OMPT_SUPPORT
|
||
|
|
|
||
|
|
#include "llvm/Support/DynamicLibrary.h"
|
||
|
|
|
||
|
|
#include <cstdlib>
|
||
|
|
#include <cstring>
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include "Shared/Debug.h"
|
||
|
|
|
||
|
|
#include "OpenMP/OMPT/Callback.h"
|
||
|
|
#include "OpenMP/OMPT/Connector.h"
|
||
|
|
|
||
|
|
using namespace llvm::omp::target::ompt;
|
||
|
|
|
||
|
|
bool llvm::omp::target::ompt::Initialized = false;
|
||
|
|
|
||
|
|
ompt_get_callback_t llvm::omp::target::ompt::lookupCallbackByCode = nullptr;
|
||
|
|
ompt_function_lookup_t llvm::omp::target::ompt::lookupCallbackByName = nullptr;
|
||
|
|
|
||
|
|
int llvm::omp::target::ompt::initializeLibrary(ompt_function_lookup_t lookup,
|
||
|
|
int initial_device_num,
|
||
|
|
ompt_data_t *tool_data) {
|
||
|
|
DP("OMPT: Executing initializeLibrary (libomptarget)\n");
|
||
|
|
#define bindOmptFunctionName(OmptFunction, DestinationFunction) \
|
||
|
|
if (lookup) \
|
||
|
|
DestinationFunction = (OmptFunction##_t)lookup(#OmptFunction); \
|
||
|
|
DP("OMPT: initializeLibrary (libomptarget) bound %s=%p\n", \
|
||
|
|
#DestinationFunction, ((void *)(uint64_t)DestinationFunction));
|
||
|
|
|
||
|
|
bindOmptFunctionName(ompt_get_callback, lookupCallbackByCode);
|
||
|
|
#undef bindOmptFunctionName
|
||
|
|
|
||
|
|
// Store pointer of 'ompt_libomp_target_fn_lookup' for use by the plugin
|
||
|
|
lookupCallbackByName = lookup;
|
||
|
|
|
||
|
|
Initialized = true;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void llvm::omp::target::ompt::finalizeLibrary(ompt_data_t *tool_data) {
|
||
|
|
DP("OMPT: Executing finalizeLibrary (libomptarget)\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
void llvm::omp::target::ompt::connectLibrary() {
|
||
|
|
DP("OMPT: Entering connectLibrary (libomptarget)\n");
|
||
|
|
/// Connect plugin instance with libomptarget
|
||
|
|
OmptLibraryConnectorTy LibomptargetConnector("libomptarget");
|
||
|
|
ompt_start_tool_result_t OmptResult;
|
||
|
|
|
||
|
|
// Initialize OmptResult with the init and fini functions that will be
|
||
|
|
// called by the connector
|
||
|
|
OmptResult.initialize = ompt::initializeLibrary;
|
||
|
|
OmptResult.finalize = ompt::finalizeLibrary;
|
||
|
|
OmptResult.tool_data.value = 0;
|
||
|
|
|
||
|
|
// Now call connect that causes the above init/fini functions to be called
|
||
|
|
LibomptargetConnector.connect(&OmptResult);
|
||
|
|
DP("OMPT: Exiting connectLibrary (libomptarget)\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|