[parser] Rename --iters to --rep (#359)

Author: xuzhao9

tritonbench/utils/parser.py

@@ -1,7 +1,7 @@
 import argparse
 
 from tritonbench.utils.env_utils import AVAILABLE_PRECISIONS, is_fbcode
-from tritonbench.utils.triton_op import DEFAULT_RUN_ITERS, DEFAULT_WARMUP
+from tritonbench.utils.triton_op import DEFAULT_REP, DEFAULT_WARMUP
 
 
 def get_parser(args=None):
@@ -52,13 +52,13 @@ def get_parser(args=None):
         "--warmup",
         type=int,
         default=DEFAULT_WARMUP,
-        help="Num of warmup runs for reach benchmark run.",
+        help="Num of warmup runs for each benchmark run.",
     )
     parser.add_argument(
-        "--iter",
+        "--rep",
         type=int,
-        default=DEFAULT_RUN_ITERS,
-        help="Num of reps for each benchmark run.",
+        default=DEFAULT_REP,
+        help="The rep time for each benchmark run.",
     )
     parser.add_argument(
         "--csv",

tritonbench/utils/triton_op.py

@@ -79,7 +79,7 @@ class BenchmarkOperatorBackend:
 
 
 DEFAULT_WARMUP = 25
-DEFAULT_RUN_ITERS = 100
+DEFAULT_REP = 100
 DEFAULT_QUANTILES = [0.5, 0.1, 0.9]
 REGISTERED_BENCHMARKS: Dict[str, OrderedDict[str, BenchmarkOperatorBackend]] = {}
 REGISTERED_METRICS: defaultdict[str, List[str]] = defaultdict(list)
@@ -127,7 +127,7 @@ def __exit__(self, *args, **kwargs):
             self.elapsed_ms = (end_time - self._start_time) * 1e3
 
 
-def do_bench_walltime(fn, warmup=25, rep=100):
+def do_bench_walltime(fn, warmup=25, rep=DEFAULT_REP):
     fn()
     torch.cuda.synchronize()
 
@@ -808,7 +808,7 @@ def fwd_no_grad_fn():
             return fwd_no_grad_fn
 
     def run(
-        self, warmup=DEFAULT_WARMUP, rep=DEFAULT_RUN_ITERS, quantiles=DEFAULT_QUANTILES
+        self, warmup=DEFAULT_WARMUP, rep=DEFAULT_REP, quantiles=DEFAULT_QUANTILES
     ) -> None:
         """Benchmarking the operator and returning its metrics."""
         metrics = []
@@ -1183,7 +1183,7 @@ def _do_bench(
         input_id: int,
         fn_name: str,
         warmup=DEFAULT_WARMUP,
-        rep=DEFAULT_RUN_ITERS,
+        rep=DEFAULT_REP,
         quantiles=DEFAULT_QUANTILES,
         baseline: bool = False,
     ) -> BenchmarkOperatorMetrics: