fix

wsmoses · wsmoses · commit 1e3dcbd010ef · 2025-11-16T01:26:02.000-06:00
diff --git a/src/enzyme_ad/jax/BUILD b/src/enzyme_ad/jax/BUILD
@@ -1280,7 +1280,8 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
 
         # Mosaic
-        "@jax//jaxlib/mosaic:tpu_dialect",
+        # Upstream is broken, re-enable when working
+	# "@jax//jaxlib/mosaic:tpu_dialect",
     ],
 )
 

Original file line number	Diff line number	Diff line change
`@@ -1280,7 +1280,8 @@ cc_library(`
`1280`	`1280`	`"@com_google_absl//absl/status:statusor",`
`1281`	`1281`
`1282`	`1282`	`# Mosaic`
`1283`		`- "@jax//jaxlib/mosaic:tpu_dialect",`
	`1283`	`+ # Upstream is broken, re-enable when working`
	`1284`	`+ # "@jax//jaxlib/mosaic:tpu_dialect",`
`1284`	`1285`	`],`
`1285`	`1286`	`)`
`1286`	`1287`