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():
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():
......
......@@ -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():
......
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