diff --git a/benchmarks/run.py b/benchmarks/run.py index 430b6b723..f0edf48c9 100644 --- a/benchmarks/run.py +++ b/benchmarks/run.py @@ -45,6 +45,11 @@ "rms_norm_tritonbench", ), "sum": ("tritonbench.operators.sum.operator", "examples.sum", "sum_tritonbench"), + "softmax": ( + "tritonbench.operators.softmax.operator", + "examples.softmax", + "softmax", + ), "jagged_mean": ( "tritonbench.operators.jagged_mean.operator", "examples.jagged_mean",