From 12c8755ecfba48670c50bde047024db6e47ce5d9 Mon Sep 17 00:00:00 2001 From: Haoyu Zhang Date: Fri, 20 Nov 2020 12:09:41 -0800 Subject: [PATCH] Enable cluster_coordinator_test on OSS. PiperOrigin-RevId: 343539254 Change-Id: I254c9f60c19ef39408e6d7982638a477baacda08 --- tensorflow/python/distribute/coordinator/BUILD | 1 + tensorflow/python/distribute/multi_worker_test_base.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/distribute/coordinator/BUILD b/tensorflow/python/distribute/coordinator/BUILD index 9c3c5bd4920..10c7c1dae84 100644 --- a/tensorflow/python/distribute/coordinator/BUILD +++ b/tensorflow/python/distribute/coordinator/BUILD @@ -40,6 +40,7 @@ tf_py_test( python_version = "PY3", shard_count = 50, tags = [ + "no_oss", # TODO(b/162119374) "notsan", # TODO(b/171040359): Flaky timeout, even if maximum shards ], deps = [ diff --git a/tensorflow/python/distribute/multi_worker_test_base.py b/tensorflow/python/distribute/multi_worker_test_base.py index ccd961d1861..e07e19f621b 100644 --- a/tensorflow/python/distribute/multi_worker_test_base.py +++ b/tensorflow/python/distribute/multi_worker_test_base.py @@ -166,8 +166,8 @@ def create_in_process_cluster(num_workers, # The cluster may hang if workers don't have enough inter_op threads. See # b/172296720 for more details. - if multiprocessing.cpu_count() < num_workers + 1: - worker_config.inter_op_parallelism_threads = num_workers + 1 + if multiprocessing.cpu_count() < 4: + worker_config.inter_op_parallelism_threads = 4 # Enable collective ops which has no impact on non-collective ops. # TODO(yuefengz, tucker): removing this after we move the initialization of