re-enable m4 cluster run

8cb7327da2cc76e2a0eb775bbe9adc0c1f3f0196 · 2024-12-12 00:01:14 +0000 · Alex Cheema

Files touched

M .github/workflows/benchmarks.yml

Diff

commit 8cb7327da2cc76e2a0eb775bbe9adc0c1f3f0196
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Thu Dec 12 00:01:14 2024 +0000

    re-enable m4 cluster run
---
 .github/workflows/benchmarks.yml | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

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

← bba0aa08 single node test 20 · back to Exo · m3 1194db6e →