tools/x264_perf_O3: support to disable loop vectorize optimization for get_ref (#1)

Reviewed-on: #1

tools/x264_perf_O3.py

@@ -1,27 +1,64 @@
 from pathlib import Path
 from subprocess import run
 
-from speccpu import get_benchspec_dir, x264
+import click
+
+from speccpu import create_spec_env, find_build, get_benchspec_dir, x264
 from speccpu.flaglib.perf import record as recordflags
 
 
-def main():
-    x264_run_id = "c56635a4-9f49-4d2b-be79-38d9e08097c5"
-    x264_outputroot = Path("/home/lyc/work-ts/local/specOutput") / x264_run_id
-
-
-    x264_dir = get_benchspec_dir(x264_outputroot) / "625.x264_s"
+@click.command()
+@click.option(
+    "--get-ref-vectorize/--no-get-ref-vectorize",
+    default=True,
+    help="Enable or disable mc.c `get_ref` vectorization",
+)
+@click.option(
+    "--spec",
+    type=click.Path(),
+    help="Path to SPEC CPU directory",
+    required=True,
+)
+@click.option(
+    "--x264-run-id",
+    type=str,
+    required=True,
+)
+@click.option(
+    "--outputroot",
+    type=click.Path(),
+    required=True,
+)
+def main(
+    get_ref_vectorize: bool,
+    spec,
+    x264_run_id: str,
+    outputroot,
+):
+    outputroot = Path(outputroot)
+    spec = Path(spec)
+    x264_dir = get_benchspec_dir(outputroot / x264_run_id) / "625.x264_s"
 
     x264_run = x264_dir / "run" / "run_base_refspeed_mytest-m64.0000"
+    x264_exe = x264_run / "x264_s_base.mytest-m64"
 
+    mc_builder = x264.MCBuilder()
 
+    if not get_ref_vectorize:
+        mc_builder.with_disabled_vectorize_get_ref()
+
+    x264_exe = x264.recompile_mc_src(
+        mc_builder.build(),
+        find_build(x264_dir / "build"),
+        create_spec_env(spec),
+    )
     run(
         [
             "perf",
             "record",
             *recordflags(
                 [
-                    str(x264_run / "x264_s_base.mytest-m64"),
+                    str(x264_exe),
                     *x264.specinvoke_args["ref"],
                 ],
                 output=str(Path.cwd() / f"x264-{x264_run_id}perf"),
@@ -31,5 +68,6 @@ def main():
         cwd=x264_run,
     )
 
+
 if __name__ == "__main__":
     main()