Skip to content
Snippets Groups Projects
Commit 7063deba authored by auphelia's avatar auphelia
Browse files

[Test] Add test for exp cycles to VVAU node test

parent d46e66ac
No related branches found
No related tags found
No related merge requests found
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment