Follow comments.

revert-3824-remove_grad_op_type
hedaoyuan 8 years ago
parent 6ba04dcd11
commit 430e0e418b

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

Loading…
Cancel
Save