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

Disable test variants that time out in CI.

Browse Source 
PiperOrigin-RevId: 489214464
This commit is contained in:
Peter Hawkins 2022-11-17 08:13:29 -08:00 committed by jax authors
parent b0d5052928
commit ebee4f4bfd
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/BUILD
View File

@ -322,6 +322,7 @@ jax_test(
name = "lax_numpy_test",
srcs = ["lax_numpy_test.py"],
backend_tags = {
"cpu": ["noasan"], # Test times out.
"tpu": ["noasan"], # Test times out.
},
pjrt_c_api_bypass = True,
@ -486,6 +487,7 @@ jax_test(
"tpu": [
"cpu:8",
"noasan", # Times out.
"nomsan", # Times out.
"nodebug", # Times out.
"notsan", # Times out.
],
@ -555,6 +557,7 @@ jax_test(
backend_tags = {
"tpu": [
"noasan", # Times out.
"nmasan", # Times out.
"nodebug", # Times out.
"notsan", # Times out.
],
@ -639,10 +642,13 @@ jax_test(
args = ["--jax_enable_custom_prng=true"],
backend_tags = {
"cpu": [
"noasan",
"noasan", # Times out under asan/tsan.
"notsan",
], # Times out under asan/tsan.
"tpu": ["optonly"],
],
"tpu": [
"noasan", # Times out under asan/tsan.
"optonly",
],
},
main = "random_test.py",
shard_count = {