From 49eb32ef52f7b2490ae7887a20cb928a70e728c2 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Sun, 16 Nov 2025 11:33:33 -0600 Subject: [PATCH] Don't build tpu dialect while broken upstream --- deps/ReactantExtra/API.cpp | 3 ++- deps/ReactantExtra/BUILD | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp index f92b54a2af..efa1a96db9 100644 --- a/deps/ReactantExtra/API.cpp +++ b/deps/ReactantExtra/API.cpp @@ -167,7 +167,8 @@ #include "xla/stream_executor/device_description.h" #endif -#include "jaxlib/mosaic/dialect/tpu/tpu_dialect.h" +// Broken upstream x/ref https://github.com/jax-ml/jax/issues/33344 +// #include "jaxlib/mosaic/dialect/tpu/tpu_dialect.h" #include "llvm/Support/ExtensibleRTTI.h" #include diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD index 7254ceb3bc..eaee062831 100644 --- a/deps/ReactantExtra/BUILD +++ b/deps/ReactantExtra/BUILD @@ -1175,7 +1175,9 @@ cc_library( "@com_google_absl//absl/log:globals", "@llvm-project//mlir:CAPIIRObjects", "@llvm-project//mlir:CAPILLVMObjects", - "@jax//jaxlib/mosaic:tpu_dialect_capi_objects", + + # Broken upstream x/ref https://github.com/jax-ml/jax/issues/33344 + # "@jax//jaxlib/mosaic:tpu_dialect_capi_objects", "@jax//jaxlib/triton:triton_dialect_capi_objects", "@xla//xla/pjrt/plugin/xla_cpu:xla_cpu_pjrt_client", "@stablehlo//:linalg_passes",