mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tag several tests as optonly to prevent test timeouts in debug mode CI builds.

Browse Source 
PiperOrigin-RevId: 488950972
This commit is contained in:
Peter Hawkins 2022-11-16 08:46:57 -08:00 committed by jax authors
parent f2bd1afb7e
commit 51c69ac594
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/BUILD
View File

@ -621,6 +621,7 @@ jax_test(
srcs = ["random_test.py"],
backend_tags = {
"cpu": ["notsan"], # Times out
"tpu": ["optonly"],
},
shard_count = {
"cpu": 30,
@ -641,6 +642,7 @@ jax_test(
"noasan",
"notsan",
], # Times out under asan/tsan.
"tpu": ["optonly"],
},
main = "random_test.py",
shard_count = {