-
Notifications
You must be signed in to change notification settings - Fork 5
/
ROCProfilerModule.cpp
73 lines (58 loc) · 1.77 KB
/
ROCProfilerModule.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
//==============================================================================
// Copyright (c) 2017 Advanced Micro Devices, Inc. All rights reserved.
/// \author AMD Developer Tools
/// \file
/// \brief This class manages the dynamic loading of ROCProfiler entry points
//==============================================================================
#include <string>
#include "ROCProfilerModule.h"
#if defined(_WIN64) || defined(_WIN32)
#pragma message("ROCProfiler does not support Windows")
const char* ROCProfilerModule::s_defaultModuleName = "";
#else
const char* ROCProfilerModule::s_defaultModuleName = "librocprofiler64.so";
#endif
ROCProfilerModule::ROCProfilerModule(void) : m_isModuleLoaded(false)
{
Initialize();
LoadModule();
}
ROCProfilerModule::~ROCProfilerModule(void)
{
UnloadModule();
}
void ROCProfilerModule::Initialize()
{
#define X(SYM) SYM = nullptr;
ROC_PROFILER_API_TABLE;
#undef X
m_isModuleLoaded = false;
}
void ROCProfilerModule::UnloadModule()
{
m_dynamicLibraryHelper.UnloadModule();
Initialize();
}
bool ROCProfilerModule::LoadModule(const std::string& moduleName)
{
// Load from specified module
bool bLoaded = m_dynamicLibraryHelper.LoadModule(moduleName);
if (!bLoaded)
{
// Load from deafult module
bLoaded = m_dynamicLibraryHelper.LoadModule(s_defaultModuleName);
}
if (bLoaded)
{
#define MAKE_STRING(s) ""#s
#define X(SYM) SYM = reinterpret_cast<SYM##_fn_t>(m_dynamicLibraryHelper.GetProcAddress(MAKE_STRING(SYM)));
ROC_PROFILER_API_TABLE;
#undef X
#undef MAKE_STRING
// Check if we initialized all the function pointers
#define X(SYM) && SYM != nullptr
m_isModuleLoaded = true ROC_PROFILER_API_TABLE;
#undef X
}
return m_isModuleLoaded;
}