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

Disable tests that timeout in debug mode in CI

Browse Source 
PiperOrigin-RevId: 476157051
This commit is contained in:
Tyler Augustine 2022-09-22 11:44:22 -07:00 committed by jax authors
parent a157982e8c
commit d52de206cb
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/BUILD
View File

@ -477,6 +477,7 @@ jax_test(
"tpu": [
"cpu:8",
"noasan", # Times out.
"nodebug", # Times out.
"notsan", # Times out.
],
},
@ -535,9 +536,10 @@ jax_test(
srcs = ["pmap_test.py"],
backend_tags = {
"tpu": [
"noasan",
"notsan",
], # Times out under asan/tsan.
"noasan", # Times out.
"nodebug", # Times out.
"notsan", # Times out.
],
},
pjrt_c_api_bypass = True,
shard_count = {