Glen 7 months ago
parent
commit
755dd477dd
2 changed files with 6 additions and 2 deletions
  1. 5 2
      .github/workflows/bench_job.yml
  2. 1 0
      .github/workflows/benchmarks.yml

+ 5 - 2
.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=$!

+ 1 - 0
.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