GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-13 20:26:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use numext::fma for sparse x dense dot product.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2025-07-02 15:51:28 -07:00
parent cc0be00435
commit 8ac2fb077d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Eigen/src/SparseCore/SparseDot.h
View File

@ -36,10 +36,10 @@ inline typename internal::traits<Derived>::Scalar SparseMatrixBase<Derived>::dot
Scalar res1(0); Scalar res1(0);
Scalar res2(0); Scalar res2(0);
for (; i; ++i) { for (; i; ++i) {
res1 += numext::conj(i.value()) * other.coeff(i.index()); res1 = numext::fma(numext::conj(i.value()), other.coeff(i.index()), res1);
++i; ++i;
if (i) { if (i) {
res2 += numext::conj(i.value()) * other.coeff(i.index()); res2 = numext::fma(numext::conj(i.value()), other.coeff(i.index()), res2);
} }
} }
return res1 + res2; return res1 + res2;