Skip to content

Commit 8c2b9c9

Browse files
committed
[Benchmark] Add fused_linear_cross_entropy to tritonbench integration
1 parent 452205e commit 8c2b9c9

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
@@ -75,6 +75,11 @@
7575
"examples.fp8_attention",
7676
"fp8_attention_tritonbench",
7777
),
78+
"fused_linear_cross_entropy": (
79+
"tritonbench.operators.fused_linear_cross_entropy.operator",
80+
"examples.fused_linear_cross_entropy",
81+
"fused_linear_cross_entropy",
82+
),
7883
}
7984

8085

0 commit comments

Comments
 (0)