From 4584f0ba760367f6d7e62e868afebd00d8bce846 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 17 Sep 2021 18:37:08 +0200
Subject: [PATCH] [Test] bugfix in test_set_folding
---
tests/fpgadataflow/test_set_folding.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/fpgadataflow/test_set_folding.py b/tests/fpgadataflow/test_set_folding.py
index f268611c2..8f4d57d3f 100644
--- a/tests/fpgadataflow/test_set_folding.py
+++ b/tests/fpgadataflow/test_set_folding.py
@@ -98,7 +98,8 @@ def make_multi_fclayer_model(ch, wdt, adt, tdt, nnodes):
model.set_tensor_datatype("outp", adt)
for i in range(1, nnodes + 1):
- model.graph.value_info.append(tensors[i])
+ if tensors[i].name != "outp":
+ model.graph.value_info.append(tensors[i])
model.set_initializer("weights_" + str(i - 1), W)
model.set_initializer("thresh_" + str(i - 1), T)
model.set_tensor_datatype("weights_" + str(i - 1), wdt)
--
GitLab