From ba4bcaf4311fd5bce10a5437beccd290fcc959c6 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 15 Jun 2020 10:53:49 +0100
Subject: [PATCH] [Core] Add validation fct for quant values for outputs in
onnx execution
---
src/finn/core/onnx_exec.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/src/finn/core/onnx_exec.py b/src/finn/core/onnx_exec.py
index b5193a015..0bacbc8b0 100644
--- a/src/finn/core/onnx_exec.py
+++ b/src/finn/core/onnx_exec.py
@@ -39,7 +39,7 @@ from finn.core.remote_exec import remote_exec
from finn.core.rtlsim_exec import rtlsim_exec
from finn.custom_op.registry import getCustomOp
import finn.analysis.topology as ta
-from finn.util.basic import update_execution_context
+from finn.util.basic import validate_quant_values
def execute_node(node, context, graph):
@@ -161,8 +161,13 @@ def execute_onnx(model, input_dict, return_full_exec_context=False):
# topologically sorted
for node in graph.node:
# call util function match input values to quantization annotation
- execution_context = update_execution_context(model, node, execution_context)
+ execution_context = validate_quant_values(
+ model, node.input, execution_context
+ )
execute_node(node, execution_context, graph)
+ execution_context = validate_quant_values(
+ model, node.output, execution_context
+ )
elif model_exec_mode == "remote_pynq":
# use remote exec metadata built into model to execute on a remote PYNQ
remote_exec(model, execution_context)
--
GitLab