From 150f4d34a5361420995c30d0982137c6316bf7bc Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 18 Aug 2020 22:51:54 +0200
Subject: [PATCH] [Test] print post-synth res from all end2end tests
---
tests/end2end/test_end2end_cnv_w1a1.py | 5 +++++
tests/end2end/test_end2end_cnv_w2a2.py | 5 +++++
tests/end2end/test_end2end_tfc_w1a1.py | 5 +++++
tests/end2end/test_end2end_tfc_w1a2.py | 5 +++++
tests/end2end/test_end2end_tfc_w2a2.py | 5 +++++
tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py | 5 +++++
tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py | 5 +++++
tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py | 5 +++++
8 files changed, 40 insertions(+)
diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py
index ebca22438..f931f91c8 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 2e3499000..239094a3c 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 f69f1b74d..1a3cc4f1b 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 755650e3d..0f066cb06 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 4b2dd9ef0..6eb613fc8 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 25cafcfd4..a272fadc1 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 ff2af7073..8b298d564 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 7b2809085..bdb24d82d 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")
--
GitLab