@ -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)
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;
}
};