diff --git a/tests/BUILD b/tests/BUILD index c020de74a..81822c5b6 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -337,6 +337,7 @@ jax_test( "tpu": 10, "iree": 10, }, + tags = ["noasan"], # Linking TF causes a linker OOM. deps = py_deps("pil") + py_deps("tensorflow_core"), ) @@ -1150,6 +1151,7 @@ jax_test( jax_test( name = "host_callback_to_tf_test", srcs = ["host_callback_to_tf_test.py"], + tags = ["noasan"], # Linking TF causes a linker OOM. deps = [ "//jax:experimental_host_callback", "//jax:ode", diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD index fd6ef2a1e..f5ab3d56d 100644 --- a/tests/pallas/BUILD +++ b/tests/pallas/BUILD @@ -93,6 +93,7 @@ jax_test( ], shard_count = 18, tags = [ + "noasan", # Times out. "nomsan", # Times out. "notsan", # Times out. ],