Alex Cheema 7 months ago
parent
commit
1194db6e65
1 changed files with 18 additions and 18 deletions
  1. 18 18
      .github/workflows/benchmarks.yml

+ 18 - 18
.github/workflows/benchmarks.yml

@@ -8,28 +8,28 @@ on:
     branches: [ '*' ]
     branches: [ '*' ]
 
 
 jobs:
 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:
   #   strategy:
   #     matrix:
   #     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
   #     # Optional: add fail-fast: false if you want all matrix jobs to continue even if one fails
   #     fail-fast: false
   #     fail-fast: false
   #   uses: ./.github/workflows/bench_job.yml
   #   uses: ./.github/workflows/bench_job.yml
   #   with:
   #   with:
-  #     config: '{"M3MAX_GPU40_128GB": 1}'
+  #     config: '{"M4PRO_GPU16_24GB": 2}'
   #     model: ${{ matrix.model }}
   #     model: ${{ matrix.model }}
-  #     calling_job_name: 'test-m3-cluster'
-  #   secrets: inherit
+  #     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