Skip to content

Commit 2608577

Browse files
[GPU] Check conditions separately by user
1 parent e77238b commit 2608577

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_buffer_fusing.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -503,11 +503,15 @@ void prepare_buffer_fusing::run(program& p) {
503503
return;
504504
if (user->is_type<loop>() || user->is_type<non_max_suppression>())
505505
return;
506+
}
507+
for (auto user : node.get_users()) {
506508
if (user->is_type<reshape>()) {
507509
auto& reshape_node = user->as<reshape>();
508510
if (can_reshape_be_optimized(reshape_node))
509511
return;
510512
}
513+
}
514+
for (auto user : node.get_users()) {
511515
if (user->is_type<experimental_detectron_roi_feature_extractor>() && user->get_dependency_index(node) == 0)
512516
return;
513517
}

0 commit comments

Comments
 (0)