Skip to content

Commit

Permalink
[GPU] False mark shapeof subgraph when shapeof has multiple user (ope…
Browse files Browse the repository at this point in the history
…nvinotoolkit#28412)

### Details:
 - False mark shapeof subgraph when shapeof has multiple user
- This is workaround for regression from
openvinotoolkit#27786.

### Tickets:
 - 159953
  • Loading branch information
wilson-seok authored Jan 15, 2025
1 parent c2a2618 commit 28473b9
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ bool mark_shape_of_subgraphs::can_mark_node(const program_node& node) {
// skip mark_node for broadcast node if dependency nodes are data and shape_of
auto& dependencies = node.get_dependencies();
if (node.is_type<broadcast>() && dependencies.size() == 2) {
if (dependencies[0].first->is_type<data>() && dependencies[1].first->is_type<shape_of>())
if (dependencies[0].first->is_type<data>() && dependencies[1].first->is_type<shape_of>() && (dependencies[1].first->get_users().size() == 1))
return false;
}

Expand Down

0 comments on commit 28473b9

Please sign in to comment.