Fix "ROCPROFSYS_USE_ROCM" runtime config setting. (#144)
[ROCm/rocprofiler-systems commit: b6b39af011]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
95503cde21
Коммит
70b456f2a3
@@ -1818,6 +1818,17 @@ get_use_amd_smi()
|
||||
#endif
|
||||
}
|
||||
|
||||
bool
|
||||
get_use_rocm()
|
||||
{
|
||||
#if defined(ROCPROFSYS_USE_ROCM) && ROCPROFSYS_USE_ROCM > 0
|
||||
static auto _v = get_config()->find("ROCPROFSYS_USE_ROCM");
|
||||
return static_cast<tim::tsettings<bool>&>(*_v->second).get();
|
||||
#else
|
||||
return false;
|
||||
#endif
|
||||
}
|
||||
|
||||
bool&
|
||||
get_use_sampling()
|
||||
{
|
||||
|
||||
@@ -204,6 +204,9 @@ get_use_timemory() ROCPROFSYS_HOT;
|
||||
bool&
|
||||
get_use_causal() ROCPROFSYS_HOT;
|
||||
|
||||
bool
|
||||
get_use_rocm() ROCPROFSYS_HOT;
|
||||
|
||||
bool
|
||||
get_use_amd_smi() ROCPROFSYS_HOT;
|
||||
|
||||
|
||||
@@ -820,7 +820,7 @@ rocprofsys_finalize_hidden(void)
|
||||
}
|
||||
|
||||
#if defined(ROCPROFSYS_USE_ROCM) && ROCPROFSYS_USE_ROCM > 0
|
||||
// TODO: option for rocm
|
||||
if(get_use_rocm())
|
||||
{
|
||||
ROCPROFSYS_VERBOSE_F(1, "Shutting down ROCm...\n");
|
||||
rocprofiler_sdk::shutdown();
|
||||
|
||||
@@ -1297,6 +1297,11 @@ rocprofiler_configure(uint32_t version, const char* runtime_version, uint32_t pr
|
||||
rocprofsys::get_state() < rocprofsys::State::Active)
|
||||
rocprofsys_init_tooling_hidden();
|
||||
|
||||
if(!rocprofsys::config::get_use_rocm())
|
||||
{
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
// set the client name
|
||||
id->name = "rocprofsys";
|
||||
|
||||
|
||||
Ссылка в новой задаче
Block a user