experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable cluster_coordinator_test on OSS.

Browse Source 
PiperOrigin-RevId: 343539254
Change-Id: I254c9f60c19ef39408e6d7982638a477baacda08
This commit is contained in:
Haoyu Zhang 2020-11-20 12:09:41 -08:00 committed by TensorFlower Gardener
parent 5abf12375e
commit 12c8755ecf
2 changed files with 3 additions and 2 deletions
tensorflow/python/distribute
coordinator
BUILD
multi_worker_test_base.py

1
tensorflow/python/distribute/coordinator/BUILD
View File

@ -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 = [

4
tensorflow/python/distribute/multi_worker_test_base.py
View File

@ -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