From c2d46c2783e589c4dd8d7a5c15592f878199f65f Mon Sep 17 00:00:00 2001 From: mmrahorovic <mmrahorovic@hotmail.com> Date: Wed, 8 Jun 2022 17:19:34 +0100 Subject: [PATCH] removed saving model & trace file used for debug purposes --- .../fpgadataflow/test_convert_to_hls_conv_layer.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/tests/fpgadataflow/test_convert_to_hls_conv_layer.py b/tests/fpgadataflow/test_convert_to_hls_conv_layer.py index 2ab0a27ae..dc102a0e5 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) -- GitLab