commit f132de21166a35ea541a82907f47f3a794d34f9b
author Sanjoy Das <sanjoy@google.com> Mon Oct 07 08:59:04 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Oct 07 09:08:18 2019 -0700
tree 3e41a31d445f0ec9262e69d19a81909d06ec7f35
parent 3d68d5909d2f1c9daf23e38f3aeed31f7f4d0059

VLOG(1) when a cluster goes megamorphic

PiperOrigin-RevId: 273302762

tensorflow/compiler/jit/xla_compilation_cache.cc

diff --git a/tensorflow/compiler/jit/xla_compilation_cache.cc b/tensorflow/compiler/jit/xla_compilation_cache.cc
index 62b0757..663a205 100644
--- a/tensorflow/compiler/jit/xla_compilation_cache.cc
+++ b/tensorflow/compiler/jit/xla_compilation_cache.cc
@@ -189,7 +189,7 @@
                      out_compilation_result, out_executable);
 }
 
-static bool IsMegamorphic(int64 compile_count, int64 execution_count) {
+static bool ShouldBeMegamorphic(int64 compile_count, int64 execution_count) {
   const int64 kCompileThreshold = 10;
   const int64 kMinExecutionsPerCompile = 50;
 
@@ -296,9 +296,15 @@
 
     // The is_megamorphic bit is "sticky".  We assume clusters that have been
     // observed to be megamorphic once stay megamorphic forever.
-    it->second.is_megamorphic |=
-        IsMegamorphic(/*compile_count=*/it->second.compile_count,
-                      /*execution_count=*/it->second.execution_count);
+    if (!it->second.is_megamorphic &&
+        ShouldBeMegamorphic(/*compile_count=*/it->second.compile_count,
+                            /*execution_count=*/it->second.execution_count)) {
+      VLOG(1) << "Marking " << function.name()
+              << " as megamorphic, compile_count=" << it->second.compile_count
+              << " execution_count=" << it->second.execution_count;
+      it->second.is_megamorphic = true;
+    }
+
     is_megamorphic = it->second.is_megamorphic;
   }