Always checkout branch

pytorch-labs · Jan 23, 2025 · 205e694 · 205e694
1 parent 0802a4b
commit 205e694
Showing 1 changed file with 4 additions and 15 deletions.
diff --git a/.github/workflows/_linux-benchmark-h100.yml b/.github/workflows/_linux-benchmark-h100.yml
@@ -30,8 +30,7 @@ jobs:
       JOB_NAME: tritonbench-h100-${{ inputs.conda_env }}-${{ inputs.benchmark_name }}
       TRITONBENCH_SCRIBE_GRAPHQL_ACCESS_TOKEN: ${{ secrets.TRITONBENCH_SCRIBE_GRAPHQL_ACCESS_TOKEN }}
     steps:
-      - name: Checkout Tritonbench (if on pull_request)
-        if: github.event_name == 'pull_request'
+      - name: Checkout Tritonbench
         uses: actions/checkout@v3
         with:
           submodules: recursive
@@ -42,13 +41,8 @@ jobs:
           nvidia-smi
       - name: Benchmarking
         run: |
-          if [ "${GITHUB_EVENT_NAME}" != "pull_request" ]; then
-            export repo_prefix=/workspace/tritonbench
-          else
-            export repo_prefix=$PWD
-          fi
-          bash "${repo_prefix}"/.ci/tritonbench/run-benchmark.sh ${{ inputs.benchmark_name }}
-          cp -r "${repo_prefix}"/.benchmarks/${{ inputs.benchmark_name }} benchmark-output
+          bash .ci/tritonbench/run-benchmark.sh ${{ inputs.benchmark_name }}
+          cp -r .benchmarks/${{ inputs.benchmark_name }} benchmark-output
       - name: Upload result to GH Actions Artifact
         uses: actions/upload-artifact@v4
         with:
@@ -57,10 +51,5 @@ jobs:
       - name: Upload result to Scribe
         run: |
           . "${SETUP_SCRIPT}"
-          if [ "${GITHUB_EVENT_NAME}" != "pull_request" ]; then
-            export repo_prefix=/workspace/tritonbench
-          else
-            export repo_prefix=$PWD
-          fi
           latest_result_json=$(find ./benchmark-output/ -name "result.json"  | sort -r | head -n 1)
-          python "${repo_prefix}"/.ci/upload/scribe.py --json ${latest_result_json}
+          python .ci/upload/scribe.py --json ${latest_result_json}