From 4561aec9421f443f438bc3543086c9402e90123d Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Thu, 3 Dec 2020 15:03:11 -0600
Subject: [PATCH] Commenting out invocation of PROMISE code gen
---
.../keras/frontend/approxhpvm_translator.py | 4 +--
.../keras/frontend/promise_translator.py | 25 +------------------
2 files changed, 3 insertions(+), 26 deletions(-)
diff --git a/llvm/projects/keras/frontend/approxhpvm_translator.py b/llvm/projects/keras/frontend/approxhpvm_translator.py
index bd30ad2a44..2c964db34b 100644
--- a/llvm/projects/keras/frontend/approxhpvm_translator.py
+++ b/llvm/projects/keras/frontend/approxhpvm_translator.py
@@ -959,8 +959,8 @@ def translate_to_approxhpvm(model, weights_dir, test_data=None, test_labels=None
input_str = tensorRtTranslator.getInputStr()
- promiseRtTranslator = PromiseRtTranslator(dfg, weight_str)
- promiseRtTranslator.translate(model, weights_dir, test_data)
+ #promiseRtTranslator = PromiseRtTranslator(dfg, weight_str)
+ #promiseRtTranslator.translate(model, weights_dir, test_data)
filter_names = tensorRtTranslator.getFilterNames()
hpvmTranslator = HPVMTranslator(dfg, weight_str, input_str, filter_names)
diff --git a/llvm/projects/keras/frontend/promise_translator.py b/llvm/projects/keras/frontend/promise_translator.py
index 54982bd7c7..30acf47cd8 100644
--- a/llvm/projects/keras/frontend/promise_translator.py
+++ b/llvm/projects/keras/frontend/promise_translator.py
@@ -386,11 +386,8 @@ class PromiseRtTranslator:
if len(cur_node.inputs) == 0:
return "input"
- #print ("Input_type = ", cur_node.inputs[0].layer_type)
-
pred_layer_type = cur_node.inputs[0].layer_type
# FIXME: Assuming the 'inference' phase - hence skipping Dropout
- #if pred_layer_type == "Flatten" or pred_layer_type == "Dropout":
if self.isSkipLayer(pred_layer_type):
cur_node = self.getPrevActiveLayer(cur_node)
@@ -1056,8 +1053,6 @@ class PromiseRtTranslator:
# Inference over test set
layer_outs = functor([x_test[start:end], 1.])
-
- ###layer_outs = functor([x_test[start:end]])
# NOTE: Saving quant ranges for input
min_val = np.amin(x_test)
@@ -1096,12 +1091,9 @@ class PromiseRtTranslator:
min_val = range[0]
if range[1] > max_val:
max_val = range[1]
-
- # print ("****** min = ", range[0], " max = ", range[1])
self.quant_ranges[layer_name] = (min_val, max_val)
-
print ("---- min = ", min_val, " max = ", max_val, " ----- \n\n")
ind += 1
@@ -1123,20 +1115,6 @@ class PromiseRtTranslator:
print ("layer_shape = ", layer.input.shape)
self.layer_input_sizes[layer_name] = layer.input.shape
- #inp = model.input # input placeholder
- #inputs = []
- # functor = K.function([inp, K.learning_phase()], outputs ) # evaluation function
- # functor2 = K.function([inp, K.learning_phase()], inputs ) # evaluation function
- # Testing
- # layer_outs = functor([x_test, 1.])
- # layer_inputs = functor2([x_test, 1.])
-
- #index = 0
- #for layer_in in layer_inputs:
- # print ("layer_in = ", layer_in.shape)
- # layer_name = model.layers[index].name
- # self.layer_input_sizes[layer_name] = layer_in.shape
- # index += 1
@@ -1323,7 +1301,6 @@ class PromiseRtTranslator:
self.findQuantizeRanges(model, x_test)
- #self.traverseSuccessors(root_node, state)
self.handleLayers(root_node, state)
print ("\n *** Generated PROMISE Layers **** \n ")
@@ -1335,4 +1312,4 @@ class PromiseRtTranslator:
- #print (self.layer_size_str)
+
--
GitLab