未验证 提交 0a704e14 编写于 作者: L limingshu 提交者: GitHub

Fix bugs in fused_linear_epilogue (#54512)

上级 8f07f503
......@@ -90,9 +90,9 @@ class FusedGemmEpilogueKernel : public framework::OpKernel<T> {
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<T>(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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册