From 3c1f9f7ae3c89ba93f1465230ef7ab83147657db Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 18 Mar 2021 02:04:59 -0500
Subject: [PATCH] Fixing tuner.json generation - tested partially with binary
tuner (runs)
---
.../keras/frontend/approxhpvm_translator.py | 14 +++++++++++---
hpvm/projects/keras/frontend/knobs.py | 4 ++--
2 files changed, 13 insertions(+), 5 deletions(-)
diff --git a/hpvm/projects/keras/frontend/approxhpvm_translator.py b/hpvm/projects/keras/frontend/approxhpvm_translator.py
index 32a4839ee6..f3bc0076c6 100644
--- a/hpvm/projects/keras/frontend/approxhpvm_translator.py
+++ b/hpvm/projects/keras/frontend/approxhpvm_translator.py
@@ -945,7 +945,7 @@ class TensorRtTranslator:
f.close()
- def dumpJsonFile(self, dir_prefix):
+ def dumpJsonFile(self, dir_prefix, weights_dir):
f = open(dir_prefix + "/tuner.json", "w+")
f.write("{ \n\n")
@@ -965,7 +965,15 @@ class TensorRtTranslator:
layer_knobs_str = " \"op_knobs\" : { \n"
layer_knobs_str += self.knobs_str[:-3]
- layer_knobs_str += " \n\n } \n\n"
+ layer_knobs_str += " \n\n }, \n\n"
+
+ layer_knobs_str += "\"baseline_knob\" : " + baseline_knobs + ", \n"
+ labels_path = weights_dir + "/tune_labels.bin"
+ layer_knobs_str += "\"tune_labels_path\" : \"" + labels_path + "\", \n"
+ layer_knobs_str += "\"conf_path\" : \"tuner_confs.txt\", \n"
+ layer_knobs_str += "\"fifo_path_r\": \"hpvm_fifo_r\", \n"
+ layer_knobs_str += "\"fifo_path_w\": \"hpvm_fifo_w\" \n"
+
f.write(layer_knobs_str)
f.write("\n\n}")
@@ -996,7 +1004,7 @@ class TensorRtTranslator:
self.generateSourceProgram(src_dir)
- self.dumpJsonFile(src_dir)
+ self.dumpJsonFile(src_dir, weights_dir)
diff --git a/hpvm/projects/keras/frontend/knobs.py b/hpvm/projects/keras/frontend/knobs.py
index 291221acb5..942df9021c 100644
--- a/hpvm/projects/keras/frontend/knobs.py
+++ b/hpvm/projects/keras/frontend/knobs.py
@@ -31,8 +31,8 @@ knobs_speedups[268] = 2
knobs_speedups[269] = 2
-conv_knobs = "12, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 261, 262, 263, 264, 265, 266, 267, 268, 269"
+conv_knobs = "\"12\", \"151\", \"152\", \"153\", \"154\", \"155\", \"156\", \"157\", \"158\", \"159\", \"160\", \"161\", \"162\", \"163\", \"164\", \"165\", \"166\", \"167\", \"168\", \"261\", \"262\", \"263\", \"264\", \"265\", \"266\", \"267\", \"268\", \"269\""
-baseline_knobs = "12"
+baseline_knobs = "\"12\""
--
GitLab