Skip to content

Commit 37b4d7a

Browse files
authored
cherry pick fused gemm bugs (#54604)
Fix fused gemm bugs
1 parent 188871e commit 37b4d7a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/fused/fused_gemm_epilogue_op.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,9 @@ class FusedGemmEpilogueKernel : public framework::OpKernel<T> {
9090
int64_t K = trans_y ? y->dims()[1] : y->dims()[0];
9191
int64_t N = trans_y ? y->dims()[0] : y->dims()[1];
9292

93-
void* reserve_data = reserve_space ? reserve_space->data() : nullptr;
9493
auto fused_type =
9594
GetFwdFusedEpilogueType<T>(dev_ctx, activation, reserve_space);
95+
void* reserve_data = reserve_space ? reserve_space->data() : nullptr;
9696

9797
VLOG(6) << "x.shape={" << x->dims() << "}, y.shape={" << y->dims()
9898
<< "}, out.shape={" << out->dims() << "}, M=" << M << ", N=" << N

0 commit comments

Comments
 (0)