From 7063debaf95e5815f0c5418a41c45376e567db93 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 29 Jul 2020 15:45:13 +0100
Subject: [PATCH] [Test] Add test for exp cycles to VVAU node test

---
 tests/fpgadataflow/test_convert_to_hls_conv_layer.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
index f87280673..81bb3a221 100644
--- a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
+++ b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py
@@ -136,3 +136,12 @@ def test_convert_to_hls_conv_layer(conv_config, depthwise, exec_mode):
         padding_node = new_model.get_nodes_by_op_type("FMPadding_Batch")[0]
         padding_inst = getCustomOp(padding_node)
         assert padding_inst.get_nodeattr("SIMD") == in_chn
+
+    if depthwise is True and exec_mode == "rtlsim":
+        node = new_model.get_nodes_by_op_type("Vector_Vector_Activate_Batch")[0]
+        inst = getCustomOp(node)
+        sim_cycles = inst.get_nodeattr("sim_cycles")
+        exp_cycles_dict = new_model.analysis(exp_cycles_per_layer)
+        exp_cycles = exp_cycles_dict[str(node)]
+        assert np.isclose(exp_cycles, sim_cycles, atol=11)
+        assert exp_cycles != 0
-- 
GitLab