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

For Runtime Metadata, HostTracer is not required, it supposed to only capture GPU events, CPU side of data are collected by other means than TraceMe.

Browse Source 
PiperOrigin-RevId: 315536038
Change-Id: Ie028fbeeaa07696eaf78e0393c52841903acea1f
This commit is contained in:
A. Unique TensorFlower 2020-06-09 12:25:55 -07:00 committed by TensorFlower Gardener
parent 9c236222b3
commit 1145cc3855
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/common_runtime/direct_session.cc
View File

@ -666,7 +666,10 @@ Status DirectSession::RunInternal(
std::unique_ptr<ProfilerSession> profiler_session;
if (run_options.trace_level() >= RunOptions::HARDWARE_TRACE) {
profiler_session = ProfilerSession::Create();
ProfileOptions options = ProfilerSession::DefaultOptions();
options.set_host_tracer_level(0);
options.set_device_type(ProfileOptions::GPU);
profiler_session = ProfilerSession::Create(options);
}
// Register this step with session's cancellation manager, so that