From 5a93e15bd782f707a3dd0dd78702c160b7689897 Mon Sep 17 00:00:00 2001
From: Shuhan Ding <sding23@apple.com>
Date: Tue, 12 Mar 2024 17:17:20 -0700
Subject: [PATCH] add to tests/BUILD

---
 tests/BUILD | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/BUILD b/tests/BUILD
index 9c8ca9310..62fb9fb8a 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -542,6 +542,21 @@ jax_test(
     ] + py_deps("numpy"),
 )
 
+jax_test(
+    name = "lax_metal_test",
+    srcs = ["lax_metal_test.py"],
+    tags = ["notap"],
+    disable_backends = [
+        "cpu",
+        "gpu",
+        "tpu",
+    ],
+    deps = [
+        "//jax:internal_test_util",
+        "//jax:lax_reference",
+    ] + py_deps("numpy"),
+)
+
 jax_test(
     name = "lax_autodiff_test",
     srcs = ["lax_autodiff_test.py"],