mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Disable more tests that are flaky in CI.

Browse Source 
PiperOrigin-RevId: 529724306
This commit is contained in:
Peter Hawkins 2023-05-05 08:32:56 -07:00 committed by jax authors
parent a6254c75e0
commit e8c735125c
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/BUILD
View File

@ -630,7 +630,10 @@ jax_test(
srcs = ["random_test.py"],
backend_tags = {
"cpu": ["notsan"], # Times out
"tpu": ["optonly"],
"tpu": [
"optonly",
"nomsan", # Times out
],
},
shard_count = {
"cpu": 30,
@ -775,6 +778,7 @@ jax_test(
srcs = ["sparsify_test.py"],
args = ["--jax_bcoo_cusparse_lowering=true"],
backend_tags = {
"cpu": ["noasan"], # Times out under asan
"tpu": [
"noasan", # Times out under asan.
],