[NPUW] Add new compute pattern (#28935)

Author: smirnov-alexey

src/plugins/intel_npu/src/plugin/npuw/partitioning/online/compiler.cpp

@@ -27,7 +27,8 @@ static const std::map<std::string, std::string> ISOL_PRESETS = {{"COMPUTE",
                                                                  "P:DQMatMulGQi4/compute,P:DQMatMulCWi4/compute,"
                                                                  "P:DQMatMulConv/compute,"
                                                                  "P:VocabMatMul/compute,"
-                                                                 "P:RMSNorm/compute,P:RMSNorm2/compute"}};
+                                                                 "P:RMSNorm/compute,P:RMSNorm2/compute,"
+                                                                 "P:VariadicSplit/compute"}};
 }
 
 // For missing declaration warning

src/plugins/intel_npu/src/plugin/npuw/partitioning/online/snapshot.cpp

@@ -478,6 +478,7 @@ void Snapshot::earlyRegroup() {
             HNDL(DQMatMulGQi4);
             HNDL(DQMatMulConv);
             HNDL(VocabMatMul);
+            HNDL(VariadicSplit);
 #undef HNDL
         }
         }

src/plugins/intel_npu/src/plugin/npuw/partitioning/patterns/compute.cpp

@@ -406,6 +406,31 @@ RMSNorm2::RMSNorm2(const std::shared_ptr<ov::npuw::online::Snapshot>& snapshot,
     register_matcher(std::make_shared<opp::Matcher>(multiply, "TagRMSNorm2"), std::move(callback));
 }
 
+// TODO: visualize
+VariadicSplit::VariadicSplit(const std::shared_ptr<ov::npuw::online::Snapshot>& snapshot, const std::string& isol_tag) {
+    auto vsplit = opp::wrap_type<ov::op::v1::VariadicSplit>({opp::any_input(), opp::any_input(), opp::any_input()});
+    auto swish = opp::wrap_type<ov::op::v4::Swish>({vsplit});
+    auto multiply = opp::wrap_type<ov::op::v1::Multiply>({vsplit, swish});
+
+    auto node_to_gptr = snapshot->getNodeToGroupMap();
+
+    // Note: Use [=] to make sure the above objects stay alive in the callback
+    auto callback = [=](ov::pass::pattern::Matcher& m) {
+        auto& node_to_output = m.get_pattern_value_map();
+
+        auto matched_vsplit = node_to_output.at(vsplit).get_node_shared_ptr();
+        auto matched_swish = node_to_output.at(swish).get_node_shared_ptr();
+        auto matched_multiply = node_to_output.at(multiply).get_node_shared_ptr();
+
+        node_to_gptr->at(matched_vsplit)->isolate(isol_tag);
+        node_to_gptr->at(matched_swish)->isolate(isol_tag);
+        node_to_gptr->at(matched_multiply)->isolate(isol_tag);
+
+        return false;  // root hasn't changed
+    };
+    register_matcher(std::make_shared<opp::Matcher>(multiply, "TagVariadicSplit"), std::move(callback));
+}
+
 }  // namespace compute
 }  // namespace patterns
 }  // namespace npuw

src/plugins/intel_npu/src/plugin/npuw/partitioning/patterns/compute.hpp

@@ -69,6 +69,12 @@ class RMSNorm2 : public ov::pass::MatcherPass {
     RMSNorm2(const std::shared_ptr<ov::npuw::online::Snapshot>& snapshot, const std::string& isol_tag);
 };
 
+class VariadicSplit : public ov::pass::MatcherPass {
+public:
+    OPENVINO_MATCHER_PASS_RTTI("npuw::patterns::compute::VariadicSplit");
+    VariadicSplit(const std::shared_ptr<ov::npuw::online::Snapshot>& snapshot, const std::string& isol_tag);
+};
+
 }  // namespace compute
 }  // namespace patterns
 }  // namespace npuw