fix fc fuse proble

PaddlePaddle · Oct 20, 2021 · 7d64b34 · 7d64b34 · paddle-bot-old · Oct 20, 2021
1 parent 4bd1977
commit 7d64b34
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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"))