talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: 1e13be30e2c6838d4a2ff768a39c414bc80534bb
This commit is contained in:
MediaPipe Team
2022-09-06 21:46:17 +00:00
committed by Sebastian Schmidt
parent 63e679d99c
commit 4dc4b19ddb
639 changed files with 71327 additions and 2078 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/profiler/graph_tracer.cc
+9 -1
View File
@@ -14,9 +14,9 @@
#include "mediapipe/framework/profiler/graph_tracer.h"
#include "absl/synchronization/mutex.h"
#include "absl/time/time.h"
#include "mediapipe/framework/calculator_context.h"
#include "mediapipe/framework/calculator_profile.pb.h"
#include "mediapipe/framework/input_stream_shard.h"
#include "mediapipe/framework/output_stream_shard.h"
#include "mediapipe/framework/packet.h"
@@ -117,14 +117,22 @@ Timestamp GraphTracer::TimestampAfter(absl::Time begin_time) {
return TraceBuilder::TimestampAfter(trace_buffer_, begin_time);
}
// The mutex to guard GraphTracer::trace_builder_.
absl::Mutex* trace_builder_mutex() {
static absl::Mutex trace_builder_mutex(absl::kConstInit);
return &trace_builder_mutex;
}
void GraphTracer::GetTrace(absl::Time begin_time, absl::Time end_time,
GraphTrace* result) {
absl::MutexLock lock(trace_builder_mutex());
trace_builder_.CreateTrace(trace_buffer_, begin_time, end_time, result);
trace_builder_.Clear();
}
void GraphTracer::GetLog(absl::Time begin_time, absl::Time end_time,
GraphTrace* result) {
absl::MutexLock lock(trace_builder_mutex());
trace_builder_.CreateLog(trace_buffer_, begin_time, end_time, result);
trace_builder_.Clear();
}