From e162800992afa1c36950b293a43e953969953f33 Mon Sep 17 00:00:00 2001
From: Will Feng <yfeng.us@gmail.com>
Date: Sun, 20 Jul 2025 23:15:31 -0700
Subject: [PATCH] [Benchmark] Add fused_linear_cross_entropy to tritonbench
 integration

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

diff --git a/benchmarks/run.py b/benchmarks/run.py
index 3a87130d..c45aa800 100644
--- a/benchmarks/run.py
+++ b/benchmarks/run.py
@@ -75,6 +75,11 @@
         "examples.fp8_attention",
         "fp8_attention_tritonbench",
     ),
+    "fused_linear_cross_entropy": (
+        "tritonbench.operators.fused_linear_cross_entropy.operator",
+        "examples.fused_linear_cross_entropy",
+        "fused_linear_cross_entropy",
+    ),
 }