commit 163a824e40bb3135bb96044c55e06361e738e544
author Gaurav Jain <gjn@google.com> Wed Sep 11 20:24:59 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Sep 11 20:31:03 2019 -0700
tree 289451f02d8934122ffdb71bc66d2ae9d9343f95
parent e8755860f875d741f8c2f4a56b7f6135562748ab

Disable remote eager benchmarks

PiperOrigin-RevId: 268599883

tensorflow/python/eager/benchmarks_test.py

diff --git a/tensorflow/python/eager/benchmarks_test.py b/tensorflow/python/eager/benchmarks_test.py
index 87ce601..3b65d12 100644
--- a/tensorflow/python/eager/benchmarks_test.py
+++ b/tensorflow/python/eager/benchmarks_test.py
@@ -1115,7 +1115,8 @@
         wall_time=mean_us,
         extras={"examples_per_sec": num_iters / total_time})
 
-  def benchmark_send_mirroring_off(self):
+  # TODO(b/136184459): Re-enabled once crash is fixed
+  def _DISABLED_benchmark_send_mirroring_off(self):
     remote.connect_to_remote_host(self._cached_server_target1)
 
     x = random_ops.random_uniform((2, 2)).cpu()
@@ -1135,7 +1136,8 @@
     # executed when their corresponding device and manager are still available.
     gc.collect()
 
-  def benchmark_send_mirroring_on(self):
+  # TODO(b/136184459): Re-enabled once crash is fixed
+  def _DISABLED_benchmark_send_mirroring_on(self):
     remote.connect_to_remote_host(self._cached_server_target1)
 
     x = random_ops.random_uniform((2, 2)).cpu()
@@ -1155,7 +1157,8 @@
     # executed when their corresponding device and manager are still available.
     gc.collect()
 
-  def benchmark_worker_mirroring_off(self):
+  # TODO(b/136184459): Re-enabled once crash is fixed
+  def _DISABLED_benchmark_worker_mirroring_off(self):
     remote.connect_to_remote_host(
         [self._cached_server_target1, self._cached_server_target2])
 
@@ -1177,7 +1180,8 @@
     # executed when their corresponding device and manager are still available.
     gc.collect()
 
-  def benchmark_worker_mirroring_on(self):
+  # TODO(b/136184459): Re-enabled once crash is fixed
+  def _DISABLED_benchmark_worker_mirroring_on(self):
     remote.connect_to_remote_host(
         [self._cached_server_target1, self._cached_server_target2])