Skip to content

Commit 17003f6

Browse files
authored
Merge pull request #1 from MooreThreads/remove_wait_for_buffer_write
Update MuptiActivityProfiler.cpp
2 parents 263555a + f4760cd commit 17003f6

File tree

1 file changed

+1
-14
lines changed

1 file changed

+1
-14
lines changed

libkineto/src/MuptiActivityProfiler.cpp

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -264,27 +264,14 @@ void MuptiActivityProfiler::processTraceInternal(ActivityLogger& logger) {
264264
addOverheadSample(flushOverhead_, mupti_.flushOverhead);
265265
}
266266
if (traceBuffers_->gpu) {
267-
auto count_and_size1 = mupti_.processActivities(
267+
auto count_and_size = mupti_.processActivities(
268268
*traceBuffers_->gpu,
269269
std::bind(
270270
&MuptiActivityProfiler::handleMuptiActivity,
271271
this,
272272
std::placeholders::_1,
273273
&logger));
274274

275-
// Wait for mupti buffer write
276-
std::this_thread::sleep_for(seconds(3));
277-
auto count_and_size2 = mupti_.processActivities(
278-
*traceBuffers_->gpu,
279-
std::bind(
280-
&MuptiActivityProfiler::handleMuptiActivity,
281-
this,
282-
std::placeholders::_1,
283-
&logger));
284-
std::pair<int, size_t> count_and_size{0, 0};
285-
count_and_size.first = count_and_size1.first + count_and_size2.first;
286-
count_and_size.second = count_and_size1.second + count_and_size2.second;
287-
288275
LOG(INFO) << "Processed " << count_and_size.first << " GPU records ("
289276
<< count_and_size.second << " bytes)";
290277
LOGGER_OBSERVER_ADD_EVENT_COUNT(count_and_size.first);

0 commit comments

Comments
 (0)