diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml index 7f0eba04..ebf7ec7f 100644 --- a/.github/workflows/benchmarks.yml +++ b/.github/workflows/benchmarks.yml @@ -8,28 +8,28 @@ on: branches: [ '*' ] jobs: - # 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 - - test-m3-single-node: + test-m4-cluster: strategy: matrix: - model: ['llama-3.2-1b'] + 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: '{"M3MAX_GPU40_128GB": 1}' + config: '{"M4PRO_GPU16_24GB": 2}' model: ${{ matrix.model }} - calling_job_name: 'test-m3-cluster' - secrets: inherit \ No newline at end of file + calling_job_name: 'test-m4-cluster' + secrets: inherit + + # test-m3-single-node: + # strategy: + # matrix: + # model: ['llama-3.2-1b'] + # # 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: '{"M3MAX_GPU40_128GB": 1}' + # model: ${{ matrix.model }} + # calling_job_name: 'test-m3-cluster' + # secrets: inherit \ No newline at end of file