From fccd5eeac8666cce817cdc6c0dfc731dc0bf5270 Mon Sep 17 00:00:00 2001
From: gongshaotian <gstian5555@outlook.com>
Date: Tue, 14 Nov 2023 06:45:24 +0000
Subject: [PATCH] Explicitly add reserve_space output tensor

---
 .../fluid/pir/transforms/fusion/fused_gemm_epilogue_pass.cc   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/pir/transforms/fusion/fused_gemm_epilogue_pass.cc b/paddle/fluid/pir/transforms/fusion/fused_gemm_epilogue_pass.cc
index 5711731fd98eb1..f8cc654333ad08 100644
--- a/paddle/fluid/pir/transforms/fusion/fused_gemm_epilogue_pass.cc
+++ b/paddle/fluid/pir/transforms/fusion/fused_gemm_epilogue_pass.cc
@@ -143,7 +143,7 @@ class FusedLinearGeluPattern
                  {"activation", act_attr}}});
     fused_gemm_epilogue_gelu(
         {&res.Tensor("x"), &res.Tensor("w"), &res.Tensor("bias")},
-        {&res.Tensor("out")});
+        {&res.Tensor("out"), &res.Tensor("reserve_space")});
   }
 };
 class FusedLinearReluPattern
@@ -181,7 +181,7 @@ class FusedLinearReluPattern
                  {"activation", act_attr}}});
     fused_gemm_epilogue_relu(
         {&res.Tensor("x"), &res.Tensor("w"), &res.Tensor("bias")},
-        {&res.Tensor("out")});
+        {&res.Tensor("out"), &res.Tensor("reserve_space")});
   }
 };