From 3df3300d5876921e778a2559a39a41cde537f1cf Mon Sep 17 00:00:00 2001 From: Haoyu Zhang Date: Thu, 30 Apr 2020 11:51:27 -0700 Subject: [PATCH] Remove async runs of cluster update with pending nodes tests. Cluster update in parallel with async nodes is not supported yet. PiperOrigin-RevId: 309269820 Change-Id: I825049e44f7098bcf323b381275b384b6636b1ff --- tensorflow/python/eager/BUILD | 1 - tensorflow/python/eager/remote_cluster_test.py | 3 --- 2 files changed, 4 deletions(-) diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index b50d06c0d53..c08cb8cc1c3 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -960,7 +960,6 @@ cuda_py_test( shard_count = 8, tags = [ "no_oss", # This test launches local server - "notap", # TODO(b/155407846) ], deps = [ "//tensorflow/python:array_ops", diff --git a/tensorflow/python/eager/remote_cluster_test.py b/tensorflow/python/eager/remote_cluster_test.py index a0ce235c282..025899e271e 100644 --- a/tensorflow/python/eager/remote_cluster_test.py +++ b/tensorflow/python/eager/remote_cluster_test.py @@ -300,7 +300,6 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase): y = worker_fn(x1) np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy()) - @test_util.run_in_async_and_sync_mode def testPendingNodesServerReplaced(self): """Update cluster when nodes are still pending on remote workers.""" with ops.device(self.device_local): @@ -371,10 +370,8 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase): for result in t1_results + t2_results: np.testing.assert_array_equal([[2, 2], [2, 2]], result) - @test_util.run_in_async_and_sync_mode def testMultiThreadPendingNodesLockFree(self): """Update cluster when other remote function calls are being launched.""" - self.skipTest("b/154053481") with ops.device(self.device_t1): x1 = array_ops.ones([2, 2])