Jonathan R. Madsen
4a8c101839
* Update .clang-format
- set SortIncludes to true
* Reformat source with includes sorted
[ROCm/rocprofiler-sdk commit: 41b1d91841]
125 rader
5.0 KiB
C++
125 rader
5.0 KiB
C++
#ifndef PC_SAMPLING_COMMON_H
|
|
#define PC_SAMPLING_COMMON_H
|
|
|
|
#include <rocprofiler/rocprofiler.h>
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
|
|
constexpr size_t BUFFER_SIZE_BYTES = 4096;
|
|
constexpr size_t WATERMARK = (BUFFER_SIZE_BYTES / 2);
|
|
|
|
#define ROCPROFILER_CALL(result, msg) \
|
|
{ \
|
|
rocprofiler_status_t status = result; \
|
|
if(status != ROCPROFILER_STATUS_SUCCESS) \
|
|
{ \
|
|
puts(#result " failed"); \
|
|
} \
|
|
}
|
|
|
|
// We might want to test the calls that fails
|
|
// e.g. calling `rocprofiler_configure_pc_sampling_service `
|
|
// after previous initialization.
|
|
#define ROCPROFILER_CALL_FAILS(result, msg) \
|
|
{ \
|
|
rocprofiler_status_t status = result; \
|
|
if(status == ROCPROFILER_STATUS_SUCCESS) \
|
|
{ \
|
|
puts(#result " succeeded"); \
|
|
} \
|
|
}
|
|
|
|
static rocprofiler_status_t
|
|
find_first_gpu_agent_impl(rocprofiler_agent_t** agents, size_t num_agents, void* data)
|
|
{
|
|
// data is required
|
|
if(!data) return ROCPROFILER_STATUS_ERROR;
|
|
|
|
auto* _out_agent = static_cast<rocprofiler_agent_t*>(data);
|
|
// find the first GPU agent
|
|
for(size_t i = 0; i < num_agents; i++)
|
|
{
|
|
if(agents[i]->type == ROCPROFILER_AGENT_TYPE_GPU)
|
|
{
|
|
*_out_agent = *agents[i];
|
|
printf("[%s] %s :: id=%zu, type=%i, num pc sample configs=%zu\n",
|
|
__FUNCTION__,
|
|
_out_agent->name,
|
|
_out_agent->id.handle,
|
|
_out_agent->type,
|
|
_out_agent->pc_sampling_configs.size);
|
|
return ROCPROFILER_STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
printf("[%s] %s :: id=%zu, type=%i, num pc sample configs=%zu\n",
|
|
__FUNCTION__,
|
|
agents[i]->name,
|
|
agents[i]->id.handle,
|
|
agents[i]->type,
|
|
agents[i]->pc_sampling_configs.size);
|
|
}
|
|
}
|
|
return ROCPROFILER_STATUS_ERROR;
|
|
}
|
|
|
|
static rocprofiler_agent_t
|
|
find_first_gpu_agent()
|
|
{
|
|
// This function returns the first gpu agent it encounters.
|
|
// TODO: write the better function querying information about the agent,
|
|
// and return if the agent is MI200.
|
|
rocprofiler_agent_t gpu_agent;
|
|
|
|
ROCPROFILER_CALL(rocprofiler_query_available_agents(&find_first_gpu_agent_impl,
|
|
sizeof(rocprofiler_agent_t),
|
|
static_cast<void*>(&gpu_agent)),
|
|
"Failed to find GPU agents");
|
|
|
|
return gpu_agent;
|
|
}
|
|
|
|
static void
|
|
rocprofiler_pc_sampling_callback(rocprofiler_context_id_t /*context_id*/,
|
|
rocprofiler_buffer_id_t /*buffer_id*/,
|
|
rocprofiler_record_header_t** headers,
|
|
size_t num_headers,
|
|
void* /*data*/,
|
|
uint64_t drop_count)
|
|
{
|
|
// Vladimir: I am not sure if this is the right way of iterating over PC sampling records.
|
|
printf(
|
|
"The number of delivered samples is: %zu, while the number of dropped samples is: %lu.\n",
|
|
num_headers,
|
|
drop_count);
|
|
|
|
for(size_t i = 0; i < num_headers; i++)
|
|
{
|
|
auto* cur_header = headers[i];
|
|
if(cur_header->kind == 0)
|
|
{
|
|
auto* pc_sample = static_cast<rocprofiler_pc_sampling_record_t*>(cur_header->payload);
|
|
printf("--- pc: %lx, dispatch_id: %lx, timestamp: %lu, hardware_id: %lu\n",
|
|
pc_sample->pc,
|
|
pc_sample->dispatch_id,
|
|
pc_sample->timestamp,
|
|
pc_sample->hardware_id);
|
|
// Vladimir: How to parse the remaining part of the `rocprofiler_pc_sampling_record_t`
|
|
// struct?
|
|
}
|
|
}
|
|
// Vladimr: We might want to add somewhere in the documentation that headars actually contain PC
|
|
// samples.
|
|
}
|
|
|
|
static void
|
|
run_HIP_app()
|
|
{
|
|
// TODO: provide the simple HIP app
|
|
}
|
|
|
|
#endif
|