Add missing parts.

Author: lialan

iree/turbine/kernel/boo/driver/utils.py

@@ -19,7 +19,14 @@ def get_timing_parser() -> argparse.ArgumentParser:
     timing_parser.add_argument(
         "--iter",
         type=int,
-        help="Number of iterations to run (floor when auto-adjusting)",
+        help="Exact number of iterations (disables auto-adjustment; "
+        "shorthand for --min-iter X --min-time 0)",
+        default=None,
+    )
+    timing_parser.add_argument(
+        "--min-iter",
+        type=int,
+        help="Minimum number of iterations when auto-adjusting (default: 100)",
         default=100,
     )
     timing_parser.add_argument(
@@ -31,6 +38,13 @@ def get_timing_parser() -> argparse.ArgumentParser:
     return timing_parser
 
 
+def resolve_timing_args(timing_args: argparse.Namespace) -> None:
+    """Resolve --iter shorthand into --min-iter and --min-time 0."""
+    if timing_args.iter is not None:
+        timing_args.min_iter = timing_args.iter
+        timing_args.min_time = 0.0
+
+
 def load_commands(commands_file: str) -> list[str]:
     """Loads commands of a given kind from a text file.