From 5346a5226c3536baacc3010bf2a58d288bfd372e Mon Sep 17 00:00:00 2001 From: Andrei Molotkov Date: Thu, 23 Dec 2021 15:04:26 +0300 Subject: [PATCH] [GPU] Mark all nodes with dynamic shape as unsupported (#9372) --- src/plugins/intel_gpu/src/plugin/plugin.cpp | 24 ++++++++++++--------- 1 file changed, 14 insertions(+), 10 deletions(-) diff --git a/src/plugins/intel_gpu/src/plugin/plugin.cpp b/src/plugins/intel_gpu/src/plugin/plugin.cpp index bb69e631749..879f15976fa 100644 --- a/src/plugins/intel_gpu/src/plugin/plugin.cpp +++ b/src/plugins/intel_gpu/src/plugin/plugin.cpp @@ -390,37 +390,41 @@ QueryNetworkResult Plugin::QueryNetwork(const CNNNetwork& network, std::vector> nextLayerDependent; auto layerIsSupported = [&](std::shared_ptr node) { + if (node->is_dynamic()) { + return false; + } if (ngraph::is_type(node) || ngraph::is_type(node) || ngraph::is_type(node) || ngraph::is_type(node)) { return false; - } else if (ngraph::is_type(node)) { + } + if (ngraph::is_type(node)) { splitNames.emplace(node->get_friendly_name()); splits.push_back(node); return false; - } else if (ngraph::is_type(node)) { + } + if (ngraph::is_type(node)) { concatNames.emplace(node->get_friendly_name()); concats.push_back(node); return false; - } else if (ngraph::is_type(node) || + } + if (ngraph::is_type(node) || ngraph::is_type(node) || ngraph::is_type(node) || ngraph::is_type(node)) { depLayerNames.emplace(node->get_friendly_name()); nextLayerDependent.push_back(node); return false; - } else if (ngraph::is_type(node)) { + } + if (ngraph::is_type(node)) { constantsNames.emplace(node->get_friendly_name()); constants.push_back(node); return false; - } else if (prog.IsOpSupported(network, node) && - !ngraph::op::is_parameter(node) && - !ngraph::op::is_output(node)) { - return true; - } else { - return false; } + return prog.IsOpSupported(network, node) && + !ngraph::op::is_parameter(node) && + !ngraph::op::is_output(node); }; // Get ops after transformations and check if it's supported