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;