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

[SDK] Add Stack IDs (#269)

Browse Source 
* Add Stack IDs

* Add memcpy test

* Add async corr id record

* Async events use `rocprofiler_async_correlation_id_t`
* Sync events use `rocprofiler_correlation_id_t`

* Update ATT to use asnyc IDs

* Review comments
This commit is contained in:
Kuricheti, Mythreya
2025-03-20 22:52:48 -07:00
committed by GitHub
parent ae0db8cee5
commit f27f76716e
36 changed files with 341 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/rocprofiler-sdk/hip/stream.cpp
+6
View File
@@ -176,6 +176,7 @@ FuncT create_write_functor(RetT (*func)(Args...))
auto tracer_data = common::init_public_api_struct(callback_api_data_t{});
auto internal_corr_id = 0;
auto ancestor_corr_id = 0;
constexpr auto stream_idx = common::mpl::index_of<hipStream_t*, function_args_type>::value;
auto stream = std::get<stream_idx>(std::make_tuple(std::forward<Args>(args)...));
@@ -194,6 +195,7 @@ FuncT create_write_functor(RetT (*func)(Args...))
thr_id,
internal_corr_id,
external_corr_ids,
ancestor_corr_id,
ROCPROFILER_CALLBACK_TRACING_HIP_STREAM_API,
ROCPROFILER_HIP_STREAM_CREATE,
tracer_data);
@@ -237,6 +239,7 @@ FuncT create_destroy_functor(RetT (*func)(Args...))
auto tracer_data = common::init_public_api_struct(callback_api_data_t{});
auto internal_corr_id = 0;
auto ancestor_corr_id = 0;
auto stream = std::get<stream_idx>(std::make_tuple(std::forward<Args>(args)...));
@@ -252,6 +255,7 @@ FuncT create_destroy_functor(RetT (*func)(Args...))
thr_id,
internal_corr_id,
external_corr_ids,
ancestor_corr_id,
ROCPROFILER_CALLBACK_TRACING_HIP_STREAM_API,
ROCPROFILER_HIP_STREAM_DESTROY,
tracer_data);
@@ -296,6 +300,7 @@ FuncT create_read_functor(RetT (*func)(Args...))
auto tracer_data = common::init_public_api_struct(callback_api_data_t{});
auto internal_corr_id = 0;
auto ancestor_corr_id = 0;
auto stream = std::get<stream_idx>(std::make_tuple(std::forward<Args>(args)...));
@@ -306,6 +311,7 @@ FuncT create_read_functor(RetT (*func)(Args...))
thr_id,
internal_corr_id,
external_corr_ids,
ancestor_corr_id,
ROCPROFILER_CALLBACK_TRACING_HIP_STREAM_API,
ROCPROFILER_HIP_STREAM_SET,
tracer_data);