[JAX] Disable qdwh_test in asan/msan/tsan configurations on TPU.

This test is flakily timing out in CI, the sanitizers probably push the test over its time bound.

PiperOrigin-RevId: 642695381
This commit is contained in:
Peter Hawkins 2024-06-12 12:16:06 -07:00 committed by jax authors
parent 987a2f0850
commit 339027d7ab

View File

@ -784,6 +784,13 @@ jax_test(
jax_test(
name = "qdwh_test",
srcs = ["qdwh_test.py"],
backend_tags = {
"tpu": [
"noasan", # Times out
"nomsan", # Times out
"notsan", # Times out
],
},
shard_count = 10,
)