From dd1bff2ec06d42b619f7dda72de2b628ab566f3c Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Fri, 2 Apr 2021 06:19:15 -0500
Subject: [PATCH] Only test first 5 configs in check-hpvm-profiler (should take
3 mins overall)
---
hpvm/test/dnn_benchmarks/profiling/test_hpvm_c_profiling.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
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 230fdf8b73..853b0dc3e2 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)
--
GitLab