Increase shard count for remote_execution_test and reenable tsan test.

PiperOrigin-RevId: 300404009
Change-Id: I6874fd4fcc07ae32212f5f73eab4631f0938d1c8
This commit is contained in:
Haoyu Zhang 2020-03-11 14:12:47 -07:00 committed by TensorFlower Gardener
parent 205b3b5e8a
commit 604ebadb89

View File

@ -873,10 +873,9 @@ cuda_py_test(
srcs = ["remote_execution_test.py"], srcs = ["remote_execution_test.py"],
grpc_enabled = True, grpc_enabled = True,
python_version = "PY3", python_version = "PY3",
shard_count = 2, shard_count = 8,
tags = [ tags = [
"no_oss", # This test launches local server "no_oss", # This test launches local server
"notsan", # TODO(b/151240519) Re-enable.
], ],
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",