From ceb723765272e7cf712fcd78f33806b5a2af9a8f Mon Sep 17 00:00:00 2001
From: itrowbri <Ian.Trowbridge@amd.com>
Date: Mon, 9 Sep 2024 11:45:57 -0500
Subject: [PATCH] SWDEV-466452: incomplete gromacs pftrace (#1040)

* SWDEV-466452: Inserted `tracing_session->FlushBlocking()` after
TRACE_EVENT_END when recording trace events to resolve incomplete
pftrace bug

* Update source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Formatting

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jonathan R. Madsen <jrmadsen@users.noreply.github.com>
---
 source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp b/source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp
index f7962bdf..a60a98ea 100644
--- a/source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp
+++ b/source/lib/rocprofiler-sdk-tool/generatePerfetto.cpp
@@ -272,6 +272,7 @@ write_perfetto(
                               itr.correlation_id.internal);
             TRACE_EVENT_END(
                 sdk::perfetto_category<sdk::category::hsa_api>::name, track, itr.end_timestamp);
+            tracing_session->FlushBlocking();
         }
 
         for(auto itr : *hip_api_data)
@@ -300,6 +301,7 @@ write_perfetto(
                               itr.correlation_id.internal);
             TRACE_EVENT_END(
                 sdk::perfetto_category<sdk::category::hip_api>::name, track, itr.end_timestamp);
+            tracing_session->FlushBlocking();
         }
 
         for(auto itr : *marker_api_data)
@@ -331,6 +333,7 @@ write_perfetto(
                               itr.correlation_id.internal);
             TRACE_EVENT_END(
                 sdk::perfetto_category<sdk::category::marker_api>::name, track, itr.end_timestamp);
+            tracing_session->FlushBlocking();
         }
 
         for(auto itr : *memory_copy_data)
@@ -365,6 +368,7 @@ write_perfetto(
                               itr.thread_id);
             TRACE_EVENT_END(
                 sdk::perfetto_category<sdk::category::memory_copy>::name, track, itr.end_timestamp);
+            tracing_session->FlushBlocking();
         }
 
         for(auto itr : *kernel_dispatch_data)
@@ -424,6 +428,7 @@ write_perfetto(
             TRACE_EVENT_END(sdk::perfetto_category<sdk::category::kernel_dispatch>::name,
                             track,
                             itr.end_timestamp);
+            tracing_session->FlushBlocking();
         }
     }
 
@@ -495,6 +500,7 @@ write_perfetto(
                               mem_cpy_tracks.at(mitr.first),
                               itr.first,
                               itr.second / bytes_multiplier);
+                tracing_session->FlushBlocking();
             }
         }
     }