From d7cce317feeab24ffd6f65e97bc32cd9d88a7ef0 Mon Sep 17 00:00:00 2001
From: Wangzheee <634486483@qq.com>
Date: Mon, 21 Aug 2023 10:20:20 +0800
Subject: [PATCH] fix_remove_padding_recover_padding_pass (#56439)

---
 .../fluid/framework/ir/remove_padding_recover_padding_pass.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/framework/ir/remove_padding_recover_padding_pass.cc b/paddle/fluid/framework/ir/remove_padding_recover_padding_pass.cc
index ffbff731836..b404d023d48 100644
--- 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) {
-- 
GitLab