未验证 提交 37b4d7a2 编写于 作者: L limingshu 提交者: GitHub

cherry pick fused gemm bugs (#54604)

Fix fused gemm bugs
上级 188871e6
...@@ -90,9 +90,9 @@ class FusedGemmEpilogueKernel : public framework::OpKernel<T> { ...@@ -90,9 +90,9 @@ class FusedGemmEpilogueKernel : public framework::OpKernel<T> {
int64_t K = trans_y ? y->dims()[1] : y->dims()[0]; int64_t K = trans_y ? y->dims()[1] : y->dims()[0];
int64_t N = trans_y ? y->dims()[0] : y->dims()[1]; int64_t N = trans_y ? y->dims()[0] : y->dims()[1];
void* reserve_data = reserve_space ? reserve_space->data() : nullptr;
auto fused_type = auto fused_type =
GetFwdFusedEpilogueType<T>(dev_ctx, activation, reserve_space); 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() VLOG(6) << "x.shape={" << x->dims() << "}, y.shape={" << y->dims()
<< "}, out.shape={" << out->dims() << "}, M=" << M << ", N=" << N << "}, 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.
先完成此消息的编辑!
想要评论请 注册