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

Allow multiple agents in a single context for agent profiling (#908)

Browse Source 
Allow multiple profiles for agent profiling

Co-authored-by: Benjamin Welton <ben@amd.com>
This commit is contained in:
Benjamin Welton
2024-06-13 12:39:16 -07:00
committed by Benjamin Welton
parent 9676295d3d
commit 06d46ba3a0
4 changed files with 281 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/rocprofiler-sdk/counters/controller.cpp
+5 -4
View File
@@ -92,10 +92,11 @@ CounterController::configure_agent_collection(rocprofiler_context_id_t
std::make_unique<rocprofiler::context::agent_counter_collection_service>();
}
ctx.agent_counter_collection->agent_id = agent_id;
ctx.agent_counter_collection->cb = cb;
ctx.agent_counter_collection->user_data = user_data;
ctx.agent_counter_collection->buffer = buffer_id;
ctx.agent_counter_collection->agent_data.emplace_back();
ctx.agent_counter_collection->agent_data.back().callback_data = {.ptr = user_data};
ctx.agent_counter_collection->agent_data.back().agent_id = agent_id;
ctx.agent_counter_collection->agent_data.back().cb = cb;
ctx.agent_counter_collection->agent_data.back().buffer = buffer_id;
return ROCPROFILER_STATUS_SUCCESS;
}