Skip to content

Commit dec8362

Browse files
committed
[Benchmark] Add fused_linear_cross_entropy to tritonbench integration
stack-info: PR: #343, branch: yf225/stack/38
1 parent 58a9633 commit dec8362

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)