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 tests that time out in CI.

Browse Source 
PiperOrigin-RevId: 532792740
This commit is contained in:
Peter Hawkins 2023-05-17 08:15:32 -07:00 committed by jax authors
parent eaf7eb2626
commit e6628e2e72
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/BUILD
View File

@ -416,6 +416,9 @@ jax_test(
jax_test(
name = "lax_test",
srcs = ["lax_test.py"],
backend_tags = {
"tpu": ["noasan"], # Times out.
},
shard_count = {
"cpu": 40,
"gpu": 40,
@ -629,7 +632,10 @@ jax_test(
name = "random_test",
srcs = ["random_test.py"],
backend_tags = {
"cpu": ["notsan"], # Times out
"cpu": [
"notsan", # Times out
"nomsan", # Times out
],
"tpu": [
"optonly",
"nomsan", # Times out