From 0a704e1488934a148393100f8ac4663da16d3146 Mon Sep 17 00:00:00 2001 From: limingshu <61349199+JamesLim-sy@users.noreply.github.com> Date: Sat, 10 Jun 2023 16:01:25 +0800 Subject: [PATCH] Fix bugs in fused_linear_epilogue (#54512) --- paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu index d516d54e4d2..6d7319c1db2 100644 --- a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu +++ b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu @@ -90,9 +90,9 @@ class FusedGemmEpilogueKernel : public framework::OpKernel { int64_t K = trans_y ? y->dims()[1] : y->dims()[0]; int64_t N = trans_y ? y->dims()[0] : y->dims()[1]; - void* reserve_data = reserve_space ? reserve_space->data() : nullptr; auto fused_type = GetFwdFusedEpilogueType(dev_ctx, activation, reserve_space); + void* reserve_data = reserve_space ? reserve_space->data() : nullptr; VLOG(6) << "x.shape={" << x->dims() << "}, y.shape={" << y->dims() << "}, out.shape={" << out->dims() << "}, M=" << M << ", N=" << N -- GitLab