diff --git a/hpvm/test/dnn_benchmarks/profiling/test_hpvm_c_profiling.py b/hpvm/test/dnn_benchmarks/profiling/test_hpvm_c_profiling.py index 230fdf8b73dfd7959cfaa98fe06eafe6a75087b1..853b0dc3e23a3ea847748ecaeda62650e99ee430 100755 --- a/hpvm/test/dnn_benchmarks/profiling/test_hpvm_c_profiling.py +++ b/hpvm/test/dnn_benchmarks/profiling/test_hpvm_c_profiling.py @@ -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)