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

Trace scheduling activities when using CPU-ONLY profiler

Browse Source 
PiperOrigin-RevId: 248008551
This commit is contained in:
Xiao Yu 2019-05-13 14:18:24 -07:00 committed by TensorFlower Gardener
parent 663e8479ce
commit 5f5340ca3e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/common_runtime/executor.cc
View File

@ -1603,8 +1603,8 @@ bool MightTrace(const NodeItem& item,
} }
auto* trace_collector = tracing::GetTraceCollector(); auto* trace_collector = tracing::GetTraceCollector();
if (trace_collector) { if (trace_collector) {
if (using_annotations) { if (using_annotations && trace_collector->IsEnabledForAnnotations()) {
return trace_collector->IsEnabledForAnnotations(); return true;
} }
} }
return profiler::TraceMeRecorder::Active( return profiler::TraceMeRecorder::Active(