From 710f965ba79ec998c0bab1e6bc54f0c5c1dd39f5 Mon Sep 17 00:00:00 2001 From: "yuan.xiong" Date: Mon, 2 Dec 2024 09:36:55 +0800 Subject: [PATCH] Change to skip mark_node in shape flow for broadcast node if dependency node is shape_of Signed-off-by: yuan.xiong --- .../src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp b/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp index bb165babb44427..d7cfdc2ceb3278 100644 --- a/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp +++ b/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp @@ -3,7 +3,6 @@ // #include "broadcast_inst.h" -#include "reorder_inst.h" #include "shape_of_inst.h" #include "read_value_inst.h" #include "reshape_inst.h" @@ -28,8 +27,8 @@ void mark_shape_of_subgraphs::look_for_shape_of_subgraph(program_node& node) { bool has_shape_of_subgraph_dep = false; for (auto& dependency : node.get_dependencies()) { if (dependency.first->is_in_shape_of_subgraph()) { - // skip mark_node for reorder node if dependency node is broadcast - if (dependency.first->is_type() && node.is_type()) { + // skip mark_node for broadcast node if dependency node is shape_of + if (dependency.first->is_type() && node.is_type()) { break; } has_shape_of_subgraph_dep = true;