← back to Exo
jobname
755dd477ddb0cabc4eab4e134dfd18095c1af364 · 2024-12-08 20:37:33 +0000 · Glen
Files touched
M .github/workflows/bench_job.ymlM .github/workflows/benchmarks.yml
Diff
commit 755dd477ddb0cabc4eab4e134dfd18095c1af364
Author: Glen <glen@Delias-MacBook-Air-2.local>
Date: Sun Dec 8 20:37:33 2024 +0000
jobname
---
.github/workflows/bench_job.yml | 7 +++++--
.github/workflows/benchmarks.yml | 1 +
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/.github/workflows/bench_job.yml b/.github/workflows/bench_job.yml
index 86f3e58d..be918ecf 100644
--- a/.github/workflows/bench_job.yml
+++ b/.github/workflows/bench_job.yml
@@ -10,6 +10,9 @@ on:
model:
required: true
type: string
+ calling_job_name: # New input parameter
+ required: true
+ type: string
jobs:
generate-matrix:
runs-on: ubuntu-latest
@@ -54,13 +57,13 @@ jobs:
aws_access_key_id: ${{ secrets.S3_EXO_BENCHMARKS_AWS_ACCESS_KEY_ID }}
aws_secret_key: ${{ secrets.S3_EXO_BENCHMARKS_AWS_SECRET_ACCESS_KEY }}
run: |
- CALLING_JOB=$(echo "${{ github.workflow_ref }}" | awk -F'/' '{print $NF}')
+ CALLING_JOB="${{ inputs.calling_job_name }}"
UNIQUE_JOB_ID="${CALLING_JOB}_${GITHUB_RUN_ID}"
ALL_NODE_IDS=$(for i in $(seq ${{ strategy.job-total }} -1 0); do echo -n "${UNIQUE_JOB_ID}_${i},"; done | sed 's/,$//')
MY_NODE_ID="${UNIQUE_JOB_ID}_${{ strategy.job-index }}"
source env/bin/activate
export PATH="/usr/local/bin:/opt/homebrew/bin:$PATH"
- exo --node-id="${MY_NODE_ID}" --node-id-filter="${ALL_NODE_IDS}" --chatgpt-api-port 52415 --disable-tui > output1.log 2>&1 &
+ exo --node-id="${MY_NODE_ID}" --node-id-filter="${ALL_NODE_IDS}" --chatgpt-api-port 52415 > output1.log 2>&1 &
PID1=$!
tail -f output1.log &
TAIL1=$!
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
index 89960472..20448937 100644
--- a/.github/workflows/benchmarks.yml
+++ b/.github/workflows/benchmarks.yml
@@ -13,4 +13,5 @@ jobs:
with:
config: '{"M4PRO_GPU16_24GB": 2}'
model: 'llama-3.2-1b'
+ calling_job_name: 'test-m4-cluster'
secrets: inherit
\ No newline at end of file
← fb44eb08 simplify bench
·
back to Exo
·
list exo processes before test, warmup req in bench 87865f0c →