From 4f0f4f2501f4611cfec6ab676a1dab5963ce5cff Mon Sep 17 00:00:00 2001
From: Jed Brown <jed@jedbrown.org>
Date: Thu, 28 Mar 2024 22:46:31 -0600
Subject: [PATCH] ci: cache Enzyme and use rustbook@main

---
 .github/workflows/enzyme-ci.yml | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/enzyme-ci.yml b/.github/workflows/enzyme-ci.yml
index 85ad9e62066fb..c233b049e1d6d 100644
--- a/.github/workflows/enzyme-ci.yml
+++ b/.github/workflows/enzyme-ci.yml
@@ -35,6 +35,15 @@ jobs:
         with:
           path: build/build/x86_64-unknown-linux-gnu/llvm
           key: ${{ matrix.os }}-llvm-${{ steps.llvm-commit.outputs.HEAD }}
+      - name: Get Enzyme commit hash
+        id: enzyme-commit
+        run: echo "HEAD=$(git -C src/tools/enzyme rev-parse HEAD)" >> $GITHUB_OUTPUT
+      - name: Cache Enzyme
+        id: cache-enzyme
+        uses: actions/cache@v4
+        with:
+          path: build/build/x86_64-unknown-linux-gnu/enzyme
+          key: ${{ matrix.os }}-enzyme-${{ steps.enzyme-commit.outputs.HEAD }}
       - name: Build
         run: |
           mkdir -p build
@@ -47,7 +56,7 @@ jobs:
         uses: actions/checkout@v4
         with:
           repository: EnzymeAD/rustbook
-          ref: mdbook-test
+          ref: main
           path: rustbook
       - name: test Enzyme/rustbook
         working-directory: rustbook