Skip to content

Commit 164fd00

Browse files
committed
Add dtype to the param cols
1 parent 7a66ce7 commit 164fd00

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
@@ -235,7 +235,7 @@ jobs:
235235
cd benchmarks/triton_kernels_benchmark
236236
python gemm_postop_addmatrix_benchmark.py --reports $REPORTS
237237
source ../../scripts/capture-hw-details.sh
238-
python ../../scripts/build_report.py $REPORTS/matmul-performance-postop-addmatrix.csv $REPORTS/gemm-postop-addmatrix-triton-report.csv --benchmark gemm-postop-addmatrix --compiler triton --param_cols "B,M,K,N" --tflops_col Triton-TFlops --hbm_col "Triton-GB/s" --tag $TAG
238+
python ../../scripts/build_report.py $REPORTS/matmul-performance-postop-addmatrix.csv $REPORTS/gemm-postop-addmatrix-triton-report.csv --benchmark gemm-postop-addmatrix --compiler triton --param_cols "B,M,K,N,dtype" --tflops_col Triton-TFlops --hbm_col "Triton-GB/s" --tag $TAG
239239
240240
- name: Run Triton FA kernel benchmark
241241
if: ${{ steps.install.outcome == 'success' && !cancelled() && !contains(fromJson(inputs.skip_benchmarks || '[]'), 'flash_attention_fwd_benchmark.py') }}

0 commit comments

Comments
 (0)