From 755dd477ddb0cabc4eab4e134dfd18095c1af364 Mon Sep 17 00:00:00 2001 From: Glen Date: Sun, 8 Dec 2024 20:37:33 +0000 Subject: [PATCH] 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 86f3e58dd..be918ecf4 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 89960472f..204489379 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