diff --git a/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu b/paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu index d516d54e4d2a21d0d1c347221903d2394d5c5125..6d7319c1db2721e0ed030ae71e1b5c4e13fc1808 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