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

AVX2: Packet4ul has pmul, abs2

Browse Source
This commit is contained in:
Charles Schlosser
2023-04-26 16:22:16 +00:00
committed by Rasmus Munk Larsen
parent eb5ff1861a
commit 8999525c29
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Eigen/src/Core/arch/AVX/PacketMath.h
View File

@@ -276,8 +276,6 @@ template<> struct packet_traits<uint64_t> : default_packet_traits
HasTranspose = 0,
HasNegate = 0,
HasSqrt = 0,
HasMul = 0,
HasAbs2 = 0,
HasCmp = 1,
HasShift = 1
};
@@ -662,6 +660,10 @@ EIGEN_STRONG_INLINE Packet4l pmul<Packet4l>(const Packet4l& a, const Packet4l& b
__m256i high = _mm256_slli_epi64(_mm256_add_epi64(mul1, mul2), 32);
return _mm256_add_epi64(high, mul3);
}
template <>
EIGEN_STRONG_INLINE Packet4ul pmul<Packet4ul>(const Packet4ul& a, const Packet4ul& b) {
return (Packet4ul)pmul<Packet4l>((Packet4l)a, (Packet4l)b);
}
#endif
template<> EIGEN_STRONG_INLINE Packet8f pset1<Packet8f>(const float& from) { return _mm256_set1_ps(from); }