diff --git a/tests/end2end/test_vitis_end2end_cnv_w1a1.py b/tests/end2end/test_vitis_end2end_cnv_w1a1.py index c34a6880be03a184f89290d106b8464012e548f6..35be815743d9625544cbe5b64396c7041f67ea71 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 7fe44ae4c7daca0a19cd041e03e7a2a2ffcc1a54..672ea93a037a47b1fcf5a5b1b866425d79c8b134 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")