From f32baaa4a909e360df7961e6edcc25a22bfa5322 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Tue, 6 Apr 2021 23:03:56 -0500
Subject: [PATCH] Fixing tuner.json generation bugs in Keras frontend
---
hpvm/projects/keras/keras_frontend/approxhpvm_translator.py | 2 +-
hpvm/projects/keras/keras_frontend/knobs.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
index 812c276798..edb8d4af29 100644
--- a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
+++ b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
@@ -960,7 +960,7 @@ class TensorRtTranslator:
layer_knobs_str += self.knobs_str[:-3]
layer_knobs_str += " \n\n }, \n\n"
- layer_knobs_str += "\"baseline_knob\" : " + baseline_knobs + ", \n"
+ layer_knobs_str += "\"baseline_knob\" : " + fp32_baseline + ", \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"
diff --git a/hpvm/projects/keras/keras_frontend/knobs.py b/hpvm/projects/keras/keras_frontend/knobs.py
index 388cbd28f8..fb0c029478 100644
--- a/hpvm/projects/keras/keras_frontend/knobs.py
+++ b/hpvm/projects/keras/keras_frontend/knobs.py
@@ -317,6 +317,7 @@ for i in range(261, 270):
conv_knobs += ", "
-baseline_knobs = "\"12\""
+fp32_baseline = "\"11\""
+baseline_knobs = "\"11\", \"12\""
--
GitLab