From c876f7b80daed2a4b8e4d232c7e51f42b7fd1a2a Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 19 Feb 2020 22:51:20 +0000
Subject: [PATCH] [Test] use eye matrix for ip_stitch fc model
---
tests/fpgadataflow/test_fpgadataflow_ip_stitch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
index e6ac6d045..421137f0a 100644
--- a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
+++ b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
@@ -71,7 +71,7 @@ def create_one_fc_model():
)
graph = helper.make_graph(
- nodes=[fc0], name="fclayer_graph", inputs=[inp], outputs=[outp],
+ nodes=[fc0], name="fclayer_graph", inputs=[inp], outputs=[outp]
)
model = helper.make_model(graph, producer_name="fclayer-model")
@@ -82,7 +82,7 @@ def create_one_fc_model():
model.set_tensor_datatype("w0", wdt)
# generate weights
- w0 = gen_finn_dt_tensor(wdt, (m, m))
+ w0 = np.eye(m, dtype=np.float32)
model.set_initializer("w0", w0)
model = model.transform(CreateDataflowPartition())
--
GitLab