Skip to content
Snippets Groups Projects
Commit 92b7ad4b authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] move post-synth res annotation to own step

parent 4a7a164e
No related branches found
No related tags found
No related merge requests found
...@@ -185,12 +185,19 @@ def test_end2end_vitis_cnv_w1a1_build(): ...@@ -185,12 +185,19 @@ def test_end2end_vitis_cnv_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED, 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")) model = model.transform(AnnotateResources("synth"))
warnings.warn( warnings.warn(
"Post-synthesis resources (excluding shell): " "Post-synthesis resources (excluding shell): "
+ model.get_metadata_prop("res_total_synth") + 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(): def test_end2end_vitis_cnv_w1a1_deploy_on_pynq():
......
...@@ -172,12 +172,19 @@ def test_end2end_vitis_tfc_w1a1_build(): ...@@ -172,12 +172,19 @@ def test_end2end_vitis_tfc_w1a1_build():
strategy=VitisOptStrategy.BUILD_SPEED, 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")) model = model.transform(AnnotateResources("synth"))
warnings.warn( warnings.warn(
"Post-synthesis resources (excluding shell): " "Post-synthesis resources (excluding shell): "
+ model.get_metadata_prop("res_total_synth") + 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(): def test_end2end_vitis_tfc_w1a1_deploy_on_pynq():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment