Skip to content

Commit 288599e

Browse files
authored
[CI][benchmarks] Fixed CI for prefix sum benchmark (#4731)
Closes #4511 Note that merging will break apart geomean charts for `prefix-sum`, but that shouldn't be a big problem, as they already broke apart recently due to agama update
1 parent fa40e4c commit 288599e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/triton-benchmarks.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -337,7 +337,7 @@ jobs:
337337
cd benchmarks/triton_kernels_benchmark
338338
python prefix_sums.py --reports $REPORTS --n_runs $N_RUNS
339339
source ../../scripts/capture-hw-details.sh
340-
python build_report.py $REPORTS/prefix-sums.csv $REPORTS/prefix_sums-triton-report.csv --benchmark prefix_sums --compiler triton --param_cols "N" --tflops_col Triton-TFlops --hbm_col "Triton-GB/s" --tag $TAG
340+
python build_report.py $REPORTS/prefix-sums.csv $REPORTS/prefix_sums-triton-report.csv --benchmark prefix_sums --compiler triton --param_cols "M,N,AXIS" --tflops_col Triton-TFlops --hbm_col "Triton-GB/s" --tag $TAG
341341
342342
- name: Run micro benchmark
343343
if: ${{ steps.install.outcome == 'success' && !cancelled() && (inputs.benchmarks == '' || contains(fromJson(inputs.benchmarks || '[]'), 'micro_benchmarks.py')) && !contains(fromJson(inputs.skip_benchmarks || '[]'), 'micro_benchmarks') }}

0 commit comments

Comments
 (0)