Print benchmark output if reporting is not requested.

annarev · tensorflower-gardener · commit 1e631f82a490 · 2017-05-17T14:32:35.000-07:00
PiperOrigin-RevId: 156354145
diff --git a/tensorflow/python/platform/benchmark.py b/tensorflow/python/platform/benchmark.py
@@ -73,11 +73,6 @@ def _global_report_benchmark(
                  cpu_time is not None else -1, throughput if
                  throughput is not None else -1, str(extras) if extras else "")
 
-  test_env = os.environ.get(TEST_REPORTER_TEST_ENV, None)
-  if test_env is None:
-    # Reporting was not requested
-    return
-
   entries = test_log_pb2.BenchmarkEntries()
   entry = entries.entry.add()
   entry.name = name
@@ -96,6 +91,12 @@ def _global_report_benchmark(
       else:
         entry.extras[k].string_value = str(v)
 
+  test_env = os.environ.get(TEST_REPORTER_TEST_ENV, None)
+  if test_env is None:
+    # Reporting was not requested, just print the proto
+    print(str(entries))
+    return
+
   serialized_entry = entries.SerializeToString()
 
   mangled_name = name.replace("/", "__")
