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

Add perfetto support for scratch memory (#303)

Browse Source 
* Add perfetto support for scratch memory

* Updated tests and docs.

* Update docs data

* Added underflow check

* Record all free events to 0 bytes

* Add format

* Address review comment

* updated tests for scratch memory

* update scratch-memory tests.
This commit is contained in:
U, Srihari
2025-07-09 21:05:45 +05:30
committed by GitHub
parent 2c4e20b951
commit 6f2a5a9646
13 changed files with 228 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/rocprofiler-sdk/hsa/scratch_memory.cpp
+9
View File
@@ -522,6 +522,7 @@ impl(Args... args)
tls.buffered_data.queue_id = {event_data.scratch_alloc_start->queue->id};
tls.buffered_data.thread_id = thr_id;
tls.buffered_data.start_timestamp = common::timestamp_ns();
tls.buffered_data.allocation_size = 0;
}
}
else if constexpr(OpPhase == ROCPROFILER_CALLBACK_PHASE_EXIT)
@@ -530,6 +531,14 @@ impl(Args... args)
{
tls.buffered_data.flags = get_flags(event_data);
tls.buffered_data.end_timestamp = common::timestamp_ns();
if constexpr(OpIdx == ROCPROFILER_SCRATCH_MEMORY_ALLOC)
{
tls.buffered_data.allocation_size = event_data.scratch_alloc_end->size;
}
else if constexpr(OpIdx == ROCPROFILER_SCRATCH_MEMORY_FREE)
{
tls.buffered_data.allocation_size = 0;
}
}
if(!tls.callback_contexts.empty())