From 83f3b835746bef3db23f5482b01fa14be6d7e474 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 24 May 2022 15:55:13 +0200
Subject: [PATCH] [Transform] add created ValueInfos to protobuf in InsertHook
---
src/finn/transformation/fpgadataflow/insert_hook.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/finn/transformation/fpgadataflow/insert_hook.py b/src/finn/transformation/fpgadataflow/insert_hook.py
index 22050c008..c1fce40c5 100644
--- a/src/finn/transformation/fpgadataflow/insert_hook.py
+++ b/src/finn/transformation/fpgadataflow/insert_hook.py
@@ -84,6 +84,9 @@ class InsertHook(Transformation):
)
# insert checksum node
graph.node.insert(node_ind + 1, chk_node)
+ # insert newly-created tensors
+ graph.value_info.append(chk_otensor)
+ graph.value_info.append(chk_result)
# set chk output tensor as new input tensor of second node
if len(consumers) == 1:
--
GitLab