74 lines
1.6 KiB
Python
74 lines
1.6 KiB
Python
from pathlib import Path
|
|
from subprocess import run
|
|
|
|
import click
|
|
|
|
from speccpu import create_spec_env, find_build, get_benchspec_dir, x264
|
|
from speccpu.flaglib.perf import record as recordflags
|
|
|
|
|
|
@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_exe),
|
|
*x264.specinvoke_args["ref"],
|
|
],
|
|
output=str(Path.cwd() / f"x264-{x264_run_id}.perf"),
|
|
),
|
|
],
|
|
check=True,
|
|
cwd=x264_run,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|