[Benchmark] Add softmax tritonbench integration (#286)

yf225 · web-flow · commit 8f5068c0317f · 2025-07-14T00:06:34.000-07:00
diff --git 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",
