Fixing tuner.json generation - tested partially with binary tuner (runs)

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)
     
 
 

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\""