We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6d9bb3a commit ecddbbbCopy full SHA for ecddbbb
patches/jax.patch
@@ -1,15 +1,19 @@
1
diff --git a/jaxlib/mosaic/BUILD b/jaxlib/mosaic/BUILD
2
-index 088411939..d11fa28d0 100644
+index 088411939..598118390 100644
3
--- a/jaxlib/mosaic/BUILD
4
+++ b/jaxlib/mosaic/BUILD
5
-@@ -54,7 +54,11 @@ cc_library(
+@@ -54,7 +54,15 @@ cc_library(
6
"dialect/tpu/tpu_dialect.h",
7
"dialect/tpu/util.h",
8
"dialect/tpu/vreg_util.h",
9
- ],
10
+ ] + glob(
11
+ [
12
+ "dialect/tpu/transforms/*.h",
13
++ ],
14
++ exclude = [
15
++ "dialect/tpu/transforms/serde.h",
16
++ "dialect/tpu/transforms/linalg_vectorization.h",
17
+ ])
18
+ ,
19
# compatible with libtpu
0 commit comments