Fix "ROCPROFSYS_USE_ROCM" runtime config setting. (#144)

[ROCm/rocprofiler-systems commit: b6b39af011]
Этот коммит содержится в:
David Galiffi
2025-03-27 16:03:46 -04:00
коммит произвёл GitHub
родитель 95503cde21
Коммит 70b456f2a3
4 изменённых файлов: 20 добавлений и 1 удалений
+11
Просмотреть файл
@@ -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()
{
+3
Просмотреть файл
@@ -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;
+1 -1
Просмотреть файл
@@ -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();
+5
Просмотреть файл
@@ -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";