diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc
index fc55b68fafeb6d68752ce6628fe2b3454e56719c..7aae6994458706f9fc077635909276cf912ffa1a 100644
--- a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc
+++ b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc
@@ -390,7 +390,7 @@ class FusedGemmEpilogueOpGradMaker : public framework::SingleGradOpMaker<T> {
     op->SetInput("X", this->Input("X"));
     op->SetInput("Y", this->Input("Y"));
     if (act_type != "none") {
-      op->SetInput("ReserveSpace", this->Input("ReserveSpace"));
+      op->SetInput("ReserveSpace", this->Output("ReserveSpace"));
     }
     op->SetInput("DOut", this->OutputGrad("Out"));