From e162800992afa1c36950b293a43e953969953f33 Mon Sep 17 00:00:00 2001 From: Will Feng 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", + ), }