feat: cache models weight for coverage and docs

kornia · Aug 18, 2024 · dfd07e8 · dfd07e8
1 parent ad135d8
commit dfd07e8
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 0 deletions.
diff --git a/.github/workflows/coverage.yml b/.github/workflows/coverage.yml
@@ -21,6 +21,15 @@ on:
         required: false
         type: string
         default: ${{ github.sha }}
+      cache-path:
+        required: false
+        type: string
+      cache-key:
+        required: false
+        type: string
+      cache-restore-keys:
+        required: false
+        type: string
 
 jobs:
   slow:
@@ -32,6 +41,14 @@ jobs:
         pytorch-dtype: ['float32', 'float64']
 
     steps:
+    - name: Restore cache
+      if: "${{ inputs.cache-path != '' }}"
+      uses: actions/cache/restore@v4
+      with:
+        path: ${{ inputs.cache-path }}
+        key: ${{ inputs.cache-key }}
+        restore-keys: ${{ inputs.cache-restore-keys }}
+
     - uses: kornia/workflows/.github/actions/[email protected]
       with:
         os: ${{ inputs.os }}
@@ -52,6 +69,14 @@ jobs:
         pytorch-dtype: ['float32']
 
     steps:
+    - name: Restore cache
+      if: "${{ inputs.cache-path != '' }}"
+      uses: actions/cache/restore@v4
+      with:
+        path: ${{ inputs.cache-path }}
+        key: ${{ inputs.cache-key }}
+        restore-keys: ${{ inputs.cache-restore-keys }}
+
     - uses: kornia/workflows/.github/actions/[email protected]
       with:
         os: ${{ inputs.os }}
@@ -72,6 +97,14 @@ jobs:
         pytorch-dtype: ['float32', 'float64']
 
     steps:
+    - name: Restore cache
+      if: "${{ inputs.cache-path != '' }}"
+      uses: actions/cache/restore@v4
+      with:
+        path: ${{ inputs.cache-path }}
+        key: ${{ inputs.cache-key }}
+        restore-keys: ${{ inputs.cache-restore-keys }}
+
     - uses: kornia/workflows/.github/actions/[email protected]
       with:
         os: ${{ inputs.os }}

diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -21,6 +21,15 @@ on:
         required: false
         type: string
         default: ${{ github.sha }}
+      cache-path:
+        required: false
+        type: string
+      cache-key:
+        required: false
+        type: string
+      cache-restore-keys:
+        required: false
+        type: string
 
 jobs:
   test-and-build:
@@ -72,6 +81,14 @@ jobs:
         pytorch-version: ${{ matrix.pytorch-version }}
         ref: ${{ inputs.ref }}
 
+    - name: Restore cache
+      if: "${{ inputs.cache-path != '' }}"
+      uses: actions/cache/restore@v4
+      with:
+        path: ${{ inputs.cache-path }}
+        key: ${{ inputs.cache-key }}
+        restore-keys: ${{ inputs.cache-restore-keys }}
+
     - name: Run doctest
       shell: bash -l {0}
       run: make doctest