diff --git a/benchmarks/run.py b/benchmarks/run.py index 5d434f30..3f9a2a7f 100644 --- a/benchmarks/run.py +++ b/benchmarks/run.py @@ -65,6 +65,11 @@ "examples.attention", "attention", ), + "cross_entropy": ( + "tritonbench.operators.cross_entropy.operator", + "examples.cross_entropy", + "cross_entropy", + ), }