From 0b91016b4f874e2b64b145179b5988e33683762f Mon Sep 17 00:00:00 2001
From: Will Feng <yfeng.us@gmail.com>
Date: Mon, 14 Jul 2025 13:06:01 -0700
Subject: [PATCH] [Benchmark] Add cross_entropy to tritonbench integration

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

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",
+    ),
 }