From f4760cd584540d004b9f1c14a9f68c34b6b41c01 Mon Sep 17 00:00:00 2001
From: zehan-mt <112623395+zehan-mt@users.noreply.github.com>
Date: Tue, 18 Jun 2024 17:51:00 +0800
Subject: [PATCH] Update MuptiActivityProfiler.cpp

resolve the issue: duplicate trace data collected from MUPTI GPU trace buffer
---
 libkineto/src/MuptiActivityProfiler.cpp | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

diff --git a/libkineto/src/MuptiActivityProfiler.cpp b/libkineto/src/MuptiActivityProfiler.cpp
index 07dde8c..5e585a8 100644
--- a/libkineto/src/MuptiActivityProfiler.cpp
+++ b/libkineto/src/MuptiActivityProfiler.cpp
@@ -264,7 +264,7 @@ void MuptiActivityProfiler::processTraceInternal(ActivityLogger& logger) {
       addOverheadSample(flushOverhead_, mupti_.flushOverhead);
     }
     if (traceBuffers_->gpu) {
-      auto count_and_size1 = mupti_.processActivities(
+      auto count_and_size = mupti_.processActivities(
           *traceBuffers_->gpu,
           std::bind(
               &MuptiActivityProfiler::handleMuptiActivity,
@@ -272,19 +272,6 @@ void MuptiActivityProfiler::processTraceInternal(ActivityLogger& logger) {
               std::placeholders::_1,
               &logger));
 
-      // Wait for mupti buffer write
-      std::this_thread::sleep_for(seconds(3));
-      auto count_and_size2 = mupti_.processActivities(
-          *traceBuffers_->gpu,
-          std::bind(
-              &MuptiActivityProfiler::handleMuptiActivity,
-              this,
-              std::placeholders::_1,
-              &logger));      
-      std::pair<int, size_t> count_and_size{0, 0};
-      count_and_size.first = count_and_size1.first + count_and_size2.first;
-      count_and_size.second = count_and_size1.second + count_and_size2.second;
-
       LOG(INFO) << "Processed " << count_and_size.first << " GPU records ("
                 << count_and_size.second << " bytes)";
       LOGGER_OBSERVER_ADD_EVENT_COUNT(count_and_size.first);