From 4a7a164e25fb593292a057f587d7e416174287d6 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 24 Aug 2020 15:40:52 +0200
Subject: [PATCH] [Test] bring back AnnotateResources to Vitis tests
---
tests/end2end/test_vitis_end2end_cnv_w1a1.py | 5 +++++
tests/end2end/test_vitis_end2end_tfc_w1a1.py | 6 ++++++
2 files changed, 11 insertions(+)
diff --git a/tests/end2end/test_vitis_end2end_cnv_w1a1.py b/tests/end2end/test_vitis_end2end_cnv_w1a1.py
index c34a6880b..35be81574 100644
--- a/tests/end2end/test_vitis_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_vitis_end2end_cnv_w1a1.py
@@ -185,6 +185,11 @@ def test_end2end_vitis_cnv_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED,
)
)
+ model = model.transform(AnnotateResources("synth"))
+ warnings.warn(
+ "Post-synthesis resources (excluding shell): "
+ + model.get_metadata_prop("res_total_synth")
+ )
model.save(build_dir + "/end2end_vitis_cnv_w1a1_build.onnx")
diff --git a/tests/end2end/test_vitis_end2end_tfc_w1a1.py b/tests/end2end/test_vitis_end2end_tfc_w1a1.py
index 7fe44ae4c..672ea93a0 100644
--- a/tests/end2end/test_vitis_end2end_tfc_w1a1.py
+++ b/tests/end2end/test_vitis_end2end_tfc_w1a1.py
@@ -64,6 +64,7 @@ from finn.transformation.fpgadataflow.vitis_build import VitisBuild, VitisOptStr
from finn.transformation.infer_data_layouts import InferDataLayouts
from finn.transformation.fpgadataflow.make_deployment import DeployToPYNQ
from pkgutil import get_data
+from finn.transformation.fpgadataflow.annotate_resources import AnnotateResources
from finn.core.onnx_exec import execute_onnx
import warnings
@@ -171,6 +172,11 @@ def test_end2end_vitis_tfc_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED,
)
)
+ model = model.transform(AnnotateResources("synth"))
+ warnings.warn(
+ "Post-synthesis resources (excluding shell): "
+ + model.get_metadata_prop("res_total_synth")
+ )
model.save(build_dir + "/end2end_vitis_tfc_w1a1_build.onnx")
--
GitLab