add missing cpu dep

EnzymeAD · Feb 8, 2025 · 2091257 · 2091257
1 parent 2601466
commit 2091257
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/enzyme_ad/jax/BUILD b/src/enzyme_ad/jax/BUILD
@@ -595,6 +595,7 @@ pybind_extension(
     ],
     visibility = ["//visibility:public"],
     deps = [
+        ":cpu",
         ":gpu",
         ":clang_compile",
         ":compile_with_xla",