Skip to content
Snippets Groups Projects
Commit c2d46c27 authored by mmrahorovic's avatar mmrahorovic
Browse files

removed saving model & trace file used for debug purposes

parent cff96cf9
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
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