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

Do not log in DefaultLogger.

Browse Source 
Instead, directly log in gpu_utils.cc, so that TF_CPP_VMODULE can be used to show the autotune results.

PiperOrigin-RevId: 319061148
Change-Id: Idd27bb2dd19353f099f3f97248f29b7a1a3f6fdb
This commit is contained in:
Reed Wanderman-Milne 2020-06-30 11:22:54 -07:00 committed by TensorFlower Gardener
parent 4f5ffbca7f
commit 056b657d58
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/gpu_utils.cc
View File

@ -163,6 +163,7 @@ void LogConvAutotuneResults(se::dnn::ConvolutionKind kind,
for (const auto& result : results) {
*log.add_results() = result;
}
VLOG(2) << log.DebugString();
Logger::GetSingleton()->LogProto(log);
}
@ -209,6 +210,7 @@ void LogFusedConvForwardAutotuneResults(
for (const auto& result : results) {
*log.add_results() = result;
}
VLOG(2) << log.DebugString();
Logger::GetSingleton()->LogProto(log);
}

4
tensorflow/core/platform/logger.cc
View File

@ -25,9 +25,7 @@ namespace {
class DefaultLogger : public Logger {
private:
void DoLogProto(google::protobuf::Any* proto) override {
VLOG(2) << proto->ShortDebugString();
}
void DoLogProto(google::protobuf::Any* proto) override {}
void DoFlush() override {}
};