Skip to content

Commit b46503c

Browse files
committed
[Benchmark] Add fp8_attention to tritonbench integration
stack-info: PR: #319, branch: yf225/stack/27
1 parent 2a8e58b commit b46503c

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
@@ -70,6 +70,11 @@
7070
"examples.cross_entropy",
7171
"cross_entropy",
7272
),
73+
"fp8_attention": (
74+
"tritonbench.operators.fp8_attention.operator",
75+
"examples.fp8_attention",
76+
"fp8_attention_tritonbench",
77+
),
7378
}
7479

7580

0 commit comments

Comments
 (0)