Increase shard_count of strategy_gather_test to avoid timeout:
PiperOrigin-RevId: 337996150 Change-Id: Ifa68a95b8a21f88c5e7fe9dd5f8796afec37a2e3
This commit is contained in:
parent
bb3dda7877
commit
d9e09b0723
@ -1760,7 +1760,7 @@ distribute_py_test(
|
||||
srcs = ["strategy_gather_test.py"],
|
||||
disable_mlir_bridge = False,
|
||||
python_version = "PY3",
|
||||
shard_count = 2,
|
||||
shard_count = 4,
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
"notsan", # TODO(b/160006974)
|
||||
|
Loading…
Reference in New Issue
Block a user