diff --git a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py index 2ab0a27ae50be3abe3bdaf6501f4667d7c735103..dc102a0e550544a61536ea6fbfc8b0dba0c7457b 100644 --- a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py +++ b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py @@ -48,7 +48,7 @@ from finn.transformation.general import GiveUniqueNodeNames from finn.transformation.infer_datatypes import InferDataTypes from finn.transformation.infer_shapes import InferShapes from finn.transformation.lower_convs_to_matmul import LowerConvsToMatMul -from finn.util.basic import gen_finn_dt_tensor, is_finn_op +from finn.util.basic import gen_finn_dt_tensor # conv_config kernel_size,stride, pad @@ -153,17 +153,6 @@ def test_convert_to_hls_conv_layer(conv_config, depthwise, exec_mode): else: raise Exception("Unknown exec_mode") - for idx, n in enumerate(new_model.graph.node): - if is_finn_op(n.domain): - inst = getCustomOp(n) - if inst.get_nodeattr("backend") == "fpgadataflow": - inst.set_nodeattr( - "rtlsim_trace", - "/scratch/users/mirzam/build_files/trace_" + str(idx) + ".vcd", - ) - - new_model.save("/scratch/users/mirzam/build_files/model.onnx") - x = gen_finn_dt_tensor(idt, input_shape) inp_dict = {model.graph.input[0].name: x} assert oxe.compare_execution(model, new_model, inp_dict)