From c97d0e88c6b51a965d020cc1d1a0785ab5eb5f5c Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 22 Jul 2020 13:30:35 +0100
Subject: [PATCH] [Test] Add exp_cycle calculation test to addstreams test
---
tests/fpgadataflow/test_fpgadataflow_addstreams.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/tests/fpgadataflow/test_fpgadataflow_addstreams.py b/tests/fpgadataflow/test_fpgadataflow_addstreams.py
index f94784457..38f3584e9 100644
--- a/tests/fpgadataflow/test_fpgadataflow_addstreams.py
+++ b/tests/fpgadataflow/test_fpgadataflow_addstreams.py
@@ -27,6 +27,7 @@
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import pytest
+import numpy as np
from onnx import TensorProto, helper
@@ -44,6 +45,8 @@ from finn.util.basic import gen_finn_dt_tensor
from finn.transformation.fpgadataflow.replace_verilog_relpaths import (
ReplaceVerilogRelPaths,
)
+from finn.custom_op.registry import getCustomOp
+from finn.analysis.fpgadataflow.exp_cycles_per_layer import exp_cycles_per_layer
def make_addstreams_modelwrapper(ch, pe, idt):
@@ -125,3 +128,10 @@ def test_fpgadataflow_addstreams(idt, ch, fold, exec_mode):
y_produced = y_produced.reshape(y_expected.shape)
assert (y_produced == y_expected).all(), exec_mode + " failed"
+
+ if exec_mode == "rtlsim":
+ inst = getCustomOp(model.graph.node[0])
+ sim_cycles = inst.get_nodeattr("sim_cycles")
+ exp_cycles_dict = model.analysis(exp_cycles_per_layer)
+ exp_cycles = exp_cycles_dict[str(model.graph.node[0])]
+ assert np.isclose(exp_cycles, sim_cycles, atol=10)
--
GitLab