diff --git a/tests/end2end/test_vitis_end2end_cnv_w1a1.py b/tests/end2end/test_vitis_end2end_cnv_w1a1.py index 35be815743d9625544cbe5b64396c7041f67ea71..4889da7a1760d7ac6f5ee12c54c2c7a7569b4790 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 672ea93a037a47b1fcf5a5b1b866425d79c8b134..8f1e2f5ab430b062d990e2c2846c4aa3b8b22211 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():