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