Internal change.
PiperOrigin-RevId: 236374895
This commit is contained in:
parent
9dc8d7323e
commit
234a8d2440
@ -2630,6 +2630,7 @@ cuda_py_test(
|
|||||||
"//tensorflow/python/eager:context",
|
"//tensorflow/python/eager:context",
|
||||||
],
|
],
|
||||||
flaky = 1, # create_local_cluster sometimes times out.
|
flaky = 1, # create_local_cluster sometimes times out.
|
||||||
|
shard_count = 10,
|
||||||
tags = ["no_rocm"],
|
tags = ["no_rocm"],
|
||||||
xla_enable_strict_auto_jit = True,
|
xla_enable_strict_auto_jit = True,
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user