Browse Source

run with three m4 pro

Alex Cheema 7 months ago
parent
commit
dbb7ad3c08
1 changed files with 15 additions and 2 deletions
  1. 15 2
      .github/workflows/benchmarks.yml

+ 15 - 2
.github/workflows/benchmarks.yml

@@ -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