From: Vladimir Paramuzov <vladimir.paramuzov@intel.com>
Date: Tue, 9 Jun 2020 09:26:28 +0000 (+0300)
Subject: [IE CLDNN] Better error message when output is not found (#824)
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b4546ad1e044357d405c7215c01859f3f983289f;p=platform%2Fupstream%2Fdldt.git

[IE CLDNN] Better error message when output is not found (#824)
---

diff --git a/inference-engine/src/cldnn_engine/cldnn_program.cpp b/inference-engine/src/cldnn_engine/cldnn_program.cpp
index 19abf81..5829900 100644
--- a/inference-engine/src/cldnn_engine/cldnn_program.cpp
+++ b/inference-engine/src/cldnn_engine/cldnn_program.cpp
@@ -4910,6 +4910,12 @@ void Program::AddOutputPrimitive(cldnn::topology& topology, std::string outputNa
 
     // Find correct output ID. Start with name stored in IR.
     std::string outputID = outLayerName;
+    // This can happen when an output has invalid connections with previous layer and
+    // it's not handled by CreateSingleLayerPrimitive method
+    if (primitiveIDs.find(outLayerName) == primitiveIDs.end()) {
+        THROW_IE_EXCEPTION << "Can't find output with name " << outLayerName;
+    }
+
     std::string finalID = primitiveIDs.at(outLayerName);
 
     while (outputID != finalID) {