56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
from pathlib import Path
|
|
from subprocess import run
|
|
|
|
import click
|
|
|
|
from speccpu import create_spec_env, get_benchspec_dir, x264
|
|
from speccpu.flaglib.perf import record as recordflags
|
|
|
|
|
|
@click.command()
|
|
@click.option(
|
|
"--disable-get-ref-vectorize",
|
|
type=bool,
|
|
default=False,
|
|
help="Enable or disable MC vectorization",
|
|
)
|
|
@click.option(
|
|
"--spec",
|
|
type=Path,
|
|
help="Path to SPEC CPU directory",
|
|
)
|
|
def main(disable_get_ref_vectorize: bool, spec: Path):
|
|
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"
|
|
|
|
x264_run = x264_dir / "run" / "run_base_refspeed_mytest-m64.0000"
|
|
x264_exe = x264_run / "x264_s_base.mytest-m64"
|
|
|
|
if disable_get_ref_vectorize:
|
|
mc_src = x264.MCBuilder().with_disabled_vectorize_get_ref().build()
|
|
x264_exe = x264.recompile_mc_src(
|
|
mc_src, 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()
|