Disabled nn_test under asan on TPU as well, since it also times out

PiperOrigin-RevId: 660950262
jax-ml · Aug 8, 2024 · d8eafc8 · d8eafc8
1 parent efb7721
commit d8eafc8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/BUILD b/tests/BUILD
@@ -692,6 +692,9 @@ jax_test(
  "gpu": [
  "noasan", # Times out under asan.
  ],
+ "tpu": [
+ "noasan", # Times out under asan.
+ ],
  },
  shard_count = {
  "cpu": 10,