diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py index ebca224389550929cebd542cf4201cf62481a169..f931f91c89f738899ff9e6584be81a3b2d542227 100644 --- a/tests/end2end/test_end2end_cnv_w1a1.py +++ b/tests/end2end/test_end2end_cnv_w1a1.py @@ -78,6 +78,7 @@ from finn.transformation.fpgadataflow.annotate_resources import AnnotateResource from finn.transformation.fpgadataflow.prepare_rtlsim import PrepareRTLSim from finn.transformation.fpgadataflow.insert_fifo import InsertFIFO from finn.core.throughput_test import throughput_test_rtlsim +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -317,6 +318,10 @@ def test_end2end_cnv_w1a1_synth_pynq_project(): ) model = model.transform(SynthPYNQProject()) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_cnv_w1a1_synth.onnx") diff --git a/tests/end2end/test_end2end_cnv_w2a2.py b/tests/end2end/test_end2end_cnv_w2a2.py index 2e34990007677ce1b8e0a9ae4a1781d4527ee040..239094a3c931c16b3afe8d1874345e4dc90334ef 100644 --- a/tests/end2end/test_end2end_cnv_w2a2.py +++ b/tests/end2end/test_end2end_cnv_w2a2.py @@ -77,6 +77,7 @@ from finn.transformation.fpgadataflow.annotate_resources import AnnotateResource from finn.transformation.fpgadataflow.prepare_rtlsim import PrepareRTLSim from finn.transformation.fpgadataflow.insert_fifo import InsertFIFO from finn.core.throughput_test import throughput_test_rtlsim +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -315,6 +316,10 @@ def test_end2end_cnv_w2a2_synth_pynq_project(): ) model = model.transform(SynthPYNQProject()) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_cnv_w2a2_synth.onnx") diff --git a/tests/end2end/test_end2end_tfc_w1a1.py b/tests/end2end/test_end2end_tfc_w1a1.py index f69f1b74d5f2ae4863ff0969c9e47c74ea78a09c..1a3cc4f1bb9232809e864bb0c784498534f63631 100644 --- a/tests/end2end/test_end2end_tfc_w1a1.py +++ b/tests/end2end/test_end2end_tfc_w1a1.py @@ -79,6 +79,7 @@ from finn.transformation.fpgadataflow.annotate_resources import AnnotateResource from finn.transformation.fpgadataflow.prepare_rtlsim import PrepareRTLSim from finn.core.throughput_test import throughput_test_rtlsim import finn.util.vcd as vcd +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -314,6 +315,10 @@ def test_end2end_tfc_w1a1_synth_pynq_project(): ) model = model.transform(SynthPYNQProject()) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_tfc_w1a1_synth.onnx") diff --git a/tests/end2end/test_end2end_tfc_w1a2.py b/tests/end2end/test_end2end_tfc_w1a2.py index 755650e3d4da6947a93495fd5bbe0464cf485193..0f066cb06c53ce118d0a357fce0999299d7f3305 100644 --- a/tests/end2end/test_end2end_tfc_w1a2.py +++ b/tests/end2end/test_end2end_tfc_w1a2.py @@ -74,6 +74,7 @@ from finn.util.basic import pynq_part_map from finn.util.test import get_test_model_trained, load_test_checkpoint_or_skip from finn.transformation.fpgadataflow.annotate_resources import AnnotateResources from finn.transformation.fpgadataflow.prepare_rtlsim import PrepareRTLSim +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -270,6 +271,10 @@ def test_end2end_tfc_w1a2_synth_pynq_project(): ) model = model.transform(SynthPYNQProject()) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_tfc_w1a2_synth.onnx") diff --git a/tests/end2end/test_end2end_tfc_w2a2.py b/tests/end2end/test_end2end_tfc_w2a2.py index 4b2dd9ef01850897d95ede1214f87e9aa5b79f63..6eb613fc877b6e6801140f2a03c3a9509c08c0cb 100644 --- a/tests/end2end/test_end2end_tfc_w2a2.py +++ b/tests/end2end/test_end2end_tfc_w2a2.py @@ -74,6 +74,7 @@ from finn.util.basic import pynq_part_map from finn.util.test import get_test_model_trained, load_test_checkpoint_or_skip from finn.transformation.fpgadataflow.annotate_resources import AnnotateResources from finn.transformation.fpgadataflow.prepare_rtlsim import PrepareRTLSim +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -270,6 +271,10 @@ def test_end2end_tfc_w2a2_synth_pynq_project(): ) model = model.transform(SynthPYNQProject()) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_tfc_w2a2_synth.onnx") diff --git a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py index 25cafcfd4c552fb368cbaca2d1d2714cf2d14011..a272fadc12f095034693e555e4d791e9e73262ab 100644 --- a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py +++ b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py @@ -63,6 +63,7 @@ from finn.transformation.lower_convs_to_matmul import LowerConvsToMatMul from finn.transformation.streamline.reorder import MakeMaxPoolNHWC from finn.transformation.infer_data_layouts import InferDataLayouts from finn.transformation.fpgadataflow.annotate_cycles import AnnotateCycles +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] @@ -178,6 +179,10 @@ def test_end2end_zynqbuild_cnv_w1a1_build(): ) model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns)) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_zynqbuild_cnv_w1a1_build.onnx") diff --git a/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py b/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py index ff2af70731d9248dd2593db5be9e465fa86157dd..8b298d5644d6d6cda038e8ca1757be7538ba9804 100644 --- a/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py +++ b/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py @@ -64,6 +64,7 @@ from finn.util.test import get_test_model_trained, load_test_checkpoint_or_skip from finn.transformation.fpgadataflow.annotate_resources import AnnotateResources from finn.transformation.infer_data_layouts import InferDataLayouts from finn.transformation.fpgadataflow.make_zynq_proj import ZynqBuild +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -160,6 +161,10 @@ def test_end2end_zynqbuild_tfc_w1a1_build(): ) model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns)) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_zynqbuild_tfc_w1a1_build.onnx") diff --git a/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py b/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py index 7b28090854adbbcb6f400f73c2b6f6557f540e5e..bdb24d82dd639abe52aac9688b0b98430f72cabd 100644 --- a/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py +++ b/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py @@ -58,6 +58,7 @@ from finn.util.basic import pynq_part_map from finn.util.test import get_test_model_trained, load_test_checkpoint_or_skip from finn.transformation.fpgadataflow.annotate_resources import AnnotateResources from finn.transformation.fpgadataflow.make_zynq_proj import ZynqBuild +import warnings build_dir = "/tmp/" + os.environ["FINN_INST_NAME"] test_pynq_board = os.getenv("PYNQ_BOARD", default="Pynq-Z1") @@ -149,6 +150,10 @@ def test_end2end_zynqbuild_tfc_w2a2_build(): ) model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns)) model = model.transform(AnnotateResources("synth")) + warnings.warn( + "Post-synthesis resources (excluding shell): " + + model.get_metadata_prop("res_total_synth") + ) model.save(build_dir + "/end2end_zynqbuild_tfc_w2a2_build.onnx")