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

Follow comments.

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