fix_remove_padding_recover_padding_pass (#56439)

d7cce317 · Wangzheee · GitHub · da1a52e5 · d7cce317
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/framework/ir/remove_padding_recover_padding_pass.cc ...fluid/framework/ir/remove_padding_recover_padding_pass.cc +3 -1

未找到文件。
--- a/paddle/fluid/framework/ir/remove_padding_recover_padding_pass.cc
+++ b/paddle/fluid/framework/ir/remove_padding_recover_padding_pass.cc
@@ -453,7 +453,9 @@ void RemovePaddingRecoverPaddingPass::ApplyImpl(ir::Graph* graph) const {
    }

    if (PADDLE_GET_CONST(
-            int, matrix_multiply_op->Op()->GetAttr("x_num_col_dims")) != 2) {
+            int, matrix_multiply_op->Op()->GetAttr("x_num_col_dims")) != 2 &&
+        PADDLE_GET_CONST(
+            int, matrix_multiply_op->Op()->GetAttr("x_num_col_dims")) != -1) {
      check_flag = false;
    }
    if (!check_flag) {