fix bug in fused_gemm_epilogue_op.cc (#50980)

064a5434 · yuehuayingxueluo · GitHub · e4fbb286 · 064a5434
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc paddle/fluid/operators/fused/fused_gemm_epilogue_op.cc +1 -1

未找到文件。
--- 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"));