[host_callback] Remove obsolete tests.

Removing tests that only work in legacy mode and with outfeed. PiperOrigin-RevId: 681435113
jax-ml · Oct 2, 2024 · b8a066a · b8a066a
1 parent 8df1623
commit b8a066a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1,221 deletions.
diff --git a/tests/BUILD b/tests/BUILD
@@ -1142,23 +1142,6 @@ jax_multiplatform_test(
     deps = ["//jax:ode"],
 )
 
-jax_multiplatform_test(
-    name = "host_callback_outfeed_test",
-    srcs = ["host_callback_test.py"],
-    args = ["--jax_host_callback_outfeed=true"],
-    shard_count = {
-        "tpu": 5,
-    },
-    tags = [
-        "noasan",  # Times out.
-    ],
-    deps = [
-        "//jax:experimental",
-        "//jax:experimental_host_callback",
-        "//jax:ode",
-    ],
-)
-
 jax_multiplatform_test(
     name = "host_callback_test",
     srcs = ["host_callback_test.py"],