graph: backend: dnnl: patterns: enable quantized gated mlp kernels

Author: TaoLv

src/graph/backend/dnnl/patterns/mlp.cpp

@@ -15,7 +15,6 @@
 *******************************************************************************/
 
 #include "graph/backend/dnnl/kernels/gated_mlp.hpp"
-#include "graph/backend/dnnl/kernels/large_partition.hpp"
 
 #include "graph/backend/dnnl/patterns/fusions.hpp"
 #include "graph/backend/dnnl/patterns/pattern_matcher_pass.hpp"
@@ -87,7 +86,7 @@ DNNL_BACKEND_REGISTER_PATTERN_MATCHER_PASS(dnnl, gated_mlp)
                             in_edges_t {in_edge(0, pre_tc, 0)});
                 })
         .set_attr<FCreateKernel>("FCreateKernel", []() -> kernel_ptr {
-            return std::make_shared<gated_mlp_base_t>();
+            return std::make_shared<gated_mlp_base_t<false>>();
         });
 
 // gated mlp with swish decomposed to sigmoid and multiply.
@@ -131,7 +130,7 @@ DNNL_BACKEND_REGISTER_PATTERN_MATCHER_PASS(dnnl, gated_mlp_v1)
                             in_edges_t {in_edge(0, pre_tc, 0)});
                 })
         .set_attr<FCreateKernel>("FCreateKernel", []() -> kernel_ptr {
-            return std::make_shared<gated_mlp_base_t>();
+            return std::make_shared<gated_mlp_base_t<false>>();
         });
 
 /*
@@ -195,7 +194,7 @@ DNNL_BACKEND_REGISTER_PATTERN_MATCHER_PASS(dnnl, quantized_gated_mlp)
                     pgraph->append_op(graph::op_kind::MatMul, fc_down_edges);
                 })
         .set_attr<FCreateKernel>("FCreateKernel", []() -> kernel_ptr {
-            return std::make_shared<larger_partition_kernel_t>();
+            return std::make_shared<gated_mlp_base_t<true>>();
         });
 
 // quantized gated mlp with swish decomposed to sigmoid and multiply.
@@ -246,7 +245,7 @@ DNNL_BACKEND_REGISTER_PATTERN_MATCHER_PASS(dnnl, quantized_gated_mlp_v1)
                     pgraph->append_op(graph::op_kind::MatMul, fc_down_edges);
                 })
         .set_attr<FCreateKernel>("FCreateKernel", []() -> kernel_ptr {
-            return std::make_shared<larger_partition_kernel_t>();
+            return std::make_shared<gated_mlp_base_t<true>>();
         });
 
 DNNL_BACKEND_REGISTER_PATTERN_DEF_END