devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix another complex div edge case.

Browse Source 
libeigen/eigen!2257
This commit is contained in:
Antonio Sánchez
2026-03-06 21:37:26 +00:00
committed by Rasmus Munk Larsen
parent 5bacb5be9a
commit 20fce70e5a
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Eigen/src/Core/arch/Default/GenericPacketMathComplex.h
View File

@@ -29,11 +29,12 @@ EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS Packet pdiv_complex(const Pa
// are a pair length-2 SIMD vectors representing a single pair of complex
// numbers x = a + i*b, y = c + i*d.
const RealPacket one = pset1<RealPacket>(RealScalar(1));
const RealPacket y_flip = pcplxflip(y).v;
// We need to avoid dividing by Inf/Inf, so use a mask to carefully
// apply the scale.
const RealPacket mask = pcmp_lt(pabs(y.v), pabs(y_flip)); // |c| < |d|
const RealPacket y_scaled = pselect(mask, pdiv(y.v, y_flip), one);
const RealPacket abs_y = pabs(y.v);
const RealPacket abs_y_flip = pcplxflip(Packet(abs_y)).v;
const RealPacket mask = pcmp_lt(abs_y, abs_y_flip); // |c| < |d|
RealPacket y_scaled = pselect(mask, pdiv(abs_y, abs_y_flip), one);
y_scaled = por(y_scaled, pandnot(y.v, abs_y)); // copy signs in case |c| == |d|
RealPacket denom = pmul(y.v, y_scaled);
denom = padd(denom, pcplxflip(Packet(denom)).v); // c * c' + d * d'
Packet num = pmul(x, pconj(Packet(y_scaled))); // a * c' + b * d', -a * d + b * c