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 sanitizer builds that are timing out or that are incompatible with test targets.

Browse Source 
PiperOrigin-RevId: 488919571
This commit is contained in:
Peter Hawkins 2022-11-16 06:00:07 -08:00 committed by jax authors
parent 5675b4572e
commit 0548c2d23b
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/BUILD
View File

@ -117,7 +117,10 @@ jax_test(
name = "fft_test",
srcs = ["fft_test.py"],
backend_tags = {
"tpu": ["notsan"], # Times out on TPU with tsan.
"tpu": [
"noasan",
"notsan",
], # Times out on TPU with asan/tsan.
},
shard_count = {
"tpu": 20,
@ -555,6 +558,7 @@ jax_test(
"nodebug", # Times out.
"notsan", # Times out.
],
"cpu": ["notsan"], # Times out
},
pjrt_c_api_bypass = True,
shard_count = {
@ -650,6 +654,12 @@ jax_test(
jax_test(
name = "scipy_fft_test",
srcs = ["scipy_fft_test.py"],
backend_tags = {
"tpu": [
"noasan",
"notsan",
], # Times out on TPU with asan/tsan.
},
shard_count = 4,
)