run with three m4 pro

This commit is contained in:
Alex Cheema
2024-12-12 14:36:18 +00:00
parent 2abe57be21
commit dbb7ad3c08

View File

@@ -8,7 +8,20 @@ on:
branches: [ '*' ]
jobs:
test-m4-cluster:
# test-m4-cluster:
# strategy:
# matrix:
# model: ['llama-3.2-1b', 'llama-3.2-3b']
# # Optional: add fail-fast: false if you want all matrix jobs to continue even if one fails
# fail-fast: false
# uses: ./.github/workflows/bench_job.yml
# with:
# config: '{"M4PRO_GPU16_24GB": 2}'
# model: ${{ matrix.model }}
# calling_job_name: 'test-m4-cluster'
# secrets: inherit
three-m4-pro-cluster:
strategy:
matrix:
model: ['llama-3.2-1b', 'llama-3.2-3b']
@@ -16,7 +29,7 @@ jobs:
fail-fast: false
uses: ./.github/workflows/bench_job.yml
with:
config: '{"M4PRO_GPU16_24GB": 2}'
config: '{"M4PRO_GPU16_24GB": 3}'
model: ${{ matrix.model }}
calling_job_name: 'test-m4-cluster'
secrets: inherit