diff --git a/examples/python/CuTeDSL/ampere/sgemm.py b/examples/python/CuTeDSL/ampere/sgemm.py index 8058f24d40..d50f700b9b 100644 --- a/examples/python/CuTeDSL/ampere/sgemm.py +++ b/examples/python/CuTeDSL/ampere/sgemm.py @@ -836,6 +836,7 @@ def parse_comma_separated_ints(s: str) -> Tuple[int, ...]: parser.add_argument("--b_major", choices=["k", "n"], default="k") parser.add_argument("--c_major", choices=["n", "m"], default="n") parser.add_argument("--warmup_iterations", default=2, type=int) + parser.add_argument("--static_shape", action="store_true") parser.add_argument("--iterations", default=100, type=int) parser.add_argument("--skip_ref_check", action="store_true") parser.add_argument(