Disable more tests under tsan/asan.

PiperOrigin-RevId: 488406459
This commit is contained in:
Peter Hawkins 2022-11-14 10:33:53 -08:00 committed by jax authors
parent 7c6a65bee2
commit da130cb074

View File

@ -606,6 +606,9 @@ jax_test(
jax_test(
name = "random_test",
srcs = ["random_test.py"],
backend_tags = {
"cpu": ["notsan"], # Times out
},
shard_count = {
"cpu": 30,
"gpu": 30,
@ -705,7 +708,10 @@ jax_test(
"tpu": 50,
"iree": 10,
},
tags = ["noasan"], # Test times out under asan.
tags = [
"noasan",
"notsan",
], # Test times out under asan/tsan.
deps = [
"//jax:experimental_sparse",
] + py_deps("scipy"),