fix fc fuse proble (#36568)

fc5db55a · Wilber · GitHub · 3f2d6a3f · fc5db55a
隐藏空白更改
内联并排

Showing with 6 addition and 1 deletion

paddle/fluid/framework/ir/fc_fuse_pass.cc paddle/fluid/framework/ir/fc_fuse_pass.cc +6 -1

未找到文件。
--- a/paddle/fluid/framework/ir/fc_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/fc_fuse_pass.cc
@@ -51,7 +51,12 @@ FCFusePass::FCFusePass() {
      .IsTensor()
      .End()
      .AddAttr("axis")
-      .IsNumGE(1)
+      .IsNumMatch<int>([](int axis) -> bool {
+        if (axis == -1 || axis >= 1) {
+          return true;
+        }
+        return false;
+      })
      .End();

  AddOpCompat(OpCompat("relu"))