diff --git a/hpvm/projects/keras/frontend/approxhpvm_translator.py b/hpvm/projects/keras/frontend/approxhpvm_translator.py index 32a4839ee6625a3d0c1c6e1f8284c8c92f512c09..f3bc0076c6024847620174d0bd311017b8f1c6e5 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 291221acb544dbcdf88c810b9401356d2da91be7..942df9021cacdffd21950db9ce98bce80957571a 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\""