Add a feature for using the same number of loops as a previous run

doc/changelog.rst

@@ -1,6 +1,9 @@
 Changelog
 =========
 
+* Add a --same-loops option to the run command to use the exact same number of
+  loops as a previous run (without recalibrating).
+
 Version 1.10.0 (2023-10-22)
 --------------
 * Add benchmark for asyncio_webockets

doc/usage.rst

@@ -140,6 +140,10 @@ options::
   -p PYTHON, --python PYTHON
                         Python executable (default: use running
                         Python)
+  --same-loops SAME_LOOPS
+                        Use the same number of loops as a previous run
+                        (i.e., don't recalibrate). Should be a path to a
+                        .json file from a previous run.
 
 show
 ----

pyperformance/cli.py

@@ -75,6 +75,10 @@ def parse_args():
     cmd.add_argument("--min-time", metavar="MIN_TIME",
                      help="Minimum duration in seconds of a single "
                      "value, used to calibrate the number of loops")
+    cmd.add_argument("--same-loops",
+                     help="Use the same number of loops as a previous run "
+                     "(i.e., don't recalibrate). Should be a path to a "
+                     ".json file from a previous run.")
     filter_opts(cmd)
 
     # show

pyperformance/data-files/benchmarks/MANIFEST

@@ -77,6 +77,9 @@ spectral_norm	<local>
 sqlalchemy_declarative	<local>
 sqlalchemy_imperative	<local>
 sqlglot	<local>
+sqlglot_parse	<local:sqlglot>
+sqlglot_transpile	<local:sqlglot>
+sqlglot_optimize	<local:sqlglot>
 sqlite_synth	<local>
 sympy	<local>
 telco	<local>

pyperformance/data-files/benchmarks/bm_sqlglot/bm_sqlglot_optimize.toml

@@ -0,0 +1,3 @@
+[tool.pyperformance]
+name = "sqlglot_optimize"
+extra_opts = ["optimize"]

pyperformance/data-files/benchmarks/bm_sqlglot/bm_sqlglot_parse.toml

@@ -0,0 +1,3 @@
+[tool.pyperformance]
+name = "sqlglot_parse"
+extra_opts = ["parse"]

pyperformance/data-files/benchmarks/bm_sqlglot/bm_sqlglot_transpile.toml

@@ -0,0 +1,3 @@
+[tool.pyperformance]
+name = "sqlglot_transpile"
+extra_opts = ["transpile"]

pyperformance/data-files/benchmarks/bm_sqlglot/pyproject.toml

@@ -10,3 +10,4 @@ dynamic = ["version"]
 
 [tool.pyperformance]
 name = "sqlglot"
+extra_opts = ["normalize"]

pyperformance/data-files/benchmarks/bm_sqlglot/run_benchmark.py

@@ -164,10 +164,31 @@ def bench_normalize(loops):
     return elapsed
 
 
+BENCHMARKS = {
+    "parse": bench_parse,
+    "transpile": bench_transpile,
+    "optimize": bench_optimize,
+    "normalize": bench_normalize
+}
+
+
+def add_cmdline_args(cmd, args):
+    cmd.append(args.benchmark)
+
+
+def add_parser_args(parser):
+    parser.add_argument(
+        "benchmark",
+        choices=BENCHMARKS,
+        help="Which benchmark to run."
+    )
+
+
 if __name__ == "__main__":
-    runner = pyperf.Runner()
+    runner = pyperf.Runner(add_cmdline_args=add_cmdline_args)
     runner.metadata['description'] = "SQLGlot benchmark"
-    runner.bench_time_func("sqlglot_parse", bench_parse)
-    runner.bench_time_func("sqlglot_transpile", bench_transpile)
-    runner.bench_time_func("sqlglot_optimize", bench_optimize)
-    runner.bench_time_func("sqlglot_normalize", bench_normalize)
+    add_parser_args(runner.argparser)
+    args = runner.parse_args()
+    benchmark = args.benchmark
+
+    runner.bench_time_func(f"sqlglot_{benchmark}", BENCHMARKS[benchmark])

pyperformance/run.py

@@ -1,5 +1,6 @@
 from collections import namedtuple
 import hashlib
+import json
 import sys
 import time
 import traceback
@@ -50,7 +51,28 @@ def get_run_id(python, bench=None):
     return RunID(py_id, compat_id, bench, ts)
 
 
+def get_loops_from_file(filename):
+    with open(filename) as fd:
+        data = json.load(fd)
+
+    loops = {}
+    for benchmark in data["benchmarks"]:
+        metadata = benchmark.get("metadata", data["metadata"])
+        name = metadata["name"]
+        if name.endswith("_none"):
+            name = name[:-len("_none")]
+        if "loops" in metadata:
+            loops[name] = metadata["loops"]
+
+    return loops
+
+
 def run_benchmarks(should_run, python, options):
+    if options.same_loops is not None:
+        loops = get_loops_from_file(options.same_loops)
+    else:
+        loops = {}
+
     to_run = sorted(should_run)
 
     info = _pythoninfo.get_info(python)
@@ -136,6 +158,10 @@ def add_bench(dest_suite, obj):
 
             return dest_suite
 
+        if name in loops:
+            pyperf_opts.append(f"--loops={loops[name]}")
+            pyperf_opts.append("--warmups=1")
+
         bench_venv, bench_runid = benchmarks.get(bench)
         if bench_venv is None:
             print("ERROR: Benchmark %s failed: could not install requirements" % name)