diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index 04cc0b0e..8fc856eb 100644 --- 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 index 0c3a2dc8..76e85d92 100644 --- 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 = [