提交 cf40daee 编写于 作者: J Jacek Czaja

- Building fix to softmax for inference

上级 8bfa1fa9
......@@ -493,7 +493,7 @@ template <typename T>
T Blas<platform::CPUDeviceContext>::ASUM(int n, T *x, int inc) const {
auto sum = static_cast<T>(0.0);
#ifdef PADDLE_WITH_MKLML
sum = Blas<T>::ASUM(n, x, inc);
sum = CBlas<T>::ASUM(n, x, inc);
#else
//TODO(jczaja): check if openblas does provide cblas_sasum/cblas_dasum
for (int c = 0; c < n; ++c) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册