Skip to content

Commit 26db8ad

Browse files
authored
Skip several benchmarks on BMG (#4357)
Signed-off-by: Pavel Chekin <[email protected]>
1 parent 4836ca4 commit 26db8ad

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

.github/workflows/triton-benchmarks-bmg.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,4 @@ jobs:
1313
uses: ./.github/workflows/triton-benchmarks.yml
1414
with:
1515
runner_label: b580
16+
skip_benchmarks: "['flash_attention_bwd_benchmark.py','flex_attention_benchmark_custom_masks.py']"

.github/workflows/triton-benchmarks-pvc.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,4 @@ jobs:
1313
uses: ./.github/workflows/triton-benchmarks.yml
1414
with:
1515
runner_label: max1550
16+
skip_benchmarks: "[]"

.github/workflows/triton-benchmarks.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ on:
5050
runner_label:
5151
description: Runner label
5252
type: string
53+
skip_benchmarks:
54+
description: JSON list of benchmarks to skip
55+
type: string
56+
default: "[]"
5357

5458
# Cancels in-progress PR runs when the PR is updated. Manual runs are never cancelled.
5559
concurrency:

0 commit comments

Comments
 (0)