[XPU] fc fusion supports sigmoid, swish and relu6 (#54486)

9b2bcfd6 · wz1qqx · GitHub · 924ca81d · 9b2bcfd6
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc +3 -0

未找到文件。
--- a/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/xpu/fc_xpu_fuse_pass.cc
@@ -257,6 +257,9 @@ void FcXPUFusePass::ApplyImpl(ir::Graph* graph) const {
                 "relu",
                 "gelu",
                 "tanh",
+                 "sigmoid",
+                 "swish",
+                 "relu6",
                 "",
             }) {
          found_subgraph_count +=