save cache

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

.github/workflows/_build.yml

@@ -117,11 +117,10 @@ jobs:
           python -c "from torch.utils import cpp_extension; print (cpp_extension.CUDA_HOME)"
 
       - name: Cache build dir
-        uses: actions/cache@v4
+        uses: actions/cache/restore@v4
         with:
           path: build
           key: build-${{ inputs.release-version }}-${{ inputs.python-version }}-${{ inputs.cuda-version }}-${{ inputs.torch-version }}-${{ inputs.cxx11-abi }}
-          save-always: true
 
       - name: Build wheel
         id: build_wheel
@@ -158,6 +157,13 @@ jobs:
           # Do not fail the job if timeout killed the build
           exit 0
 
+      - name: Save cache
+        if: always()
+        uses: actions/cache/save@v4
+        with:
+          key: build-${{ inputs.release-version }}-${{ inputs.python-version }}-${{ inputs.cuda-version }}-${{ inputs.torch-version }}-${{ inputs.cxx11-abi }}
+          path: build
+
       - name: Log Built Wheels
         run: |
           ls dist