[GPU][Transformations] Fixed up predicates (#28607)

### Details: - *Fixed up predicates* Signed-off-by: Evgeniia Nugmanova <[email protected]>
openvinotoolkit · Jan 22, 2025 · 8a7c974 · 8a7c974
1 parent db41516
commit 8a7c974
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 15 deletions.
diff --git a/src/plugins/intel_gpu/src/plugin/transformations/convert_stridedslices_to_variadicsplit.cpp b/src/plugins/intel_gpu/src/plugin/transformations/convert_stridedslices_to_variadicsplit.cpp
@@ -31,7 +31,7 @@ ConvertStridedSlicesToVariadicSplit::ConvertStridedSlicesToVariadicSplit() {
                 return false;
             user_count++;
         }
-        return (user_count == num_users_to_fuse) && consumers_count(num_users_to_fuse);
+        return (user_count == num_users_to_fuse) && consumers_count(num_users_to_fuse)(output);
     };
 
     auto data_m = any_input();

diff --git a/...lugins/intel_gpu/src/plugin/transformations/unsqueeze_broadcast_reshape_matmul_fusion.cpp b/...lugins/intel_gpu/src/plugin/transformations/unsqueeze_broadcast_reshape_matmul_fusion.cpp
@@ -26,20 +26,16 @@ UnsqueezeBroadcastReshapeMatmulFusion::UnsqueezeBroadcastReshapeMatmulFusion() {
         return ov::as_type_ptr<ov::op::v1::Reshape>(output.get_node_shared_ptr()) == nullptr;
     };
 
-    auto unsqueeze_predicate = [](const ov::Output<ov::Node>& output) -> bool {
-        return rank_equals(5)(output) && consumers_count(1);
-    };
+    auto unsqueeze_predicate = rank_equals(5) && consumers_count(1);
 
     auto broadcast_predicate = [](const ov::Output<ov::Node>& output) -> bool {
         const auto broadcast = ov::as_type_ptr<ov::op::v3::Broadcast>(output.get_node_shared_ptr());
         if (!broadcast || broadcast->get_broadcast_spec().m_type != ov::op::BroadcastType::BIDIRECTIONAL)
             return false;
-        return rank_equals(5)(output) && consumers_count(1);
+        return rank_equals(5)(output) && consumers_count(1)(output);
     };
 
-    auto reshape_predicate = [](const ov::Output<ov::Node>& output) -> bool {
-        return rank_equals(4)(output) && consumers_count(1);
-    };
+    auto reshape_predicate = rank_equals(4) && consumers_count(1);
 
     auto input_a_m = any_input(not_reshape);
     auto input_b_m = wrap_type<ov::intel_gpu::op::KVCache>({any_input(), any_input()});

diff --git a/src/plugins/intel_gpu/src/plugin/transformations/unsqueeze_broadcast_reshape_sdpa_fusion.cpp b/src/plugins/intel_gpu/src/plugin/transformations/unsqueeze_broadcast_reshape_sdpa_fusion.cpp
@@ -23,20 +23,16 @@ using ov::pass::pattern::op::Or;
 UnsqueezeBroadcastReshapeSDPAFusion::UnsqueezeBroadcastReshapeSDPAFusion() {
     using namespace ov::pass::pattern;
 
-    auto unsqueeze_predicate = [](const ov::Output<ov::Node>& output) -> bool {
-        return rank_equals(5)(output) && consumers_count(1);
-    };
+    auto unsqueeze_predicate = rank_equals(5) && consumers_count(1);
 
     auto broadcast_predicate = [](const ov::Output<ov::Node>& output) -> bool {
         const auto broadcast = ov::as_type_ptr<ov::op::v3::Broadcast>(output.get_node_shared_ptr());
         if (!broadcast || broadcast->get_broadcast_spec().m_type != ov::op::BroadcastType::BIDIRECTIONAL)
             return false;
-        return rank_equals(5)(output) && consumers_count(1);
+        return rank_equals(5)(output) && consumers_count(1)(output);
     };
 
-    auto reshape_predicate = [](const ov::Output<ov::Node>& output) -> bool {
-        return rank_equals(4)(output) && consumers_count(1);
-    };
+    auto reshape_predicate = rank_equals(4) && consumers_count(1);
 
     auto input_a_m = any_input();
     auto input_attn_mask = any_input();