Dropping tests for non-EagerService code path in remote_cluster_test.
PiperOrigin-RevId: 302922575 Change-Id: I8b335e1f0ff8ed5a3f47b1fbc77302e500d7de37
This commit is contained in:
parent
858bd4506a
commit
f1e0098f2a
@ -891,7 +891,7 @@ cuda_py_test(
|
||||
srcs = ["remote_cluster_test.py"],
|
||||
grpc_enabled = True,
|
||||
python_version = "PY3",
|
||||
shard_count = 16,
|
||||
shard_count = 8,
|
||||
tags = [
|
||||
"no_oss", # This test launches local server
|
||||
"notsan", # TODO(b/152075365)
|
||||
|
@ -495,28 +495,6 @@ class DynamicClusterTest(test.TestCase, parameterized.TestCase):
|
||||
context.check_alive("/job:remote_device/replica:0/task:10")
|
||||
|
||||
|
||||
class DynamicClusterWithoutLazyRemoteInputsCopyTest(DynamicClusterTest):
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
super(DynamicClusterWithoutLazyRemoteInputsCopyTest, cls).setUpClass()
|
||||
context._reset_context()
|
||||
context.context().lazy_remote_inputs_copy = False
|
||||
|
||||
@classmethod
|
||||
def tearDownClass(cls):
|
||||
super(DynamicClusterWithoutLazyRemoteInputsCopyTest, cls).tearDownClass()
|
||||
context._reset_context()
|
||||
context.context().lazy_remote_inputs_copy = True
|
||||
|
||||
# TODO(haoyuzhang): When lazyh remote inputs copy is disabled, we use the
|
||||
# WorkerService RunGraph request to execute component functions in distributed
|
||||
# function execution. We currently do not have access control in WorkerService
|
||||
# to allow concurrent cluster update and function execution.
|
||||
def testMultiThreadPendingNodesLockFree(self):
|
||||
self.skipTest("Unsupported case")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
ops.enable_eager_execution()
|
||||
test.main()
|
||||
|
Loading…
x
Reference in New Issue
Block a user