IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[VPU] Fix precisions for execGraph (#9767)

Browse Source 
ExecGraph didn't contain the parameter node and precisions
65013
This commit is contained in:
Anton Dudchenko 2022-02-03 13:20:59 +03:00 committed by GitHub
parent ccf4f4e420
commit 711d6de33b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/plugins/intel_myriad/graph_transformer/src/utils/runtime_graph.cpp
View File

@ -77,11 +77,22 @@ std::shared_ptr<ngraph::Function> buildRuntimeGraph(GraphMetaInfo& graphMetaInfo
const auto inputs = getInputs(stageMeta);
std::shared_ptr<ngraph::Node> node;
if (stageMeta.stageType == "Input") {
if (stageMeta.layerType == "Input") {
params.emplace_back(std::make_shared<ngraph::op::Parameter>());
for (const auto& outPrecision : stageMeta.outPrecisions) {
if (outPrecision != Precision::FP16) {
params.back()->set_element_type(InferenceEngine::details::convertPrecision(outPrecision));
break;
}
}
node = params.back();
} else if (stageMeta.childrenNum == 0) {
results.emplace_back(std::make_shared<ngraph::op::Result>(inputs.back()));
auto inputNode = inputs.back().get_node();
if (stageMeta.outPrecisions[0] != Precision::FP16) {
inputNode->set_output_type(0, InferenceEngine::details::convertPrecision(stageMeta.outPrecisions[0]),
ngraph::PartialShape(stageMeta.inputDims[0]));
}
node = results.back();
} else {
node = std::make_shared<ExecGraphInfoSerialization::ExecutionNode>(inputs, 0);
@ -102,7 +113,6 @@ std::shared_ptr<ngraph::Function> buildRuntimeGraph(GraphMetaInfo& graphMetaInfo
for (std::size_t i = 0; i < graphMetaInfo.stagesMeta.size(); i++) {
const auto stageMeta = graphMetaInfo.stagesMeta[i];
if (stageMeta.status == ie::InferenceEngineProfileInfo::LayerStatus::OPTIMIZED_OUT ||
stageMetaIndexToNode.count(i) != 0 ||
stageMeta.stageName == "<Receive-Tensor>" ||

1
src/tests/functional/plugin/myriad/shared_tests_instances/skip_tests_config.cpp
View File

@ -58,7 +58,6 @@ std::vector<std::string> disabledTestPatterns() {
// TODO: Issue 73501
R"(.*_Hetero_Behavior.*OVExecGraphImportExportTest.*)",
// TODO: Issue 65013
R"(.*LoadNetworkCreateDefaultExecGraphResult.*)",
R"(.*importExportedFunctionConstantResultOnly.*elementType=(f32|f16).*)",
// Not expected behavior
R"(.*Behavior.*ExecNetSetPrecision.*canSetOutputPrecisionForNetwork.*U8.*)",