[Mosaic] Don't link CAPIIR into _tpu_ext, link into jaxlib_mlir_capi_…

…shared_library instead PiperOrigin-RevId: 579881376
skondla · Nov 6, 2023 · c186928 · c186928
1 parent 41430ac
commit c186928
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jaxlib/mlir/_mlir_libs/BUILD.bazel b/jaxlib/mlir/_mlir_libs/BUILD.bazel
@@ -98,7 +98,7 @@ py_extension(
     deps = [
         ":jaxlib_mlir_capi_shared_library",
         "//jaxlib/mosaic:tpu_dialect_capi_headers",
-        "@llvm-project//mlir:CAPIIR",
+        "@llvm-project//mlir:CAPIIRHeaders",
         "@llvm-project//mlir:MLIRBindingsPythonHeadersAndDeps",
         "@pybind11",
     ],
@@ -242,6 +242,7 @@ cc_library(
         ":jax_dialects_capi",
         "//jaxlib/mosaic:tpu_dialect_capi_objects",
         "@llvm-project//mlir:CAPIArithObjects",
+        "@llvm-project//mlir:CAPIIRObjects",
         "@llvm-project//mlir:CAPIMathObjects",
         "@llvm-project//mlir:CAPIMemRefObjects",
         "@llvm-project//mlir:CAPISparseTensorObjects",