fixup options

This commit is contained in:
2025-04-27 19:14:41 +08:00
parent d91125723e
commit db474163ea

View File

@@ -3,37 +3,55 @@ from subprocess import run
import click
from speccpu import create_spec_env, get_benchspec_dir, x264
from speccpu import create_spec_env, find_build, get_benchspec_dir, x264
from speccpu.flaglib.perf import record as recordflags
@click.command()
@click.option(
"--disable-get-ref-vectorize",
type=bool,
default=False,
"--get-ref-vectorize/--no-get-ref-vectorize",
default=True,
help="Enable or disable MC vectorization",
)
@click.option(
"--spec",
type=Path,
type=click.Path(),
help="Path to SPEC CPU directory",
required=True,
)
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"
@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"
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)
)
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",