69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
import os
|
|
from pathlib import Path
|
|
from subprocess import Popen
|
|
|
|
from speccpu import SPEC, check_bench_dir, find_build, get_benchspec_dir, x264
|
|
|
|
|
|
def x264_disable_mc_mul(
|
|
spec_dir: Path,
|
|
spec_config: str,
|
|
spec_outputroot: Path,
|
|
):
|
|
spec = SPEC(spec_dir)
|
|
spec_env = spec.env()
|
|
|
|
x264_dir = get_benchspec_dir(spec_outputroot) / "625.x264_s"
|
|
|
|
assert check_bench_dir(x264_dir)
|
|
x264_build = find_build(x264_dir / "build")
|
|
|
|
assert x264_build.exists()
|
|
assert "x264_src" in os.listdir(x264_build)
|
|
|
|
mc_path = x264.get_mc_path(x264_build)
|
|
assert mc_path.exists()
|
|
|
|
mc_src = x264.MCBuilder().with_disabled_iscale_mul().build()
|
|
|
|
# Write mc_src to mc_path, recompile it, and replace resulting exe
|
|
with open(mc_path, "w") as f:
|
|
f.write(mc_src)
|
|
|
|
# Recompile
|
|
recompiled_x264_path = x264.recompile_mc(mc_path, x264_build, spec_env)
|
|
assert recompiled_x264_path.exists()
|
|
assert recompiled_x264_path.is_file()
|
|
|
|
# Execute that new binary
|
|
x264.update_exe(recompiled_x264_path, x264_dir / "exe")
|
|
|
|
with Popen(
|
|
spec.mkcmd_runcpu(
|
|
spec_config,
|
|
["x264_s"],
|
|
setprocgroup=True,
|
|
workload="ref",
|
|
output_root=spec_outputroot,
|
|
),
|
|
env=spec_env,
|
|
) as process:
|
|
process.wait()
|
|
|
|
|
|
def main():
|
|
spec_path = Path("/home/lyc/spec2017-1.1.8-new")
|
|
x264_run_id = "f7e8503c-b8ca-47c9-879f-d1e73bbedc99"
|
|
x264_config = spec_path / "config" / f"{x264_run_id}.cfg"
|
|
x264_outputroot = Path("/home/lyc/work-ts/local/specOutput") / x264_run_id
|
|
|
|
x264_disable_mc_mul(
|
|
spec_path,
|
|
str(x264_config),
|
|
x264_outputroot,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|