From 40e267fb3403b4db16d309e861e83ad3d1b2f017 Mon Sep 17 00:00:00 2001
From: Ammar ELWazir <Ammar.ELWazir@amd.com>
Date: Tue, 4 Jul 2023 18:22:43 +0000
Subject: [PATCH] Fixing AMD SYS

Change-Id: I0f47f8e2a4275e0ef0f406e4f3efd9f6c2c4acd3
---
 src/tools/tool.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/tools/tool.cpp b/src/tools/tool.cpp
index de2571c254..7b1e1f8749 100644
--- a/src/tools/tool.cpp
+++ b/src/tools/tool.cpp
@@ -493,6 +493,7 @@ void wait_for_amdsys() {
             for ([[maybe_unused]] rocprofiler_buffer_id_t buffer_id : buffer_ids) {
               CHECK_ROCPROFILER(rocprofiler_flush_data(session_id, buffer_id));
             }
+            rocprofiler::TraceBufferBase::FlushAll();
           }
           break;
         }
@@ -506,6 +507,7 @@ void wait_for_amdsys() {
             for ([[maybe_unused]] rocprofiler_buffer_id_t buffer_id : buffer_ids) {
               CHECK_ROCPROFILER(rocprofiler_flush_data(session_id, buffer_id));
             }
+            rocprofiler::TraceBufferBase::FlushAll();
           }
           amd_sys_handler.exchange(false, std::memory_order_release);
           flag = true;