From 7ce8bc7733253085247d9019bff968b5b32378eb Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 10 Mar 2020 15:01:37 +0000
Subject: [PATCH] [Test] Add test cases for decoupled mode
---
tests/fpgadataflow/test_fpgadataflow_fclayer.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/fpgadataflow/test_fpgadataflow_fclayer.py b/tests/fpgadataflow/test_fpgadataflow_fclayer.py
index e5bbb85c3..c53b735dd 100644
--- a/tests/fpgadataflow/test_fpgadataflow_fclayer.py
+++ b/tests/fpgadataflow/test_fpgadataflow_fclayer.py
@@ -27,6 +27,7 @@
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import pytest
+import csv
import numpy as np
from onnx import TensorProto, helper
@@ -208,6 +209,7 @@ def test_fpgadataflow_fclayer_npysim(mem_mode, idt, wdt, act, nf, sf, mw, mh):
y_expected = y.reshape(oshape)
# execute model
y_produced = oxe.execute_onnx(model, input_dict)["outp"]
+
y_produced =y_produced.reshape(y_expected.shape)
assert (y_produced == y_expected).all(), "npysim failed"
--
GitLab