Skip to content

Commit 0fbe25b

Browse files
committed
Test skiplist approach
1 parent c67ca36 commit 0fbe25b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/triton-benchmarks.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ jobs:
168168
python ../../scripts/build_report.py $REPORTS/matmul-performance-adv-path.csv $REPORTS/gemm-triton-advanced-report.csv --benchmark gemm --compiler triton --param_cols "B,M,K,N" --tflops_col Triton-TFlops --hbm_col "Triton-GB/s" --tag $TAG
169169
170170
- name: Run Triton GEMM (A@B^t) kernel benchmark
171-
if: ${{ steps.install.outcome == 'success' && !cancelled() && !(inputs.only_subset || false) }}
171+
if: ${{ steps.install.outcome == 'success' && !cancelled() && contains(inputs.skip_benchmarks, 'gemm_benchmark.py_abt') }}
172172
run: |
173173
cd benchmarks/triton_kernels_benchmark
174174
TRANSPOSE_B=1 python gemm_benchmark.py --reports $REPORTS
@@ -179,7 +179,7 @@ jobs:
179179
python ../../scripts/build_report.py $REPORTS/matmul-performance-bt.csv $REPORTS/gemm-bt-onednn-report.csv --benchmark gemm-bt --compiler onednn --param_cols "B,M,K,N" --tflops_col onednn-TFlops --hbm_col "onednn-GB/s" --tag $TAG
180180
181181
- name: Run Triton GEMM (A^t@B) kernel benchmark
182-
if: ${{ steps.install.outcome == 'success' && !cancelled() && !(inputs.only_subset || false) }}
182+
if: ${{ steps.install.outcome == 'success' && !cancelled() && contains(inputs.skip_benchmarks, 'gemm_benchmark.py_atb') }}
183183
run: |
184184
cd benchmarks/triton_kernels_benchmark
185185
TRANSPOSE_A=1 python gemm_benchmark.py --reports $REPORTS

0 commit comments

Comments
 (0)