Skip to content
Snippets Groups Projects
Commit dd1bff2e authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Only test first 5 configs in check-hpvm-profiler (should take 3 mins overall)

parent 04d867ce
No related branches found
No related tags found
No related merge requests found
......@@ -2,7 +2,7 @@
from pathlib import Path
from sys import argv
from hpvm_profiler import profile_configs, read_hpvm_configs
from hpvm_profiler import profile_configs, read_hpvm_configs, write_hpvm_configs
# relative to cwd()
benchmarks_bindir = Path("../hpvm-c")
......@@ -17,4 +17,6 @@ dnn = argv[1]
bench_bin_file = benchmarks_bindir / f"hpvm_{dnn}"
config_file = benchmarks_srcdir / dnn / "data/tuner_confs.txt"
out_config_file = f"./{dnn}.txt"
profile_configs(bench_bin_file, config_file, out_config_file)
header, configs = read_hpvm_configs(config_file)
profile_configs(bench_bin_file, configs[1:6], configs[0], progress_bar=False)
write_hpvm_configs(header, configs[:6], out_config_file)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment