From cc95ff1f89a783d9fed2489bb76db61a1830d0a5 Mon Sep 17 00:00:00 2001
From: Will Feng <yfeng.us@gmail.com>
Date: Sat, 12 Jul 2025 01:51:49 -0700
Subject: [PATCH] [Benchmark] Add softmax tritonbench integration

stack-info: PR: https://github.com/pytorch-labs/helion/pull/286, branch: yf225/stack/17
---
 benchmarks/run.py | 5 +++++
 1 file changed, 5 insertions(+)

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",