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

PC sampling client: using raw pointers (#902)

Browse Source 
* PC sampling client: using raw pointers to prevent premature destruction of buffers

* PCS client: freeing buffer_ids

[ROCm/rocprofiler-sdk commit: b0c41827c3]
This commit is contained in:
Vladimir Indic
2024-06-04 19:26:46 +02:00
committed by Ammar ELWazir
parent 1c0dcebe94
commit cb69616d46
3 changed files with 60 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-sdk/samples/pc_sampling/pcs.hpp
+14 -3
View File
@@ -35,8 +35,9 @@ constexpr size_t BUFFER_SIZE_BYTES = 8192;
constexpr size_t WATERMARK = (BUFFER_SIZE_BYTES / 4);
struct tool_agent_info;
using avail_configs_vec_t = std::vector<rocprofiler_pc_sampling_configuration_t>;
using tool_agent_info_vec_t = std::vector<std::unique_ptr<tool_agent_info>>;
using avail_configs_vec_t = std::vector<rocprofiler_pc_sampling_configuration_t>;
using tool_agent_info_vec_t = std::vector<std::unique_ptr<tool_agent_info>>;
using pc_sampling_buffer_id_vec_t = std::vector<rocprofiler_buffer_id_t>;
struct tool_agent_info
{
@@ -50,8 +51,18 @@ struct tool_agent_info
// meaning we were not able to enable PC sampling service.
// Check the `tool_init` for more information.
extern tool_agent_info_vec_t gpu_agents;
// Must be called first (prior to any other function from this namespace)
void
init();
// Must be called at the end of the `tool_fini`
void
fini();
// Ids of the buffers used as containers for PC sampling records
extern std::vector<rocprofiler_buffer_id_t> buffer_ids;
pc_sampling_buffer_id_vec_t*
get_pc_sampling_buffer_ids();
void
find_all_gpu_agents_supporting_pc_sampling();