Skip to content

Commit

Permalink
Merge pull request #3375 from ollama/mxyng/conditional-generate
Browse files Browse the repository at this point in the history
only generate cuda/rocm when changes to llm detected
  • Loading branch information
mxyng committed Mar 27, 2024
2 parents 913306f + 5b0c48d commit c363282
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions .github/workflows/test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,24 @@ on:
- '!README.md'

jobs:
changes:
runs-on: ubuntu-latest
outputs:
GENERATE_CUDA: ${{ steps.changes.outputs.GENERATE_CUDA }}
GENERATE_ROCM: ${{ steps.changes.outputs.GENERATE_ROCM }}
steps:
- id: changes
run: |
changed() {
git diff-tree -r --no-commit-id --name-only $GITHUB_BASE_REF $GITHUB_HEAD_REF \
| xargs python3 -c "import sys; print(any([x.startswith('$1') for x in sys.argv[1:]]))"
}
{
echo GENERATE_CUDA=$(changed llm)
echo GENERATE_ROCM=$(changed llm)
} >>$GITHUB_OUTPUT
generate:
strategy:
matrix:
Expand Down Expand Up @@ -46,6 +64,8 @@ jobs:
name: ${{ matrix.os }}-${{ matrix.arch }}-libraries
path: llm/llama.cpp/build/**/lib/*
generate-cuda:
needs: [changes]
if: ${{ needs.changes.outputs.GENERATE_CUDA == 'True' }}
strategy:
matrix:
cuda-version:
Expand Down Expand Up @@ -75,6 +95,8 @@ jobs:
name: cuda-${{ matrix.cuda-version }}-libraries
path: llm/llama.cpp/build/**/lib/*
generate-rocm:
needs: [changes]
if: ${{ needs.changes.outputs.GENERATE_ROCM == 'True' }}
strategy:
matrix:
rocm-version:
Expand Down

0 comments on commit c363282

Please sign in to comment.