full all shapees for rms_norm

yf225 · yf225 · commit 9fa5ecc96c94 · 2025-08-18T10:50:40.000-07:00
diff --git a/benchmarks/run.py b/benchmarks/run.py
@@ -42,9 +42,6 @@
         "tritonbench.operators.rms_norm.operator",
         "examples.rms_norm",
         "rms_norm_tritonbench",
-        {
-            "num_inputs": 3
-        },  # TODO(yf225): reduction dim size = 8192 currently throws error
     ),
     "layer_norm": (
         "tritonbench.operators.layer_norm.operator",
