diff --git a/.github/workflows/BuildCWrapper.yaml b/.github/workflows/BuildCWrapper.yaml index 0f1d585..5d66e79 100644 --- 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