From 92b7ad4b2b9d75e6c519c46b63999e4465aa76b4 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 25 Aug 2020 11:19:09 +0200
Subject: [PATCH] [Test] move post-synth res annotation to own step
---
tests/end2end/test_vitis_end2end_cnv_w1a1.py | 9 ++++++++-
tests/end2end/test_vitis_end2end_tfc_w1a1.py | 9 ++++++++-
2 files changed, 16 insertions(+), 2 deletions(-)
diff --git a/tests/end2end/test_vitis_end2end_cnv_w1a1.py b/tests/end2end/test_vitis_end2end_cnv_w1a1.py
index 35be81574..4889da7a1 100644
--- a/tests/end2end/test_vitis_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_vitis_end2end_cnv_w1a1.py
@@ -185,12 +185,19 @@ def test_end2end_vitis_cnv_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED,
)
)
+ model.save(build_dir + "/end2end_vitis_cnv_w1a1_build.onnx")
+
+
+def test_end2end_vitis_cnv_w1a1_annotate_resources():
+ model = load_test_checkpoint_or_skip(
+ build_dir + "/end2end_vitis_cnv_w1a1_build.onnx"
+ )
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")
+ model.save(build_dir + "/end2end_vitis_cnv_w1a1_annotate_resources.onnx")
def test_end2end_vitis_cnv_w1a1_deploy_on_pynq():
diff --git a/tests/end2end/test_vitis_end2end_tfc_w1a1.py b/tests/end2end/test_vitis_end2end_tfc_w1a1.py
index 672ea93a0..8f1e2f5ab 100644
--- a/tests/end2end/test_vitis_end2end_tfc_w1a1.py
+++ b/tests/end2end/test_vitis_end2end_tfc_w1a1.py
@@ -172,12 +172,19 @@ def test_end2end_vitis_tfc_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED,
)
)
+ model.save(build_dir + "/end2end_vitis_tfc_w1a1_build.onnx")
+
+
+def test_end2end_vitis_tfc_w1a1_annotate_resources():
+ model = load_test_checkpoint_or_skip(
+ build_dir + "/end2end_vitis_tfc_w1a1_build.onnx"
+ )
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")
+ model.save(build_dir + "/end2end_vitis_tfc_w1a1_annotate_resources.onnx")
def test_end2end_vitis_tfc_w1a1_deploy_on_pynq():
--
GitLab