Fix

EnzymeAD · Aug 25, 2024 · fb461ff · fb461ff
1 parent de342ac
commit fb461ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -55,5 +55,7 @@ steps:
       cp bazel-bin/*.whl .
     artifact_paths:
       - "*.whl"
+      - "bazel-out/k8*/testlogs/test/llama/test.log"
+      - "bazel-out/k8*/testlogs/test/llama/bench_vs_xla.log"
 
     timeout_in_minutes: 180
diff --git a/workspace.bzl b/workspace.bzl
@@ -1,7 +1,7 @@
 JAX_COMMIT = "3713b966c2a868e948a663193282deba7ba14842"
 JAX_SHA256 = "6b0265a1c58b6c050f334e6b91ce1a69b47b1144670b87bcc30fc90172b5cbf4"
 
-ENZYME_COMMIT = "cc65abdb55e5e2e142d773e0508e1083d4b3ac52"
+ENZYME_COMMIT = "786a76c0ecb8c0f7df887ceba8a6defb866eb54f"
 ENZYME_SHA256 = ""
 
 XLA_PATCHES = [