diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py index d7f59ef35aaf61891937dcaa105cf1392133e732..8046d9bb8538f87eb6a76614b6fa9101931bde82 100644 --- a/tests/end2end/test_end2end_cnv_w1a1.py +++ b/tests/end2end/test_end2end_cnv_w1a1.py @@ -178,7 +178,7 @@ def test_end2end_cnv_w1a1_gen_hls_ip(): def test_end2end_cnv_w1a1_ip_stitch(): model = ModelWrapper(build_dir + "/end2end_cnv_w1a1_ipgen.onnx") model = model.transform(ReplaceVerilogRelPaths()) - model = model.transform(CreateStitchedIP(test_fpga_part)) + model = model.transform(CreateStitchedIP(test_fpga_part, target_clk_ns)) model.save(build_dir + "/end2end_cnv_w1a1_ipstitch.onnx") diff --git a/tests/end2end/test_end2end_tfc_w1a2.py b/tests/end2end/test_end2end_tfc_w1a2.py index ecc0d48a6af37bc2bdd48f9306976aa8582ca1b0..5ee2942845c41f4c6705b4ee3ecee89154d9faa9 100644 --- a/tests/end2end/test_end2end_tfc_w1a2.py +++ b/tests/end2end/test_end2end_tfc_w1a2.py @@ -156,7 +156,7 @@ def test_end2end_tfc_w1a2_gen_hls_ip(): def test_end2end_tfc_w1a2_ip_stitch(): model = ModelWrapper(build_dir + "/end2end_tfc_w1a2_ipgen.onnx") model = model.transform(ReplaceVerilogRelPaths()) - model = model.transform(CreateStitchedIP(test_fpga_part)) + model = model.transform(CreateStitchedIP(test_fpga_part, target_clk_ns)) model.save(build_dir + "/end2end_tfc_w1a2_ipstitch.onnx") diff --git a/tests/end2end/test_end2end_tfc_w2a2.py b/tests/end2end/test_end2end_tfc_w2a2.py index 8c13352d9e9d146d58d76b1cf1e17878f27513f5..2477318efd1e02b0865dadb40bad1a74ac8ea0b4 100644 --- a/tests/end2end/test_end2end_tfc_w2a2.py +++ b/tests/end2end/test_end2end_tfc_w2a2.py @@ -156,7 +156,7 @@ def test_end2end_tfc_w2a2_gen_hls_ip(): def test_end2end_tfc_w2a2_ip_stitch(): model = ModelWrapper(build_dir + "/end2end_tfc_w2a2_ipgen.onnx") model = model.transform(ReplaceVerilogRelPaths()) - model = model.transform(CreateStitchedIP(test_fpga_part)) + model = model.transform(CreateStitchedIP(test_fpga_part, target_clk_ns)) model.save(build_dir + "/end2end_tfc_w2a2_ipstitch.onnx") diff --git a/tests/fpgadataflow/test_fpgadataflow_fifo.py b/tests/fpgadataflow/test_fpgadataflow_fifo.py index fe27d7d4273be2b938e5bf70338bb374ce16b6b2..9158a0b0e72017b2468627e4f30fd3432c418d38 100644 --- a/tests/fpgadataflow/test_fpgadataflow_fifo.py +++ b/tests/fpgadataflow/test_fpgadataflow_fifo.py @@ -98,7 +98,7 @@ def test_fpgadataflow_fifo_rtlsim(Shape, folded_shape, depth, finn_dtype): assert y.shape == tuple(Shape), """The output shape is incorrect.""" model = model.transform(ReplaceVerilogRelPaths()) - model = model.transform(CreateStitchedIP(test_fpga_part)) + model = model.transform(CreateStitchedIP(test_fpga_part, target_clk_ns)) model = model.transform(MakePYNQProject(test_pynq_board)) model = model.transform(SynthPYNQProject()) model = model.transform(MakePYNQDriver()) diff --git a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py index f26ba428bf4cbe174c048dcd35a4d63dc58519ab..30b86d639ae52143320dfdfeb25488bae865b4d2 100644 --- a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py +++ b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py @@ -220,7 +220,7 @@ def test_fpgadataflow_ipstitch_do_stitch(): ip_stitch_model_dir + "/test_fpgadataflow_ipstitch_gen_model.onnx" ) model = model.transform(rvp.ReplaceVerilogRelPaths()) - model = model.transform(CreateStitchedIP(test_fpga_part)) + model = model.transform(CreateStitchedIP(test_fpga_part, 5)) vivado_stitch_proj_dir = model.get_metadata_prop("vivado_stitch_proj") assert vivado_stitch_proj_dir is not None assert os.path.isdir(vivado_stitch_proj_dir)