提交 430e0e41 编写于 作者: H hedaoyuan

Follow comments.

上级 6ba04dcd
......@@ -5,7 +5,9 @@ list(APPEND h_files Function.h)
list(APPEND cpp_files Function.cpp)
list(APPEND cpp_files BufferArg.cpp)
list(APPEND cpp_files GemmFunctor.cpp)
list(APPEND cpp_files EigenGemm.cpp)
if(USE_EIGEN_FOR_BLAS)
list(APPEND cpp_files EigenGemm.cpp)
endif(USE_EIGEN_FOR_BLAS)
if(WITH_GPU)
file(GLOB cu_files . *OpGpu.cu)
......
......@@ -77,8 +77,7 @@ struct EigenBlasGemm {
} else if (alpha == T(1) && beta == T(1)) {
c.device(device) += a.contract(b, dims);
} else {
c.device(device) =
c.constant(alpha) * a.contract(b, dims) + c.constant(beta) * c;
c.device(device) = alpha * a.contract(b, dims) + beta * c;
}
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册