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 AVX512FP16 build.

Browse Source 
libeigen/eigen!2100

Closes #3013
This commit is contained in:
Antonio Sánchez
2025-12-29 18:31:48 +00:00
committed by Charles Schlosser
parent 2ac496ff8a
commit 5793499a55
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Eigen/src/Core/arch/AVX512/PacketMathFP16.h
View File

@@ -735,9 +735,9 @@ EIGEN_STRONG_INLINE half predux<Packet8h>(const Packet8h& a) {
return half(_mm_reduce_add_ph(a)); return half(_mm_reduce_add_ph(a));
} }
// predux_half_dowto4 // predux_half
template <> template <>
EIGEN_STRONG_INLINE Packet16h predux_half_dowto4<Packet32h>(const Packet32h& a) { EIGEN_STRONG_INLINE Packet16h predux_half<Packet32h>(const Packet32h& a) {
const __m512i bits = _mm512_castph_si512(a); const __m512i bits = _mm512_castph_si512(a);
Packet16h lo = _mm256_castsi256_ph(_mm512_castsi512_si256(bits)); Packet16h lo = _mm256_castsi256_ph(_mm512_castsi512_si256(bits));
Packet16h hi = _mm256_castsi256_ph(_mm512_extracti64x4_epi64(bits, 1)); Packet16h hi = _mm256_castsi256_ph(_mm512_extracti64x4_epi64(bits, 1));
@@ -745,7 +745,7 @@ EIGEN_STRONG_INLINE Packet16h predux_half_dowto4<Packet32h>(const Packet32h& a)
} }
template <> template <>
EIGEN_STRONG_INLINE Packet8h predux_half_dowto4<Packet16h>(const Packet16h& a) { EIGEN_STRONG_INLINE Packet8h predux_half<Packet16h>(const Packet16h& a) {
Packet8h lo = _mm_castsi128_ph(_mm256_castsi256_si128(_mm256_castph_si256(a))); Packet8h lo = _mm_castsi128_ph(_mm256_castsi256_si128(_mm256_castph_si256(a)));
Packet8h hi = _mm_castps_ph(_mm256_extractf128_ps(_mm256_castph_ps(a), 1)); Packet8h hi = _mm_castps_ph(_mm256_extractf128_ps(_mm256_castph_ps(a), 1));
return padd(lo, hi); return padd(lo, hi);