CI: Cache Torch

FluxML · Apr 29, 2024 · 0ef32d1 · 0ef32d1
1 parent aa10911
commit 0ef32d1
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/.github/workflows/BuildCWrapper.yaml b/.github/workflows/BuildCWrapper.yaml
@@ -32,7 +32,15 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - run: ./.dev/install_build_deps.sh $GCC_VERSION
-      - run: ./.dev/install_torch.sh $TORCH_VARIANT $TORCH_VERSION
+      - name: Cache Torch
+        id: cache-torch
+        uses: actions/cache@v4
+        with:
+          path: /usr/local/libtorch
+          key: torch-${{ env.TORCH_VERSION }}-${{ runner.arch }}-${{ runner.os }}-cpu
+      - name: Install Torch
+        if: steps.cache-torch.outputs.cache-hit != 'true'
+        run: ./.dev/install_torch.sh $TORCH_VARIANT $TORCH_VERSION
       - name: build
         run: |
           export CMAKE_PREFIX_PATH=/usr/local/libtorch
@@ -84,7 +92,15 @@ jobs:
         run: |
           ./.dev/install_cuda_sdk.sh $CUDA_VERSION
           ./.dev/install_cudnn.sh $CUDA_VERSION $CUDNN_VERSION
-      - run: |
+      - name: Cache Torch
+        id: cache-torch
+        uses: actions/cache@v4
+        with:
+          path: /usr/local/libtorch
+          key: torch-${{ matrix.torch_version }}-${{ runner.arch }}-${{ runner.os }}-cuda${{ matrix.cuda_version }}
+      - name: Install Torch
+        if: steps.cache-torch.outputs.cache-hit != 'true'
+        run: |
           export TORCH_VARIANT="cu$(echo $CUDA_VERSION | cut -d . -f 1-2 | tr -d '.')"
           ./.dev/install_torch.sh $TORCH_VARIANT $TORCH_VERSION
       - name: build