From 25499411efb47adcf242329b81e4dd59a3916597 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 13 Sep 2022 23:35:47 +0200
Subject: [PATCH] [Util] set property on SDP nodes to get full context
---
src/finn/util/test.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/util/test.py b/src/finn/util/test.py
index f5d3b1c30..bfe4aa0bb 100644
--- a/src/finn/util/test.py
+++ b/src/finn/util/test.py
@@ -180,6 +180,7 @@ def execute_parent(parent_path, child_path, input_tensor_npy, return_full_ctx=Fa
sdp_node = parent_model.get_nodes_by_op_type("StreamingDataflowPartition")[0]
sdp_node = getCustomOp(sdp_node)
sdp_node.set_nodeattr("model", child_path)
+ sdp_node.set_nodeattr("return_full_exec_context", 1 if return_full_ctx else 0)
ret = execute_onnx(parent_model, {iname: input_tensor_npy}, True)
if return_full_ctx:
return ret
--
GitLab