From 37b4d7a229983c6d8a62a27edb7a5c0223a29de4 Mon Sep 17 00:00:00 2001 From: limingshu <61349199+JamesLim-sy@users.noreply.github.com> Date: Wed, 14 Jun 2023 14:57:06 +0800 Subject: [PATCH] cherry pick fused gemm bugs (#54604) Fix fused gemm bugs --- 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