add test build file

wsmoses · wsmoses · commit ad97658da2d8 · 2024-02-20T11:24:27.000-05:00
diff --git a/test/BUILD b/test/BUILD
@@ -0,0 +1,22 @@
+load("@rules_python//python:py_test.bzl", "py_test")
+
+py_test(
+    name = "test",
+    srcs = [
+        "test.py",
+    ],
+    deps = [
+        "//src/enzyme_ad/jax:enzyme_jax_internal",
+    ],
+)
+
+py_test(
+    name = "bench_vs_xla",
+    srcs = [
+        "bench_vs_xla.py",
+    ],
+    deps = [
+        "//src/enzyme_ad/jax:enzyme_jax_internal",
+    ],
+)
+
