Skip to content

Commit

Permalink
[CI][Microbench] Renamed FA benchmark for consistency (#1994)
Browse files Browse the repository at this point in the history
Right now we have `attn` benchmark that is being run on `llvm-target`
branch and old `flash_attention` benchmark from POC branch. Looks like
we want to compare these 2 runs, so we need to rename one of them for
consistency.

Closes #1996
  • Loading branch information
Egor-Krivov authored Aug 23, 2024
1 parent 0455e4a commit 6fdd0ed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/triton-benchmarks.yml
Original file line number Diff line number Diff line change
Expand Up @@ -253,7 +253,7 @@ jobs:
sed -E 's/, /,/g;s/ /,/g' summary.csv > attention-results.csv
source ../../scripts/capture-hw-details.sh
TAG=${{ inputs.tag || 'ci' }}-adv
python ./build_report.py attention-results.csv $REPORTS/attention-triton-advanced-report.csv --benchmark flash_attention --compiler triton --tflops_col max_tflops --param_cols "Z,H,N_CTX,D_HEAD" --tag $TAG
python ./build_report.py attention-results.csv $REPORTS/attention-triton-advanced-report.csv --benchmark attn --compiler triton --tflops_col max_tflops --param_cols "Z,H,N_CTX,D_HEAD" --tag $TAG
- name: Save pip cache
if: ${{ steps.pip-cache.outputs.status == 'miss' }}
Expand Down

0 comments on commit 6fdd0ed

Please sign in to comment.