commit 056b657d58f781a16351ab78606c2702aef838f4
author Reed Wanderman-Milne <reedwm@google.com> Tue Jun 30 11:22:54 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Jun 30 11:30:29 2020 -0700
tree 96e8dab107690be2e6868509f813197b88cb9cb2
parent 4f5ffbca7f4c63fa085c12464891250766aa652a

Do not log in DefaultLogger.

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

tensorflow/core/kernels/gpu_utils.cc

diff --git a/tensorflow/core/kernels/gpu_utils.cc b/tensorflow/core/kernels/gpu_utils.cc
index 5c51c2e..7da1963 100644
--- a/tensorflow/core/kernels/gpu_utils.cc
+++ b/tensorflow/core/kernels/gpu_utils.cc
@@ -163,6 +163,7 @@
   for (const auto& result : results) {
     *log.add_results() = result;
   }
+  VLOG(2) << log.DebugString();
   Logger::GetSingleton()->LogProto(log);
 }
 
@@ -209,6 +210,7 @@
   for (const auto& result : results) {
     *log.add_results() = result;
   }
+  VLOG(2) << log.DebugString();
   Logger::GetSingleton()->LogProto(log);
 }
 

tensorflow/core/platform/logger.cc

diff --git a/tensorflow/core/platform/logger.cc b/tensorflow/core/platform/logger.cc
index 32fdaa5..653c4fb 100644
--- a/tensorflow/core/platform/logger.cc
+++ b/tensorflow/core/platform/logger.cc
@@ -25,9 +25,7 @@
 
 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 {}
 };