Skip to content

Commit 47878bf

Browse files
authored
[Benchmark] Add attention tritonbench integration (#284)
1 parent e8cf178 commit 47878bf

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

benchmarks/run.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,11 @@
5555
"examples.fp8_gemm",
5656
"fp8_gemm_tritonbench",
5757
),
58+
"flash_attention": (
59+
"tritonbench.operators.flash_attention.operator",
60+
"examples.attention",
61+
"attention",
62+
),
5863
}
5964

6065

0 commit comments

Comments
 (0)