experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Only save per-thread events when non-empty

Browse Source 
PiperOrigin-RevId: 283439027
Change-Id: I33118a45a69fff30b3c29d927c8a8380536dd1af
This commit is contained in:
Jose Baiocchi 2019-12-02 16:09:38 -08:00 committed by TensorFlower Gardener
parent 6a0e85b2da
commit 71681bd691
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/profiler/internal/traceme_recorder.cc
View File

@ -199,7 +199,9 @@ void TraceMeRecorder::RegisterThread(int32 tid, ThreadLocalRecorder* thread) {
void TraceMeRecorder::UnregisterThread(TraceMeRecorder::ThreadEvents&& events) {
mutex_lock lock(mutex_);
threads_.erase(events.thread.tid);
orphaned_events_.push_back(std::move(events));
if (!events.events.empty()) {
orphaned_events_.push_back(std::move(events));
}
}
// This method is performance critical and should be kept fast. It is called
@ -211,7 +213,10 @@ TraceMeRecorder::Events TraceMeRecorder::Clear() {
std::swap(orphaned_events_, result);
for (const auto& entry : threads_) {
auto* recorder = entry.second;
result.push_back(recorder->Clear());
TraceMeRecorder::ThreadEvents events = recorder->Clear();
if (!events.events.empty()) {
result.push_back(std::move(events));
}
}
return result;
}