AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use rocprofiler-SDK for OMPT tracing (#702)

Browse Source 
Switch to using SDK for OMPT tracing and remove older OMPT code path
This commit is contained in:
Kian Cossettini
2025-08-26 16:54:01 -04:00
committed by GitHub
parent 5f4e0dc889
commit 07a7b9b845
16 changed files with 294 additions and 682 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-systems/source/lib/core/rocprofiler-sdk.cpp
+10
View File
@@ -414,6 +414,7 @@ get_callback_domains()
{
// Argument tracing is supported in rocprofiler-sdk 0.6.0 and later
supported.emplace(ROCPROFILER_CALLBACK_TRACING_RCCL_API);
supported.emplace(ROCPROFILER_CALLBACK_TRACING_OMPT);
supported.emplace(ROCPROFILER_CALLBACK_TRACING_ROCDECODE_API);
}
# endif
@@ -436,6 +437,15 @@ get_callback_domains()
_data.emplace(ROCPROFILER_CALLBACK_TRACING_RCCL_API);
}
# if ROCPROFILER_VERSION >= 600
if(config::get_use_ompt() && _version.formatted >= 600)
{
// Translate some configuration settings to rocprofiler domains
_data.emplace(ROCPROFILER_CALLBACK_TRACING_OMPT);
}
# endif
// Check that the domains are valid
const auto valid_choices =
settings::instance()->at("ROCPROFSYS_ROCM_DOMAINS")->get_choices();