Jonathan R. Madsen
fa4295db6b
* Initial Implementation: include * Initial Implementation: lib details (#11) * Initial Implementation: lib details * Initial Implementation: lib (#12) * Initial Implementation: lib * Initial Implementation: source (#13) * Initial Implementation: source * Initial Implementation: samples (#14) * Initial Implementation: samples * Initial Implementation: tests (#15) * Initial Implementation: tests * Initial Implementation: scripts (#16) * Initial Implementation: scripts * Initial Implementation: cmake (#17) * Initial Implementation: cmake * Initial Implementation: top-level files (#18) * Initial Implementation: top-level files - clang-format - clang-tidy - cmake-format - ignore build and cache directories - main CMakeLists.txt - pyproject.toml (python formatting) - VERSION file * Initial Implementation: workflow (#19) * Fix unused variable - rocprofiler_register_warn_level
102 lines
3.2 KiB
C++
102 lines
3.2 KiB
C++
#pragma once
|
|
|
|
#include "fwd.h"
|
|
|
|
#include <dlfcn.h>
|
|
#include <cstdint>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
|
|
#ifndef ROCP_REG_FILE_NAME
|
|
# define ROCP_REG_FILE_NAME \
|
|
::std::string{ __FILE__ } \
|
|
.substr(::std::string_view{ __FILE__ }.find_last_of('/') + 1) \
|
|
.c_str()
|
|
#endif
|
|
|
|
namespace
|
|
{
|
|
decltype(hip_init)* hip_init_fn = nullptr;
|
|
decltype(hsa_init)* hsa_init_fn = nullptr;
|
|
decltype(roctxRangePush)* roctxRangePush_fn = nullptr;
|
|
decltype(roctxRangePush)* roctxRangePop_fn = nullptr;
|
|
|
|
enum rocp_reg_test_modes : uint8_t
|
|
{
|
|
ROCP_REG_TEST_NONE = 0x0,
|
|
ROCP_REG_TEST_HIP = (1 << 0),
|
|
ROCP_REG_TEST_HSA = (1 << 1),
|
|
ROCP_REG_TEST_ROCTX = (1 << 2),
|
|
};
|
|
|
|
template <uint8_t Idx = ROCP_REG_TEST_NONE>
|
|
inline void
|
|
resolve_symbols(int _open_mode = RTLD_LOCAL | RTLD_LAZY)
|
|
{
|
|
auto* _open_mode_env = std::getenv("ROCP_REG_TEST_OPEN_MODE");
|
|
if(_open_mode_env)
|
|
{
|
|
constexpr auto npos = std::string_view::npos;
|
|
auto _open_mode_v = std::string_view{ _open_mode_env };
|
|
if(_open_mode_v.find("RTLD_GLOBAL") != npos)
|
|
_open_mode = RTLD_GLOBAL;
|
|
else if(_open_mode_v.find("RTLD_NOLOAD") != npos)
|
|
_open_mode = RTLD_NOLOAD;
|
|
else
|
|
_open_mode = RTLD_LOCAL;
|
|
|
|
if(_open_mode_v.find("RTLD_NOW") != npos)
|
|
_open_mode |= RTLD_NOW;
|
|
else
|
|
_open_mode |= RTLD_LAZY;
|
|
}
|
|
|
|
auto _resolve_dlopen = [_open_mode](void*& _handle, const char* _lib_name) {
|
|
fprintf(
|
|
stderr, "[%s] dlopen %s, %i\n", ROCP_REG_FILE_NAME, _lib_name, _open_mode);
|
|
_handle = dlopen(_lib_name, _open_mode);
|
|
if(!_handle)
|
|
{
|
|
fprintf(stderr, "Failure opening '%s'\n", _lib_name);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
};
|
|
|
|
auto _resolve_dlsym = [](auto& _func, void* _handle, const char* _func_name) {
|
|
if(!_func && _handle && _func_name)
|
|
{
|
|
auto* _func_v = dlsym(_handle, _func_name);
|
|
if(_func_v) *(void**) (&_func) = _func_v;
|
|
}
|
|
};
|
|
|
|
void* amdhip_handle = nullptr;
|
|
void* hsart_handle = nullptr;
|
|
void* roctx_handle = nullptr;
|
|
|
|
if constexpr((Idx & ROCP_REG_TEST_HIP) == ROCP_REG_TEST_HIP)
|
|
{
|
|
hip_init_fn = hip_init;
|
|
if(!hip_init_fn) _resolve_dlopen(amdhip_handle, "libamdhip64.so");
|
|
_resolve_dlsym(hip_init_fn, amdhip_handle, "hip_init");
|
|
}
|
|
|
|
if constexpr((Idx & ROCP_REG_TEST_HSA) == ROCP_REG_TEST_HSA)
|
|
{
|
|
hsa_init_fn = hsa_init;
|
|
if(!hsa_init_fn) _resolve_dlopen(hsart_handle, "libhsa-runtime64.so");
|
|
_resolve_dlsym(hsa_init_fn, hsart_handle, "hsa_init");
|
|
}
|
|
|
|
if constexpr((Idx & ROCP_REG_TEST_ROCTX) == ROCP_REG_TEST_ROCTX)
|
|
{
|
|
roctxRangePush_fn = roctxRangePush;
|
|
roctxRangePop_fn = roctxRangePop;
|
|
if(!roctxRangePush_fn || !roctxRangePop_fn)
|
|
_resolve_dlopen(roctx_handle, "libroctx64.so");
|
|
_resolve_dlsym(roctxRangePush_fn, roctx_handle, "roctxRangePush");
|
|
_resolve_dlsym(roctxRangePop_fn, roctx_handle, "roctxRangePop");
|
|
}
|
|
}
|
|
} // namespace
|