diff --git a/benchmarks/run.py b/benchmarks/run.py index 3a87130d..c45aa800 100644 --- a/benchmarks/run.py +++ b/benchmarks/run.py @@ -75,6 +75,11 @@ "examples.fp8_attention", "fp8_attention_tritonbench", ), + "fused_linear_cross_entropy": ( + "tritonbench.operators.fused_linear_cross_entropy.operator", + "examples.fused_linear_cross_entropy", + "fused_linear_cross_entropy", + ), }