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