diff --git a/paddle/function/CMakeLists.txt b/paddle/function/CMakeLists.txt index 9187294a49c7a61ee7bd7b9fe5910e504ff75ae0..c572a9d433bc16e6733b8fc9367970bef28e699a 100644 --- a/paddle/function/CMakeLists.txt +++ b/paddle/function/CMakeLists.txt @@ -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) diff --git a/paddle/function/EigenGemm.cpp b/paddle/function/EigenGemm.cpp index 0b4220fcbef175fc45c2ccfe6edf878b30953cb7..674141ed39b7f5573948348e3ba3bb526ae43c66 100644 --- a/paddle/function/EigenGemm.cpp +++ b/paddle/function/EigenGemm.cpp @@ -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; } } };