From ddf1bbc7522ec5900579e3e348475366e845515b Mon Sep 17 00:00:00 2001
From: Will Feng <yfeng.us@gmail.com>
Date: Mon, 27 Oct 2025 11:17:06 -0700
Subject: [PATCH] [Benchmark CI] Set welford num_inputs to 6 to avoid timeout

---
 benchmarks/run.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/benchmarks/run.py b/benchmarks/run.py
index c2b103a02..750833422 100644
--- a/benchmarks/run.py
+++ b/benchmarks/run.py
@@ -279,6 +279,9 @@ class RunResult:
         "tritonbench.operators.welford.operator",
         "examples.welford",
         "welford",
+        {
+            "num_inputs": 6,  # welford takes long time on Benchmark CI, so use fewer inputs instead.
+        },
     ),
     "gather_gemv": (
         "tritonbench.operators.gather_gemv.operator",