| // This file is part of Eigen, a lightweight C++ template library |
| // for linear algebra. |
| // |
| // Copyright (C) 2016 Benoit Steiner (benoit.steiner.goog@gmail.com) |
| // |
| // This Source Code Form is subject to the terms of the Mozilla |
| // Public License v. 2.0. If a copy of the MPL was not distributed |
| // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. |
| |
| #ifndef EIGEN_PACKET_MATH_AVX512_H |
| #define EIGEN_PACKET_MATH_AVX512_H |
| |
| // IWYU pragma: private |
| #include "../../InternalHeaderCheck.h" |
| |
| namespace Eigen { |
| |
| namespace internal { |
| |
| #ifndef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD |
| #define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 8 |
| #endif |
| |
| #ifndef EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS |
| #define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS 32 |
| #endif |
| |
| #ifdef EIGEN_VECTORIZE_FMA |
| #ifndef EIGEN_HAS_SINGLE_INSTRUCTION_MADD |
| #define EIGEN_HAS_SINGLE_INSTRUCTION_MADD |
| #endif |
| #endif |
| |
| typedef __m512 Packet16f; |
| typedef __m512i Packet16i; |
| typedef __m512d Packet8d; |
| // TODO(rmlarsen): Add support for Packet8l. |
| #ifndef EIGEN_VECTORIZE_AVX512FP16 |
| typedef eigen_packet_wrapper<__m256i, 1> Packet16h; |
| #endif |
| typedef eigen_packet_wrapper<__m256i, 2> Packet16bf; |
| |
| template <> |
| struct is_arithmetic<__m512> { |
| enum { value = true }; |
| }; |
| template <> |
| struct is_arithmetic<__m512i> { |
| enum { value = true }; |
| }; |
| template <> |
| struct is_arithmetic<__m512d> { |
| enum { value = true }; |
| }; |
| |
| #ifndef EIGEN_VECTORIZE_AVX512FP16 |
| template <> |
| struct is_arithmetic<Packet16h> { |
| enum { value = true }; |
| }; |
| |
| template <> |
| struct packet_traits<half> : default_packet_traits { |
| typedef Packet16h type; |
| // There is no half-size packet for Packet16h. |
| typedef Packet16h half; |
| enum { |
| Vectorizable = 1, |
| AlignedOnScalar = 1, |
| size = 16, |
| |
| HasCmp = 1, |
| HasAdd = 1, |
| HasSub = 1, |
| HasMul = 1, |
| HasDiv = 1, |
| HasNegate = 1, |
| HasAbs = 1, |
| HasAbs2 = 0, |
| HasMin = 1, |
| HasMax = 1, |
| HasConj = 1, |
| HasSetLinear = 0, |
| HasSqrt = 1, |
| HasRsqrt = 1, |
| HasLog = 1, |
| HasLog1p = 1, |
| HasExp = 1, |
| HasExpm1 = 1, |
| HasBessel = 1, |
| HasNdtri = 1, |
| HasSin = EIGEN_FAST_MATH, |
| HasCos = EIGEN_FAST_MATH, |
| HasTanh = EIGEN_FAST_MATH, |
| HasErf = EIGEN_FAST_MATH, |
| HasBlend = 0, |
| HasRound = 1, |
| HasFloor = 1, |
| HasCeil = 1, |
| HasRint = 1 |
| }; |
| }; |
| #endif |
| |
| template <> |
| struct packet_traits<float> : default_packet_traits { |
| typedef Packet16f type; |
| typedef Packet8f half; |
| enum { |
| Vectorizable = 1, |
| AlignedOnScalar = 1, |
| size = 16, |
| |
| HasAbs = 1, |
| HasMin = 1, |
| HasMax = 1, |
| HasConj = 1, |
| HasBlend = 1, |
| HasSin = EIGEN_FAST_MATH, |
| HasCos = EIGEN_FAST_MATH, |
| HasACos = 1, |
| HasASin = 1, |
| HasATan = 1, |
| HasATanh = 1, |
| HasSqrt = 1, |
| HasRsqrt = 1, |
| HasLog = 1, |
| HasLog1p = 1, |
| HasExpm1 = 1, |
| HasNdtri = 1, |
| HasBessel = 1, |
| HasExp = 1, |
| HasReciprocal = EIGEN_FAST_MATH, |
| HasTanh = EIGEN_FAST_MATH, |
| HasErf = EIGEN_FAST_MATH, |
| HasCmp = 1, |
| HasDiv = 1, |
| HasRound = 1, |
| HasFloor = 1, |
| HasCeil = 1, |
| HasRint = 1 |
| }; |
| }; |
| template <> |
| struct packet_traits<double> : default_packet_traits { |
| typedef Packet8d type; |
| typedef Packet4d half; |
| enum { |
| Vectorizable = 1, |
| AlignedOnScalar = 1, |
| size = 8, |
| HasBlend = 1, |
| HasSqrt = 1, |
| HasRsqrt = 1, |
| HasLog = 1, |
| HasExp = 1, |
| HasATan = 1, |
| HasCmp = 1, |
| HasDiv = 1, |
| HasRound = 1, |
| HasFloor = 1, |
| HasCeil = 1, |
| HasRint = 1 |
| }; |
| }; |
| |
| template <> |
| struct packet_traits<int> : default_packet_traits { |
| typedef Packet16i type; |
| typedef Packet8i half; |
| enum { Vectorizable = 1, AlignedOnScalar = 1, HasBlend = 0, HasCmp = 1, HasDiv = 1, size = 16 }; |
| }; |
| |
| template <> |
| struct unpacket_traits<Packet16f> { |
| typedef float type; |
| typedef Packet8f half; |
| typedef Packet16i integer_packet; |
| typedef uint16_t mask_t; |
| enum { |
| size = 16, |
| alignment = Aligned64, |
| vectorizable = true, |
| masked_load_available = true, |
| masked_store_available = true, |
| masked_fpops_available = true |
| }; |
| }; |
| template <> |
| struct unpacket_traits<Packet8d> { |
| typedef double type; |
| typedef Packet4d half; |
| typedef uint8_t mask_t; |
| enum { |
| size = 8, |
| alignment = Aligned64, |
| vectorizable = true, |
| masked_load_available = true, |
| masked_store_available = true, |
| masked_fpops_available = true |
| }; |
| }; |
| template <> |
| struct unpacket_traits<Packet16i> { |
| typedef int type; |
| typedef Packet8i half; |
| enum { |
| size = 16, |
| alignment = Aligned64, |
| vectorizable = true, |
| masked_load_available = false, |
| masked_store_available = false |
| }; |
| }; |
| |
| #ifndef EIGEN_VECTORIZE_AVX512FP16 |
| template <> |
| struct unpacket_traits<Packet16h> { |
| typedef Eigen::half type; |
| typedef Packet8h half; |
| enum { |
| size = 16, |
| alignment = Aligned32, |
| vectorizable = true, |
| masked_load_available = false, |
| masked_store_available = false |
| }; |
| }; |
| #endif |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pset1<Packet16f>(const float& from) { |
| return _mm512_set1_ps(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pset1<Packet8d>(const double& from) { |
| return _mm512_set1_pd(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pset1<Packet16i>(const int& from) { |
| return _mm512_set1_epi32(from); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pset1frombits<Packet16f>(unsigned int from) { |
| return _mm512_castsi512_ps(_mm512_set1_epi32(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pset1frombits<Packet8d>(const numext::uint64_t from) { |
| return _mm512_castsi512_pd(_mm512_set1_epi64(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pzero(const Packet16f& /*a*/) { |
| return _mm512_setzero_ps(); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pzero(const Packet8d& /*a*/) { |
| return _mm512_setzero_pd(); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pzero(const Packet16i& /*a*/) { |
| return _mm512_setzero_si512(); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f peven_mask(const Packet16f& /*a*/) { |
| return _mm512_castsi512_ps(_mm512_set_epi32(0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i peven_mask(const Packet16i& /*a*/) { |
| return _mm512_set_epi32(0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1, 0, -1); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d peven_mask(const Packet8d& /*a*/) { |
| return _mm512_castsi512_pd(_mm512_set_epi32(0, 0, -1, -1, 0, 0, -1, -1, 0, 0, -1, -1, 0, 0, -1, -1)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pload1<Packet16f>(const float* from) { |
| #if (EIGEN_COMP_GNUC != 0) || (EIGEN_COMP_CLANG != 0) |
| // Inline asm here helps reduce some register spilling in TRSM kernels. |
| // See note in unrolls::gemm::microKernel in TrsmKernel.h |
| Packet16f ret; |
| __asm__("vbroadcastss %[mem], %[dst]" : [dst] "=v"(ret) : [mem] "m"(*from)); |
| return ret; |
| #else |
| return _mm512_broadcastss_ps(_mm_load_ps1(from)); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pload1<Packet8d>(const double* from) { |
| #if (EIGEN_COMP_GNUC != 0) || (EIGEN_COMP_CLANG != 0) |
| Packet8d ret; |
| __asm__("vbroadcastsd %[mem], %[dst]" : [dst] "=v"(ret) : [mem] "m"(*from)); |
| return ret; |
| #else |
| return _mm512_set1_pd(*from); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f plset<Packet16f>(const float& a) { |
| return _mm512_add_ps(_mm512_set1_ps(a), _mm512_set_ps(15.0f, 14.0f, 13.0f, 12.0f, 11.0f, 10.0f, 9.0f, 8.0f, 7.0f, |
| 6.0f, 5.0f, 4.0f, 3.0f, 2.0f, 1.0f, 0.0f)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d plset<Packet8d>(const double& a) { |
| return _mm512_add_pd(_mm512_set1_pd(a), _mm512_set_pd(7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 0.0)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i plset<Packet16i>(const int& a) { |
| return _mm512_add_epi32(_mm512_set1_epi32(a), _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f padd<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return _mm512_add_ps(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d padd<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return _mm512_add_pd(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i padd<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_add_epi32(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f padd<Packet16f>(const Packet16f& a, const Packet16f& b, uint16_t umask) { |
| __mmask16 mask = static_cast<__mmask16>(umask); |
| return _mm512_maskz_add_ps(mask, a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d padd<Packet8d>(const Packet8d& a, const Packet8d& b, uint8_t umask) { |
| __mmask8 mask = static_cast<__mmask8>(umask); |
| return _mm512_maskz_add_pd(mask, a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f psub<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return _mm512_sub_ps(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d psub<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return _mm512_sub_pd(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i psub<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_sub_epi32(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pnegate(const Packet16f& a) { |
| // NOTE: MSVC seems to struggle with _mm512_set1_epi32, leading to random results. |
| // The intel docs give it a relatively high latency as well, so we're probably |
| // better off with using _mm512_set_epi32 directly anyways. |
| const __m512i mask = |
| _mm512_set_epi32(0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, |
| 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000, 0x80000000); |
| return _mm512_castsi512_ps(_mm512_xor_epi32(_mm512_castps_si512(a), mask)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pnegate(const Packet8d& a) { |
| const __m512i mask = |
| _mm512_set_epi64(0x8000000000000000ULL, 0x8000000000000000ULL, 0x8000000000000000ULL, 0x8000000000000000ULL, |
| 0x8000000000000000ULL, 0x8000000000000000ULL, 0x8000000000000000ULL, 0x8000000000000000ULL); |
| return _mm512_castsi512_pd(_mm512_xor_epi64(_mm512_castpd_si512(a), mask)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pnegate(const Packet16i& a) { |
| return _mm512_sub_epi32(_mm512_setzero_si512(), a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pconj(const Packet16f& a) { |
| return a; |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pconj(const Packet8d& a) { |
| return a; |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pconj(const Packet16i& a) { |
| return a; |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmul<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return _mm512_mul_ps(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmul<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return _mm512_mul_pd(a, b); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pmul<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_mullo_epi32(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pdiv<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return _mm512_div_ps(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pdiv<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return _mm512_div_pd(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pdiv<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| Packet8i q_lo = pdiv<Packet8i>(_mm512_extracti64x4_epi64(a, 0), _mm512_extracti64x4_epi64(b, 0)); |
| Packet8i q_hi = pdiv<Packet8i>(_mm512_extracti64x4_epi64(a, 1), _mm512_extracti64x4_epi64(b, 1)); |
| return _mm512_inserti64x4(_mm512_castsi256_si512(q_lo), q_hi, 1); |
| } |
| |
| #ifdef EIGEN_VECTORIZE_FMA |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmadd(const Packet16f& a, const Packet16f& b, const Packet16f& c) { |
| return _mm512_fmadd_ps(a, b, c); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmadd(const Packet8d& a, const Packet8d& b, const Packet8d& c) { |
| return _mm512_fmadd_pd(a, b, c); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmsub(const Packet16f& a, const Packet16f& b, const Packet16f& c) { |
| return _mm512_fmsub_ps(a, b, c); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmsub(const Packet8d& a, const Packet8d& b, const Packet8d& c) { |
| return _mm512_fmsub_pd(a, b, c); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pnmadd(const Packet16f& a, const Packet16f& b, const Packet16f& c) { |
| return _mm512_fnmadd_ps(a, b, c); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pnmadd(const Packet8d& a, const Packet8d& b, const Packet8d& c) { |
| return _mm512_fnmadd_pd(a, b, c); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pnmsub(const Packet16f& a, const Packet16f& b, const Packet16f& c) { |
| return _mm512_fnmsub_ps(a, b, c); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pnmsub(const Packet8d& a, const Packet8d& b, const Packet8d& c) { |
| return _mm512_fnmsub_pd(a, b, c); |
| } |
| #endif |
| |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet16f pselect(const Packet16f& mask, const Packet16f& a, const Packet16f& b) { |
| __mmask16 mask16 = _mm512_cmpeq_epi32_mask(_mm512_castps_si512(mask), _mm512_setzero_epi32()); |
| return _mm512_mask_blend_ps(mask16, a, b); |
| } |
| |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet16i pselect(const Packet16i& mask, const Packet16i& a, const Packet16i& b) { |
| __mmask16 mask16 = _mm512_cmpeq_epi32_mask(mask, _mm512_setzero_epi32()); |
| return _mm512_mask_blend_epi32(mask16, a, b); |
| } |
| |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet8d pselect(const Packet8d& mask, const Packet8d& a, const Packet8d& b) { |
| __mmask8 mask8 = _mm512_cmp_epi64_mask(_mm512_castpd_si512(mask), _mm512_setzero_epi32(), _MM_CMPINT_EQ); |
| return _mm512_mask_blend_pd(mask8, a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmin<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| // Arguments are reversed to match NaN propagation behavior of std::min. |
| return _mm512_min_ps(b, a); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmin<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| // Arguments are reversed to match NaN propagation behavior of std::min. |
| return _mm512_min_pd(b, a); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pmin<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_min_epi32(b, a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmax<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| // Arguments are reversed to match NaN propagation behavior of std::max. |
| return _mm512_max_ps(b, a); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmax<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| // Arguments are reversed to match NaN propagation behavior of std::max. |
| return _mm512_max_pd(b, a); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pmax<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_max_epi32(b, a); |
| } |
| |
| // Add specializations for min/max with prescribed NaN progation. |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmin<PropagateNumbers, Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return pminmax_propagate_numbers(a, b, pmin<Packet16f>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmin<PropagateNumbers, Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return pminmax_propagate_numbers(a, b, pmin<Packet8d>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmax<PropagateNumbers, Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return pminmax_propagate_numbers(a, b, pmax<Packet16f>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmax<PropagateNumbers, Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return pminmax_propagate_numbers(a, b, pmax<Packet8d>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmin<PropagateNaN, Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return pminmax_propagate_nan(a, b, pmin<Packet16f>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmin<PropagateNaN, Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return pminmax_propagate_nan(a, b, pmin<Packet8d>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pmax<PropagateNaN, Packet16f>(const Packet16f& a, const Packet16f& b) { |
| return pminmax_propagate_nan(a, b, pmax<Packet16f>); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pmax<PropagateNaN, Packet8d>(const Packet8d& a, const Packet8d& b) { |
| return pminmax_propagate_nan(a, b, pmax<Packet8d>); |
| } |
| |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| template <int I_> |
| EIGEN_STRONG_INLINE Packet8f extract256(Packet16f x) { |
| return _mm512_extractf32x8_ps(x, I_); |
| } |
| template <int I_> |
| EIGEN_STRONG_INLINE Packet2d extract128(Packet8d x) { |
| return _mm512_extractf64x2_pd(x, I_); |
| } |
| EIGEN_STRONG_INLINE Packet16f cat256(Packet8f a, Packet8f b) { |
| return _mm512_insertf32x8(_mm512_castps256_ps512(a), b, 1); |
| } |
| EIGEN_STRONG_INLINE Packet16i cat256i(Packet8i a, Packet8i b) { |
| return _mm512_inserti32x8(_mm512_castsi256_si512(a), b, 1); |
| } |
| #else |
| // AVX512F does not define _mm512_extractf32x8_ps to extract _m256 from _m512 |
| template <int I_> |
| EIGEN_STRONG_INLINE Packet8f extract256(Packet16f x) { |
| return _mm256_castsi256_ps(_mm512_extracti64x4_epi64(_mm512_castps_si512(x), I_)); |
| } |
| |
| // AVX512F does not define _mm512_extractf64x2_pd to extract _m128 from _m512 |
| template <int I_> |
| EIGEN_STRONG_INLINE Packet2d extract128(Packet8d x) { |
| return _mm_castsi128_pd(_mm512_extracti32x4_epi32(_mm512_castpd_si512(x), I_)); |
| } |
| |
| EIGEN_STRONG_INLINE Packet16f cat256(Packet8f a, Packet8f b) { |
| return _mm512_castsi512_ps( |
| _mm512_inserti64x4(_mm512_castsi256_si512(_mm256_castps_si256(a)), _mm256_castps_si256(b), 1)); |
| } |
| EIGEN_STRONG_INLINE Packet16i cat256i(Packet8i a, Packet8i b) { |
| return _mm512_inserti64x4(_mm512_castsi256_si512(a), b, 1); |
| } |
| #endif |
| |
| // Helper function for bit packing snippet of low precision comparison. |
| // It packs the flags from 32x16 to 16x16. |
| EIGEN_STRONG_INLINE __m256i Pack32To16(Packet16f rf) { |
| // Split data into small pieces and handle with AVX instructions |
| // to guarantee internal order of vector. |
| // Operation: |
| // dst[15:0] := Saturate16(rf[31:0]) |
| // dst[31:16] := Saturate16(rf[63:32]) |
| // ... |
| // dst[255:240] := Saturate16(rf[255:224]) |
| __m256i lo = _mm256_castps_si256(extract256<0>(rf)); |
| __m256i hi = _mm256_castps_si256(extract256<1>(rf)); |
| __m128i result_lo = _mm_packs_epi32(_mm256_extractf128_si256(lo, 0), _mm256_extractf128_si256(lo, 1)); |
| __m128i result_hi = _mm_packs_epi32(_mm256_extractf128_si256(hi, 0), _mm256_extractf128_si256(hi, 1)); |
| return _mm256_insertf128_si256(_mm256_castsi128_si256(result_lo), result_hi, 1); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pisnan(const Packet16f& a) { |
| __mmask16 mask = _mm512_cmp_ps_mask(a, a, _CMP_UNORD_Q); |
| return _mm512_castsi512_ps(_mm512_maskz_set1_epi32(mask, 0xffffffffu)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pcmp_eq(const Packet16f& a, const Packet16f& b) { |
| __mmask16 mask = _mm512_cmp_ps_mask(a, b, _CMP_EQ_OQ); |
| return _mm512_castsi512_ps(_mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pcmp_le(const Packet16f& a, const Packet16f& b) { |
| __mmask16 mask = _mm512_cmp_ps_mask(a, b, _CMP_LE_OQ); |
| return _mm512_castsi512_ps(_mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pcmp_lt(const Packet16f& a, const Packet16f& b) { |
| __mmask16 mask = _mm512_cmp_ps_mask(a, b, _CMP_LT_OQ); |
| return _mm512_castsi512_ps(_mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pcmp_lt_or_nan(const Packet16f& a, const Packet16f& b) { |
| __mmask16 mask = _mm512_cmp_ps_mask(a, b, _CMP_NGE_UQ); |
| return _mm512_castsi512_ps(_mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pcmp_eq(const Packet16i& a, const Packet16i& b) { |
| __mmask16 mask = _mm512_cmp_epi32_mask(a, b, _MM_CMPINT_EQ); |
| return _mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pcmp_le(const Packet16i& a, const Packet16i& b) { |
| __mmask16 mask = _mm512_cmp_epi32_mask(a, b, _MM_CMPINT_LE); |
| return _mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pcmp_lt(const Packet16i& a, const Packet16i& b) { |
| __mmask16 mask = _mm512_cmp_epi32_mask(a, b, _MM_CMPINT_LT); |
| return _mm512_mask_set1_epi32(_mm512_setzero_epi32(), mask, 0xffffffffu); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pcmp_eq(const Packet8d& a, const Packet8d& b) { |
| __mmask8 mask = _mm512_cmp_pd_mask(a, b, _CMP_EQ_OQ); |
| return _mm512_castsi512_pd(_mm512_mask_set1_epi64(_mm512_setzero_epi32(), mask, 0xffffffffffffffffu)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pcmp_le(const Packet8d& a, const Packet8d& b) { |
| __mmask8 mask = _mm512_cmp_pd_mask(a, b, _CMP_LE_OQ); |
| return _mm512_castsi512_pd(_mm512_mask_set1_epi64(_mm512_setzero_epi32(), mask, 0xffffffffffffffffu)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pcmp_lt(const Packet8d& a, const Packet8d& b) { |
| __mmask8 mask = _mm512_cmp_pd_mask(a, b, _CMP_LT_OQ); |
| return _mm512_castsi512_pd(_mm512_mask_set1_epi64(_mm512_setzero_epi32(), mask, 0xffffffffffffffffu)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pcmp_lt_or_nan(const Packet8d& a, const Packet8d& b) { |
| __mmask8 mask = _mm512_cmp_pd_mask(a, b, _CMP_NGE_UQ); |
| return _mm512_castsi512_pd(_mm512_mask_set1_epi64(_mm512_setzero_epi32(), mask, 0xffffffffffffffffu)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f print<Packet16f>(const Packet16f& a) { |
| return _mm512_roundscale_ps(a, _MM_FROUND_CUR_DIRECTION); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d print<Packet8d>(const Packet8d& a) { |
| return _mm512_roundscale_pd(a, _MM_FROUND_CUR_DIRECTION); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pceil<Packet16f>(const Packet16f& a) { |
| return _mm512_roundscale_ps(a, _MM_FROUND_TO_POS_INF); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pceil<Packet8d>(const Packet8d& a) { |
| return _mm512_roundscale_pd(a, _MM_FROUND_TO_POS_INF); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pfloor<Packet16f>(const Packet16f& a) { |
| return _mm512_roundscale_ps(a, _MM_FROUND_TO_NEG_INF); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pfloor<Packet8d>(const Packet8d& a) { |
| return _mm512_roundscale_pd(a, _MM_FROUND_TO_NEG_INF); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i ptrue<Packet16i>(const Packet16i& /*a*/) { |
| return _mm512_set1_epi32(0xffffffffu); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f ptrue<Packet16f>(const Packet16f& a) { |
| return _mm512_castsi512_ps(ptrue<Packet16i>(_mm512_castps_si512(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ptrue<Packet8d>(const Packet8d& a) { |
| return _mm512_castsi512_pd(ptrue<Packet16i>(_mm512_castpd_si512(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pand<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_and_si512(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pand<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_and_ps(a, b); |
| #else |
| return _mm512_castsi512_ps(pand(_mm512_castps_si512(a), _mm512_castps_si512(b))); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pand<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_and_pd(a, b); |
| #else |
| Packet8d res = _mm512_undefined_pd(); |
| Packet4d lane0_a = _mm512_extractf64x4_pd(a, 0); |
| Packet4d lane0_b = _mm512_extractf64x4_pd(b, 0); |
| res = _mm512_insertf64x4(res, _mm256_and_pd(lane0_a, lane0_b), 0); |
| |
| Packet4d lane1_a = _mm512_extractf64x4_pd(a, 1); |
| Packet4d lane1_b = _mm512_extractf64x4_pd(b, 1); |
| return _mm512_insertf64x4(res, _mm256_and_pd(lane1_a, lane1_b), 1); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i por<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_or_si512(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f por<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_or_ps(a, b); |
| #else |
| return _mm512_castsi512_ps(por(_mm512_castps_si512(a), _mm512_castps_si512(b))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d por<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_or_pd(a, b); |
| #else |
| return _mm512_castsi512_pd(por(_mm512_castpd_si512(a), _mm512_castpd_si512(b))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pxor<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_xor_si512(a, b); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pxor<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_xor_ps(a, b); |
| #else |
| return _mm512_castsi512_ps(pxor(_mm512_castps_si512(a), _mm512_castps_si512(b))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pxor<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_xor_pd(a, b); |
| #else |
| return _mm512_castsi512_pd(pxor(_mm512_castpd_si512(a), _mm512_castpd_si512(b))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pandnot<Packet16i>(const Packet16i& a, const Packet16i& b) { |
| return _mm512_andnot_si512(b, a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pandnot<Packet16f>(const Packet16f& a, const Packet16f& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_andnot_ps(b, a); |
| #else |
| return _mm512_castsi512_ps(pandnot(_mm512_castps_si512(a), _mm512_castps_si512(b))); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pandnot<Packet8d>(const Packet8d& a, const Packet8d& b) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_andnot_pd(b, a); |
| #else |
| return _mm512_castsi512_pd(pandnot(_mm512_castpd_si512(a), _mm512_castpd_si512(b))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pround<Packet16f>(const Packet16f& a) { |
| // Work-around for default std::round rounding mode. |
| const Packet16f mask = pset1frombits<Packet16f>(static_cast<numext::uint32_t>(0x80000000u)); |
| const Packet16f prev0dot5 = pset1frombits<Packet16f>(static_cast<numext::uint32_t>(0x3EFFFFFFu)); |
| return _mm512_roundscale_ps(padd(por(pand(a, mask), prev0dot5), a), _MM_FROUND_TO_ZERO); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pround<Packet8d>(const Packet8d& a) { |
| // Work-around for default std::round rounding mode. |
| const Packet8d mask = pset1frombits<Packet8d>(static_cast<numext::uint64_t>(0x8000000000000000ull)); |
| const Packet8d prev0dot5 = pset1frombits<Packet8d>(static_cast<numext::uint64_t>(0x3FDFFFFFFFFFFFFFull)); |
| return _mm512_roundscale_pd(padd(por(pand(a, mask), prev0dot5), a), _MM_FROUND_TO_ZERO); |
| } |
| |
| template <int N> |
| EIGEN_STRONG_INLINE Packet16i parithmetic_shift_right(Packet16i a) { |
| return _mm512_srai_epi32(a, N); |
| } |
| |
| template <int N> |
| EIGEN_STRONG_INLINE Packet16i plogical_shift_right(Packet16i a) { |
| return _mm512_srli_epi32(a, N); |
| } |
| |
| template <int N> |
| EIGEN_STRONG_INLINE Packet16i plogical_shift_left(Packet16i a) { |
| return _mm512_slli_epi32(a, N); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pload<Packet16f>(const float* from) { |
| EIGEN_DEBUG_ALIGNED_LOAD return _mm512_load_ps(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pload<Packet8d>(const double* from) { |
| EIGEN_DEBUG_ALIGNED_LOAD return _mm512_load_pd(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pload<Packet16i>(const int* from) { |
| EIGEN_DEBUG_ALIGNED_LOAD return _mm512_load_si512(reinterpret_cast<const __m512i*>(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f ploadu<Packet16f>(const float* from) { |
| EIGEN_DEBUG_UNALIGNED_LOAD return _mm512_loadu_ps(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ploadu<Packet8d>(const double* from) { |
| EIGEN_DEBUG_UNALIGNED_LOAD return _mm512_loadu_pd(from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i ploadu<Packet16i>(const int* from) { |
| EIGEN_DEBUG_UNALIGNED_LOAD return _mm512_loadu_si512(reinterpret_cast<const __m512i*>(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f ploadu<Packet16f>(const float* from, uint16_t umask) { |
| __mmask16 mask = static_cast<__mmask16>(umask); |
| EIGEN_DEBUG_UNALIGNED_LOAD return _mm512_maskz_loadu_ps(mask, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ploadu<Packet8d>(const double* from, uint8_t umask) { |
| __mmask8 mask = static_cast<__mmask8>(umask); |
| EIGEN_DEBUG_UNALIGNED_LOAD return _mm512_maskz_loadu_pd(mask, from); |
| } |
| |
| // Loads 8 floats from memory a returns the packet |
| // {a0, a0 a1, a1, a2, a2, a3, a3, a4, a4, a5, a5, a6, a6, a7, a7} |
| template <> |
| EIGEN_STRONG_INLINE Packet16f ploaddup<Packet16f>(const float* from) { |
| // an unaligned load is required here as there is no requirement |
| // on the alignment of input pointer 'from' |
| __m256i low_half = _mm256_loadu_si256(reinterpret_cast<const __m256i*>(from)); |
| __m512 even_elements = _mm512_castsi512_ps(_mm512_cvtepu32_epi64(low_half)); |
| __m512 pairs = _mm512_permute_ps(even_elements, _MM_SHUFFLE(2, 2, 0, 0)); |
| return pairs; |
| } |
| |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| // FIXME: this does not look optimal, better load a Packet4d and shuffle... |
| // Loads 4 doubles from memory a returns the packet {a0, a0 a1, a1, a2, a2, a3, |
| // a3} |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ploaddup<Packet8d>(const double* from) { |
| __m512d x = _mm512_setzero_pd(); |
| x = _mm512_insertf64x2(x, _mm_loaddup_pd(&from[0]), 0); |
| x = _mm512_insertf64x2(x, _mm_loaddup_pd(&from[1]), 1); |
| x = _mm512_insertf64x2(x, _mm_loaddup_pd(&from[2]), 2); |
| x = _mm512_insertf64x2(x, _mm_loaddup_pd(&from[3]), 3); |
| return x; |
| } |
| #else |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ploaddup<Packet8d>(const double* from) { |
| __m512d x = _mm512_setzero_pd(); |
| x = _mm512_mask_broadcastsd_pd(x, 0x3 << 0, _mm_load_sd(from + 0)); |
| x = _mm512_mask_broadcastsd_pd(x, 0x3 << 2, _mm_load_sd(from + 1)); |
| x = _mm512_mask_broadcastsd_pd(x, 0x3 << 4, _mm_load_sd(from + 2)); |
| x = _mm512_mask_broadcastsd_pd(x, 0x3 << 6, _mm_load_sd(from + 3)); |
| return x; |
| } |
| #endif |
| |
| // Loads 8 integers from memory and returns the packet |
| // {a0, a0 a1, a1, a2, a2, a3, a3, a4, a4, a5, a5, a6, a6, a7, a7} |
| template <> |
| EIGEN_STRONG_INLINE Packet16i ploaddup<Packet16i>(const int* from) { |
| __m256i low_half = _mm256_loadu_si256(reinterpret_cast<const __m256i*>(from)); |
| __m512 even_elements = _mm512_castsi512_ps(_mm512_cvtepu32_epi64(low_half)); |
| __m512 pairs = _mm512_permute_ps(even_elements, _MM_SHUFFLE(2, 2, 0, 0)); |
| return _mm512_castps_si512(pairs); |
| } |
| |
| // Loads 4 floats from memory a returns the packet |
| // {a0, a0 a0, a0, a1, a1, a1, a1, a2, a2, a2, a2, a3, a3, a3, a3} |
| template <> |
| EIGEN_STRONG_INLINE Packet16f ploadquad<Packet16f>(const float* from) { |
| Packet16f tmp = _mm512_castps128_ps512(ploadu<Packet4f>(from)); |
| const Packet16i scatter_mask = _mm512_set_epi32(3, 3, 3, 3, 2, 2, 2, 2, 1, 1, 1, 1, 0, 0, 0, 0); |
| return _mm512_permutexvar_ps(scatter_mask, tmp); |
| } |
| |
| // Loads 2 doubles from memory a returns the packet |
| // {a0, a0 a0, a0, a1, a1, a1, a1} |
| template <> |
| EIGEN_STRONG_INLINE Packet8d ploadquad<Packet8d>(const double* from) { |
| __m256d lane0 = _mm256_set1_pd(*from); |
| __m256d lane1 = _mm256_set1_pd(*(from + 1)); |
| __m512d tmp = _mm512_undefined_pd(); |
| tmp = _mm512_insertf64x4(tmp, lane0, 0); |
| return _mm512_insertf64x4(tmp, lane1, 1); |
| } |
| |
| // Loads 4 integers from memory and returns the packet |
| // {a0, a0 a0, a0, a1, a1, a1, a1, a2, a2, a2, a2, a3, a3, a3, a3} |
| template <> |
| EIGEN_STRONG_INLINE Packet16i ploadquad<Packet16i>(const int* from) { |
| Packet16i tmp = _mm512_castsi128_si512(ploadu<Packet4i>(from)); |
| const Packet16i scatter_mask = _mm512_set_epi32(3, 3, 3, 3, 2, 2, 2, 2, 1, 1, 1, 1, 0, 0, 0, 0); |
| return _mm512_permutexvar_epi32(scatter_mask, tmp); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pstore<float>(float* to, const Packet16f& from) { |
| EIGEN_DEBUG_ALIGNED_STORE _mm512_store_ps(to, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstore<double>(double* to, const Packet8d& from) { |
| EIGEN_DEBUG_ALIGNED_STORE _mm512_store_pd(to, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstore<int>(int* to, const Packet16i& from) { |
| EIGEN_DEBUG_ALIGNED_STORE _mm512_storeu_si512(reinterpret_cast<__m512i*>(to), from); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet16f& from) { |
| EIGEN_DEBUG_UNALIGNED_STORE _mm512_storeu_ps(to, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet8d& from) { |
| EIGEN_DEBUG_UNALIGNED_STORE _mm512_storeu_pd(to, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet16i& from) { |
| EIGEN_DEBUG_UNALIGNED_STORE _mm512_storeu_si512(reinterpret_cast<__m512i*>(to), from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<float>(float* to, const Packet16f& from, uint16_t umask) { |
| __mmask16 mask = static_cast<__mmask16>(umask); |
| EIGEN_DEBUG_UNALIGNED_STORE return _mm512_mask_storeu_ps(to, mask, from); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<double>(double* to, const Packet8d& from, uint8_t umask) { |
| __mmask8 mask = static_cast<__mmask8>(umask); |
| EIGEN_DEBUG_UNALIGNED_STORE return _mm512_mask_storeu_pd(to, mask, from); |
| } |
| |
| template <typename Scalar, typename Packet> |
| EIGEN_DEVICE_FUNC inline Packet pgather(const Packet& src, const Scalar* from, Index stride, |
| typename unpacket_traits<Packet>::mask_t umask); |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet16f pgather<float, Packet16f>(const Packet16f& src, const float* from, Index stride, |
| uint16_t umask) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| __mmask16 mask = static_cast<__mmask16>(umask); |
| |
| return _mm512_mask_i32gather_ps(src, mask, indices, from, 4); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet8d pgather<double, Packet8d>(const Packet8d& src, const double* from, Index stride, |
| uint8_t umask) { |
| Packet8i stride_vector = _mm256_set1_epi32(convert_index<int>(stride)); |
| Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); |
| Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); |
| __mmask8 mask = static_cast<__mmask8>(umask); |
| |
| return _mm512_mask_i32gather_pd(src, mask, indices, from, 8); |
| } |
| |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet16f pgather<float, Packet16f>(const float* from, Index stride) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| |
| return _mm512_i32gather_ps(indices, from, 4); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet8d pgather<double, Packet8d>(const double* from, Index stride) { |
| Packet8i stride_vector = _mm256_set1_epi32(convert_index<int>(stride)); |
| Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); |
| Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); |
| |
| return _mm512_i32gather_pd(indices, from, 8); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline Packet16i pgather<int, Packet16i>(const int* from, Index stride) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| return _mm512_i32gather_epi32(indices, from, 4); |
| } |
| |
| template <typename Scalar, typename Packet> |
| EIGEN_DEVICE_FUNC inline void pscatter(Scalar* to, const Packet& from, Index stride, |
| typename unpacket_traits<Packet>::mask_t umask); |
| template <> |
| EIGEN_DEVICE_FUNC inline void pscatter<float, Packet16f>(float* to, const Packet16f& from, Index stride, |
| uint16_t umask) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| __mmask16 mask = static_cast<__mmask16>(umask); |
| _mm512_mask_i32scatter_ps(to, mask, indices, from, 4); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline void pscatter<double, Packet8d>(double* to, const Packet8d& from, Index stride, |
| uint8_t umask) { |
| Packet8i stride_vector = _mm256_set1_epi32(convert_index<int>(stride)); |
| Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); |
| Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); |
| __mmask8 mask = static_cast<__mmask8>(umask); |
| _mm512_mask_i32scatter_pd(to, mask, indices, from, 8); |
| } |
| |
| template <> |
| EIGEN_DEVICE_FUNC inline void pscatter<float, Packet16f>(float* to, const Packet16f& from, Index stride) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| _mm512_i32scatter_ps(to, indices, from, 4); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline void pscatter<double, Packet8d>(double* to, const Packet8d& from, Index stride) { |
| Packet8i stride_vector = _mm256_set1_epi32(convert_index<int>(stride)); |
| Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); |
| Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); |
| _mm512_i32scatter_pd(to, indices, from, 8); |
| } |
| template <> |
| EIGEN_DEVICE_FUNC inline void pscatter<int, Packet16i>(int* to, const Packet16i& from, Index stride) { |
| Packet16i stride_vector = _mm512_set1_epi32(convert_index<int>(stride)); |
| Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); |
| Packet16i indices = _mm512_mullo_epi32(stride_vector, stride_multiplier); |
| _mm512_i32scatter_epi32(to, indices, from, 4); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pstore1<Packet16f>(float* to, const float& a) { |
| Packet16f pa = pset1<Packet16f>(a); |
| pstore(to, pa); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstore1<Packet8d>(double* to, const double& a) { |
| Packet8d pa = pset1<Packet8d>(a); |
| pstore(to, pa); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void pstore1<Packet16i>(int* to, const int& a) { |
| Packet16i pa = pset1<Packet16i>(a); |
| pstore(to, pa); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void prefetch<float>(const float* addr) { |
| _mm_prefetch((SsePrefetchPtrType)(addr), _MM_HINT_T0); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void prefetch<double>(const double* addr) { |
| _mm_prefetch((SsePrefetchPtrType)(addr), _MM_HINT_T0); |
| } |
| template <> |
| EIGEN_STRONG_INLINE void prefetch<int>(const int* addr) { |
| _mm_prefetch((SsePrefetchPtrType)(addr), _MM_HINT_T0); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE float pfirst<Packet16f>(const Packet16f& a) { |
| return _mm_cvtss_f32(_mm512_extractf32x4_ps(a, 0)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE double pfirst<Packet8d>(const Packet8d& a) { |
| return _mm_cvtsd_f64(_mm256_extractf128_pd(_mm512_extractf64x4_pd(a, 0), 0)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE int pfirst<Packet16i>(const Packet16i& a) { |
| return _mm_extract_epi32(_mm512_extracti32x4_epi32(a, 0), 0); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f preverse(const Packet16f& a) { |
| return _mm512_permutexvar_ps(_mm512_set_epi32(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15), a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d preverse(const Packet8d& a) { |
| return _mm512_permutexvar_pd(_mm512_set_epi32(0, 0, 0, 1, 0, 2, 0, 3, 0, 4, 0, 5, 0, 6, 0, 7), a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16i preverse(const Packet16i& a) { |
| return _mm512_permutexvar_epi32(_mm512_set_epi32(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15), a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pabs(const Packet16f& a) { |
| // _mm512_abs_ps intrinsic not found, so hack around it |
| return _mm512_castsi512_ps(_mm512_and_si512(_mm512_castps_si512(a), _mm512_set1_epi32(0x7fffffff))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pabs(const Packet8d& a) { |
| // _mm512_abs_ps intrinsic not found, so hack around it |
| return _mm512_castsi512_pd(_mm512_and_si512(_mm512_castpd_si512(a), _mm512_set1_epi64(0x7fffffffffffffff))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16i pabs(const Packet16i& a) { |
| return _mm512_abs_epi32(a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h psignbit(const Packet16h& a) { |
| return _mm256_srai_epi16(a, 15); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16bf psignbit(const Packet16bf& a) { |
| return _mm256_srai_epi16(a, 15); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16f psignbit(const Packet16f& a) { |
| return _mm512_castsi512_ps(_mm512_srai_epi32(_mm512_castps_si512(a), 31)); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d psignbit(const Packet8d& a) { |
| return _mm512_castsi512_pd(_mm512_srai_epi64(_mm512_castpd_si512(a), 63)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pfrexp<Packet16f>(const Packet16f& a, Packet16f& exponent) { |
| return pfrexp_generic(a, exponent); |
| } |
| |
| // Extract exponent without existence of Packet8l. |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pfrexp_generic_get_biased_exponent(const Packet8d& a) { |
| const Packet8d cst_exp_mask = pset1frombits<Packet8d>(static_cast<uint64_t>(0x7ff0000000000000ull)); |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| return _mm512_cvtepi64_pd(_mm512_srli_epi64(_mm512_castpd_si512(pand(a, cst_exp_mask)), 52)); |
| #else |
| return _mm512_cvtepi32_pd(_mm512_cvtepi64_epi32(_mm512_srli_epi64(_mm512_castpd_si512(pand(a, cst_exp_mask)), 52))); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pfrexp<Packet8d>(const Packet8d& a, Packet8d& exponent) { |
| return pfrexp_generic(a, exponent); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pldexp<Packet16f>(const Packet16f& a, const Packet16f& exponent) { |
| return pldexp_generic(a, exponent); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pldexp<Packet8d>(const Packet8d& a, const Packet8d& exponent) { |
| // Clamp exponent to [-2099, 2099] |
| const Packet8d max_exponent = pset1<Packet8d>(2099.0); |
| const Packet8i e = _mm512_cvtpd_epi32(pmin(pmax(exponent, pnegate(max_exponent)), max_exponent)); |
| |
| // Split 2^e into four factors and multiply. |
| const Packet8i bias = pset1<Packet8i>(1023); |
| Packet8i b = parithmetic_shift_right<2>(e); // floor(e/4) |
| |
| // 2^b |
| const Packet8i permute_idx = _mm256_setr_epi32(0, 4, 1, 5, 2, 6, 3, 7); |
| Packet8i hi = _mm256_permutevar8x32_epi32(padd(b, bias), permute_idx); |
| Packet8i lo = _mm256_slli_epi64(hi, 52); |
| hi = _mm256_slli_epi64(_mm256_srli_epi64(hi, 32), 52); |
| Packet8d c = _mm512_castsi512_pd(_mm512_inserti64x4(_mm512_castsi256_si512(lo), hi, 1)); |
| Packet8d out = pmul(pmul(pmul(a, c), c), c); // a * 2^(3b) |
| |
| // 2^(e - 3b) |
| b = psub(psub(psub(e, b), b), b); // e - 3b |
| hi = _mm256_permutevar8x32_epi32(padd(b, bias), permute_idx); |
| lo = _mm256_slli_epi64(hi, 52); |
| hi = _mm256_slli_epi64(_mm256_srli_epi64(hi, 32), 52); |
| c = _mm512_castsi512_pd(_mm512_inserti64x4(_mm512_castsi256_si512(lo), hi, 1)); |
| out = pmul(out, c); // a * 2^e |
| return out; |
| } |
| |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| // AVX512F does not define _mm512_extractf32x8_ps to extract _m256 from _m512 |
| #define EIGEN_EXTRACT_8f_FROM_16f(INPUT, OUTPUT) \ |
| __m256 OUTPUT##_0 = _mm512_extractf32x8_ps(INPUT, 0); \ |
| __m256 OUTPUT##_1 = _mm512_extractf32x8_ps(INPUT, 1) |
| |
| // AVX512F does not define _mm512_extracti32x8_epi32 to extract _m256i from _m512i |
| #define EIGEN_EXTRACT_8i_FROM_16i(INPUT, OUTPUT) \ |
| __m256i OUTPUT##_0 = _mm512_extracti32x8_epi32(INPUT, 0); \ |
| __m256i OUTPUT##_1 = _mm512_extracti32x8_epi32(INPUT, 1) |
| #else |
| #define EIGEN_EXTRACT_8f_FROM_16f(INPUT, OUTPUT) \ |
| __m256 OUTPUT##_0 = _mm256_insertf128_ps(_mm256_castps128_ps256(_mm512_extractf32x4_ps(INPUT, 0)), \ |
| _mm512_extractf32x4_ps(INPUT, 1), 1); \ |
| __m256 OUTPUT##_1 = _mm256_insertf128_ps(_mm256_castps128_ps256(_mm512_extractf32x4_ps(INPUT, 2)), \ |
| _mm512_extractf32x4_ps(INPUT, 3), 1) |
| |
| #define EIGEN_EXTRACT_8i_FROM_16i(INPUT, OUTPUT) \ |
| __m256i OUTPUT##_0 = _mm256_insertf128_si256(_mm256_castsi128_si256(_mm512_extracti32x4_epi32(INPUT, 0)), \ |
| _mm512_extracti32x4_epi32(INPUT, 1), 1); \ |
| __m256i OUTPUT##_1 = _mm256_insertf128_si256(_mm256_castsi128_si256(_mm512_extracti32x4_epi32(INPUT, 2)), \ |
| _mm512_extracti32x4_epi32(INPUT, 3), 1) |
| #endif |
| |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| #define EIGEN_INSERT_8f_INTO_16f(OUTPUT, INPUTA, INPUTB) \ |
| OUTPUT = _mm512_insertf32x8(_mm512_castps256_ps512(INPUTA), INPUTB, 1); |
| |
| #define EIGEN_INSERT_8i_INTO_16i(OUTPUT, INPUTA, INPUTB) \ |
| OUTPUT = _mm512_inserti32x8(_mm512_castsi256_si512(INPUTA), INPUTB, 1); |
| #else |
| #define EIGEN_INSERT_8f_INTO_16f(OUTPUT, INPUTA, INPUTB) \ |
| OUTPUT = _mm512_undefined_ps(); \ |
| OUTPUT = _mm512_insertf32x4(OUTPUT, _mm256_extractf128_ps(INPUTA, 0), 0); \ |
| OUTPUT = _mm512_insertf32x4(OUTPUT, _mm256_extractf128_ps(INPUTA, 1), 1); \ |
| OUTPUT = _mm512_insertf32x4(OUTPUT, _mm256_extractf128_ps(INPUTB, 0), 2); \ |
| OUTPUT = _mm512_insertf32x4(OUTPUT, _mm256_extractf128_ps(INPUTB, 1), 3); |
| |
| #define EIGEN_INSERT_8i_INTO_16i(OUTPUT, INPUTA, INPUTB) \ |
| OUTPUT = _mm512_undefined_epi32(); \ |
| OUTPUT = _mm512_inserti32x4(OUTPUT, _mm256_extractf128_si256(INPUTA, 0), 0); \ |
| OUTPUT = _mm512_inserti32x4(OUTPUT, _mm256_extractf128_si256(INPUTA, 1), 1); \ |
| OUTPUT = _mm512_inserti32x4(OUTPUT, _mm256_extractf128_si256(INPUTB, 0), 2); \ |
| OUTPUT = _mm512_inserti32x4(OUTPUT, _mm256_extractf128_si256(INPUTB, 1), 3); |
| #endif |
| |
| template <> |
| EIGEN_STRONG_INLINE float predux<Packet16f>(const Packet16f& a) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| __m256 lane0 = _mm512_extractf32x8_ps(a, 0); |
| __m256 lane1 = _mm512_extractf32x8_ps(a, 1); |
| Packet8f x = _mm256_add_ps(lane0, lane1); |
| return predux<Packet8f>(x); |
| #else |
| __m128 lane0 = _mm512_extractf32x4_ps(a, 0); |
| __m128 lane1 = _mm512_extractf32x4_ps(a, 1); |
| __m128 lane2 = _mm512_extractf32x4_ps(a, 2); |
| __m128 lane3 = _mm512_extractf32x4_ps(a, 3); |
| __m128 sum = _mm_add_ps(_mm_add_ps(lane0, lane1), _mm_add_ps(lane2, lane3)); |
| sum = _mm_hadd_ps(sum, sum); |
| sum = _mm_hadd_ps(sum, _mm_permute_ps(sum, 1)); |
| return _mm_cvtss_f32(sum); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE double predux<Packet8d>(const Packet8d& a) { |
| __m256d lane0 = _mm512_extractf64x4_pd(a, 0); |
| __m256d lane1 = _mm512_extractf64x4_pd(a, 1); |
| __m256d sum = _mm256_add_pd(lane0, lane1); |
| __m256d tmp0 = _mm256_hadd_pd(sum, _mm256_permute2f128_pd(sum, sum, 1)); |
| return _mm_cvtsd_f64(_mm256_castpd256_pd128(_mm256_hadd_pd(tmp0, tmp0))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE int predux<Packet16i>(const Packet16i& a) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| __m256i lane0 = _mm512_extracti32x8_epi32(a, 0); |
| __m256i lane1 = _mm512_extracti32x8_epi32(a, 1); |
| Packet8i x = _mm256_add_epi32(lane0, lane1); |
| return predux<Packet8i>(x); |
| #else |
| __m128i lane0 = _mm512_extracti32x4_epi32(a, 0); |
| __m128i lane1 = _mm512_extracti32x4_epi32(a, 1); |
| __m128i lane2 = _mm512_extracti32x4_epi32(a, 2); |
| __m128i lane3 = _mm512_extracti32x4_epi32(a, 3); |
| __m128i sum = _mm_add_epi32(_mm_add_epi32(lane0, lane1), _mm_add_epi32(lane2, lane3)); |
| sum = _mm_hadd_epi32(sum, sum); |
| sum = _mm_hadd_epi32(sum, _mm_castps_si128(_mm_permute_ps(_mm_castsi128_ps(sum), 1))); |
| return _mm_cvtsi128_si32(sum); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8f predux_half_dowto4<Packet16f>(const Packet16f& a) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| __m256 lane0 = _mm512_extractf32x8_ps(a, 0); |
| __m256 lane1 = _mm512_extractf32x8_ps(a, 1); |
| return _mm256_add_ps(lane0, lane1); |
| #else |
| __m128 lane0 = _mm512_extractf32x4_ps(a, 0); |
| __m128 lane1 = _mm512_extractf32x4_ps(a, 1); |
| __m128 lane2 = _mm512_extractf32x4_ps(a, 2); |
| __m128 lane3 = _mm512_extractf32x4_ps(a, 3); |
| __m128 sum0 = _mm_add_ps(lane0, lane2); |
| __m128 sum1 = _mm_add_ps(lane1, lane3); |
| return _mm256_insertf128_ps(_mm256_castps128_ps256(sum0), sum1, 1); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet4d predux_half_dowto4<Packet8d>(const Packet8d& a) { |
| __m256d lane0 = _mm512_extractf64x4_pd(a, 0); |
| __m256d lane1 = _mm512_extractf64x4_pd(a, 1); |
| return _mm256_add_pd(lane0, lane1); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8i predux_half_dowto4<Packet16i>(const Packet16i& a) { |
| #ifdef EIGEN_VECTORIZE_AVX512DQ |
| __m256i lane0 = _mm512_extracti32x8_epi32(a, 0); |
| __m256i lane1 = _mm512_extracti32x8_epi32(a, 1); |
| return _mm256_add_epi32(lane0, lane1); |
| #else |
| __m128i lane0 = _mm512_extracti32x4_epi32(a, 0); |
| __m128i lane1 = _mm512_extracti32x4_epi32(a, 1); |
| __m128i lane2 = _mm512_extracti32x4_epi32(a, 2); |
| __m128i lane3 = _mm512_extracti32x4_epi32(a, 3); |
| __m128i sum0 = _mm_add_epi32(lane0, lane2); |
| __m128i sum1 = _mm_add_epi32(lane1, lane3); |
| return _mm256_inserti128_si256(_mm256_castsi128_si256(sum0), sum1, 1); |
| #endif |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE float predux_mul<Packet16f>(const Packet16f& a) { |
| // #ifdef EIGEN_VECTORIZE_AVX512DQ |
| #if 0 |
| Packet8f lane0 = _mm512_extractf32x8_ps(a, 0); |
| Packet8f lane1 = _mm512_extractf32x8_ps(a, 1); |
| Packet8f res = pmul(lane0, lane1); |
| res = pmul(res, _mm256_permute2f128_ps(res, res, 1)); |
| res = pmul(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 3, 2))); |
| return pfirst(pmul(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 0, 1)))); |
| #else |
| __m128 lane0 = _mm512_extractf32x4_ps(a, 0); |
| __m128 lane1 = _mm512_extractf32x4_ps(a, 1); |
| __m128 lane2 = _mm512_extractf32x4_ps(a, 2); |
| __m128 lane3 = _mm512_extractf32x4_ps(a, 3); |
| __m128 res = pmul(pmul(lane0, lane1), pmul(lane2, lane3)); |
| res = pmul(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 3, 2))); |
| return pfirst(pmul(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 0, 1)))); |
| #endif |
| } |
| template <> |
| EIGEN_STRONG_INLINE double predux_mul<Packet8d>(const Packet8d& a) { |
| __m256d lane0 = _mm512_extractf64x4_pd(a, 0); |
| __m256d lane1 = _mm512_extractf64x4_pd(a, 1); |
| __m256d res = pmul(lane0, lane1); |
| res = pmul(res, _mm256_permute2f128_pd(res, res, 1)); |
| return pfirst(pmul(res, _mm256_shuffle_pd(res, res, 1))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE float predux_min<Packet16f>(const Packet16f& a) { |
| __m128 lane0 = _mm512_extractf32x4_ps(a, 0); |
| __m128 lane1 = _mm512_extractf32x4_ps(a, 1); |
| __m128 lane2 = _mm512_extractf32x4_ps(a, 2); |
| __m128 lane3 = _mm512_extractf32x4_ps(a, 3); |
| __m128 res = _mm_min_ps(_mm_min_ps(lane0, lane1), _mm_min_ps(lane2, lane3)); |
| res = _mm_min_ps(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 3, 2))); |
| return pfirst(_mm_min_ps(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 0, 1)))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE double predux_min<Packet8d>(const Packet8d& a) { |
| __m256d lane0 = _mm512_extractf64x4_pd(a, 0); |
| __m256d lane1 = _mm512_extractf64x4_pd(a, 1); |
| __m256d res = _mm256_min_pd(lane0, lane1); |
| res = _mm256_min_pd(res, _mm256_permute2f128_pd(res, res, 1)); |
| return pfirst(_mm256_min_pd(res, _mm256_shuffle_pd(res, res, 1))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE float predux_max<Packet16f>(const Packet16f& a) { |
| __m128 lane0 = _mm512_extractf32x4_ps(a, 0); |
| __m128 lane1 = _mm512_extractf32x4_ps(a, 1); |
| __m128 lane2 = _mm512_extractf32x4_ps(a, 2); |
| __m128 lane3 = _mm512_extractf32x4_ps(a, 3); |
| __m128 res = _mm_max_ps(_mm_max_ps(lane0, lane1), _mm_max_ps(lane2, lane3)); |
| res = _mm_max_ps(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 3, 2))); |
| return pfirst(_mm_max_ps(res, _mm_permute_ps(res, _MM_SHUFFLE(0, 0, 0, 1)))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE double predux_max<Packet8d>(const Packet8d& a) { |
| __m256d lane0 = _mm512_extractf64x4_pd(a, 0); |
| __m256d lane1 = _mm512_extractf64x4_pd(a, 1); |
| __m256d res = _mm256_max_pd(lane0, lane1); |
| res = _mm256_max_pd(res, _mm256_permute2f128_pd(res, res, 1)); |
| return pfirst(_mm256_max_pd(res, _mm256_shuffle_pd(res, res, 1))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE bool predux_any(const Packet16f& x) { |
| Packet16i xi = _mm512_castps_si512(x); |
| __mmask16 tmp = _mm512_test_epi32_mask(xi, xi); |
| return !_mm512_kortestz(tmp, tmp); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE bool predux_any(const Packet16i& x) { |
| __mmask16 tmp = _mm512_test_epi32_mask(x, x); |
| return !_mm512_kortestz(tmp, tmp); |
| } |
| |
| #define PACK_OUTPUT(OUTPUT, INPUT, INDEX, STRIDE) \ |
| EIGEN_INSERT_8f_INTO_16f(OUTPUT[INDEX], INPUT[INDEX], INPUT[INDEX + STRIDE]); |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet16f, 16>& kernel) { |
| __m512 T0 = _mm512_unpacklo_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T1 = _mm512_unpackhi_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T2 = _mm512_unpacklo_ps(kernel.packet[2], kernel.packet[3]); |
| __m512 T3 = _mm512_unpackhi_ps(kernel.packet[2], kernel.packet[3]); |
| __m512 T4 = _mm512_unpacklo_ps(kernel.packet[4], kernel.packet[5]); |
| __m512 T5 = _mm512_unpackhi_ps(kernel.packet[4], kernel.packet[5]); |
| __m512 T6 = _mm512_unpacklo_ps(kernel.packet[6], kernel.packet[7]); |
| __m512 T7 = _mm512_unpackhi_ps(kernel.packet[6], kernel.packet[7]); |
| __m512 T8 = _mm512_unpacklo_ps(kernel.packet[8], kernel.packet[9]); |
| __m512 T9 = _mm512_unpackhi_ps(kernel.packet[8], kernel.packet[9]); |
| __m512 T10 = _mm512_unpacklo_ps(kernel.packet[10], kernel.packet[11]); |
| __m512 T11 = _mm512_unpackhi_ps(kernel.packet[10], kernel.packet[11]); |
| __m512 T12 = _mm512_unpacklo_ps(kernel.packet[12], kernel.packet[13]); |
| __m512 T13 = _mm512_unpackhi_ps(kernel.packet[12], kernel.packet[13]); |
| __m512 T14 = _mm512_unpacklo_ps(kernel.packet[14], kernel.packet[15]); |
| __m512 T15 = _mm512_unpackhi_ps(kernel.packet[14], kernel.packet[15]); |
| __m512 S0 = _mm512_shuffle_ps(T0, T2, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S1 = _mm512_shuffle_ps(T0, T2, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S2 = _mm512_shuffle_ps(T1, T3, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S3 = _mm512_shuffle_ps(T1, T3, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S4 = _mm512_shuffle_ps(T4, T6, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S5 = _mm512_shuffle_ps(T4, T6, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S6 = _mm512_shuffle_ps(T5, T7, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S7 = _mm512_shuffle_ps(T5, T7, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S8 = _mm512_shuffle_ps(T8, T10, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S9 = _mm512_shuffle_ps(T8, T10, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S10 = _mm512_shuffle_ps(T9, T11, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S11 = _mm512_shuffle_ps(T9, T11, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S12 = _mm512_shuffle_ps(T12, T14, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S13 = _mm512_shuffle_ps(T12, T14, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S14 = _mm512_shuffle_ps(T13, T15, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S15 = _mm512_shuffle_ps(T13, T15, _MM_SHUFFLE(3, 2, 3, 2)); |
| |
| EIGEN_EXTRACT_8f_FROM_16f(S0, S0); |
| EIGEN_EXTRACT_8f_FROM_16f(S1, S1); |
| EIGEN_EXTRACT_8f_FROM_16f(S2, S2); |
| EIGEN_EXTRACT_8f_FROM_16f(S3, S3); |
| EIGEN_EXTRACT_8f_FROM_16f(S4, S4); |
| EIGEN_EXTRACT_8f_FROM_16f(S5, S5); |
| EIGEN_EXTRACT_8f_FROM_16f(S6, S6); |
| EIGEN_EXTRACT_8f_FROM_16f(S7, S7); |
| EIGEN_EXTRACT_8f_FROM_16f(S8, S8); |
| EIGEN_EXTRACT_8f_FROM_16f(S9, S9); |
| EIGEN_EXTRACT_8f_FROM_16f(S10, S10); |
| EIGEN_EXTRACT_8f_FROM_16f(S11, S11); |
| EIGEN_EXTRACT_8f_FROM_16f(S12, S12); |
| EIGEN_EXTRACT_8f_FROM_16f(S13, S13); |
| EIGEN_EXTRACT_8f_FROM_16f(S14, S14); |
| EIGEN_EXTRACT_8f_FROM_16f(S15, S15); |
| |
| PacketBlock<Packet8f, 32> tmp; |
| |
| tmp.packet[0] = _mm256_permute2f128_ps(S0_0, S4_0, 0x20); |
| tmp.packet[1] = _mm256_permute2f128_ps(S1_0, S5_0, 0x20); |
| tmp.packet[2] = _mm256_permute2f128_ps(S2_0, S6_0, 0x20); |
| tmp.packet[3] = _mm256_permute2f128_ps(S3_0, S7_0, 0x20); |
| tmp.packet[4] = _mm256_permute2f128_ps(S0_0, S4_0, 0x31); |
| tmp.packet[5] = _mm256_permute2f128_ps(S1_0, S5_0, 0x31); |
| tmp.packet[6] = _mm256_permute2f128_ps(S2_0, S6_0, 0x31); |
| tmp.packet[7] = _mm256_permute2f128_ps(S3_0, S7_0, 0x31); |
| |
| tmp.packet[8] = _mm256_permute2f128_ps(S0_1, S4_1, 0x20); |
| tmp.packet[9] = _mm256_permute2f128_ps(S1_1, S5_1, 0x20); |
| tmp.packet[10] = _mm256_permute2f128_ps(S2_1, S6_1, 0x20); |
| tmp.packet[11] = _mm256_permute2f128_ps(S3_1, S7_1, 0x20); |
| tmp.packet[12] = _mm256_permute2f128_ps(S0_1, S4_1, 0x31); |
| tmp.packet[13] = _mm256_permute2f128_ps(S1_1, S5_1, 0x31); |
| tmp.packet[14] = _mm256_permute2f128_ps(S2_1, S6_1, 0x31); |
| tmp.packet[15] = _mm256_permute2f128_ps(S3_1, S7_1, 0x31); |
| |
| // Second set of _m256 outputs |
| tmp.packet[16] = _mm256_permute2f128_ps(S8_0, S12_0, 0x20); |
| tmp.packet[17] = _mm256_permute2f128_ps(S9_0, S13_0, 0x20); |
| tmp.packet[18] = _mm256_permute2f128_ps(S10_0, S14_0, 0x20); |
| tmp.packet[19] = _mm256_permute2f128_ps(S11_0, S15_0, 0x20); |
| tmp.packet[20] = _mm256_permute2f128_ps(S8_0, S12_0, 0x31); |
| tmp.packet[21] = _mm256_permute2f128_ps(S9_0, S13_0, 0x31); |
| tmp.packet[22] = _mm256_permute2f128_ps(S10_0, S14_0, 0x31); |
| tmp.packet[23] = _mm256_permute2f128_ps(S11_0, S15_0, 0x31); |
| |
| tmp.packet[24] = _mm256_permute2f128_ps(S8_1, S12_1, 0x20); |
| tmp.packet[25] = _mm256_permute2f128_ps(S9_1, S13_1, 0x20); |
| tmp.packet[26] = _mm256_permute2f128_ps(S10_1, S14_1, 0x20); |
| tmp.packet[27] = _mm256_permute2f128_ps(S11_1, S15_1, 0x20); |
| tmp.packet[28] = _mm256_permute2f128_ps(S8_1, S12_1, 0x31); |
| tmp.packet[29] = _mm256_permute2f128_ps(S9_1, S13_1, 0x31); |
| tmp.packet[30] = _mm256_permute2f128_ps(S10_1, S14_1, 0x31); |
| tmp.packet[31] = _mm256_permute2f128_ps(S11_1, S15_1, 0x31); |
| |
| // Pack them into the output |
| PACK_OUTPUT(kernel.packet, tmp.packet, 0, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 1, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 2, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 3, 16); |
| |
| PACK_OUTPUT(kernel.packet, tmp.packet, 4, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 5, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 6, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 7, 16); |
| |
| PACK_OUTPUT(kernel.packet, tmp.packet, 8, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 9, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 10, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 11, 16); |
| |
| PACK_OUTPUT(kernel.packet, tmp.packet, 12, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 13, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 14, 16); |
| PACK_OUTPUT(kernel.packet, tmp.packet, 15, 16); |
| } |
| #define PACK_OUTPUT_2(OUTPUT, INPUT, INDEX, STRIDE) \ |
| EIGEN_INSERT_8f_INTO_16f(OUTPUT[INDEX], INPUT[2 * INDEX], INPUT[2 * INDEX + STRIDE]); |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet16f, 8>& kernel) { |
| __m512 T0 = _mm512_unpacklo_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T1 = _mm512_unpackhi_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T2 = _mm512_unpacklo_ps(kernel.packet[2], kernel.packet[3]); |
| __m512 T3 = _mm512_unpackhi_ps(kernel.packet[2], kernel.packet[3]); |
| __m512 T4 = _mm512_unpacklo_ps(kernel.packet[4], kernel.packet[5]); |
| __m512 T5 = _mm512_unpackhi_ps(kernel.packet[4], kernel.packet[5]); |
| __m512 T6 = _mm512_unpacklo_ps(kernel.packet[6], kernel.packet[7]); |
| __m512 T7 = _mm512_unpackhi_ps(kernel.packet[6], kernel.packet[7]); |
| |
| kernel.packet[0] = _mm512_castpd_ps(_mm512_unpacklo_pd(_mm512_castps_pd(T0), _mm512_castps_pd(T2))); |
| kernel.packet[1] = _mm512_castpd_ps(_mm512_unpackhi_pd(_mm512_castps_pd(T0), _mm512_castps_pd(T2))); |
| kernel.packet[2] = _mm512_castpd_ps(_mm512_unpacklo_pd(_mm512_castps_pd(T1), _mm512_castps_pd(T3))); |
| kernel.packet[3] = _mm512_castpd_ps(_mm512_unpackhi_pd(_mm512_castps_pd(T1), _mm512_castps_pd(T3))); |
| kernel.packet[4] = _mm512_castpd_ps(_mm512_unpacklo_pd(_mm512_castps_pd(T4), _mm512_castps_pd(T6))); |
| kernel.packet[5] = _mm512_castpd_ps(_mm512_unpackhi_pd(_mm512_castps_pd(T4), _mm512_castps_pd(T6))); |
| kernel.packet[6] = _mm512_castpd_ps(_mm512_unpacklo_pd(_mm512_castps_pd(T5), _mm512_castps_pd(T7))); |
| kernel.packet[7] = _mm512_castpd_ps(_mm512_unpackhi_pd(_mm512_castps_pd(T5), _mm512_castps_pd(T7))); |
| |
| T0 = _mm512_shuffle_f32x4(kernel.packet[0], kernel.packet[4], 0x44); |
| T1 = _mm512_shuffle_f32x4(kernel.packet[0], kernel.packet[4], 0xee); |
| T2 = _mm512_shuffle_f32x4(kernel.packet[1], kernel.packet[5], 0x44); |
| T3 = _mm512_shuffle_f32x4(kernel.packet[1], kernel.packet[5], 0xee); |
| T4 = _mm512_shuffle_f32x4(kernel.packet[2], kernel.packet[6], 0x44); |
| T5 = _mm512_shuffle_f32x4(kernel.packet[2], kernel.packet[6], 0xee); |
| T6 = _mm512_shuffle_f32x4(kernel.packet[3], kernel.packet[7], 0x44); |
| T7 = _mm512_shuffle_f32x4(kernel.packet[3], kernel.packet[7], 0xee); |
| |
| kernel.packet[0] = _mm512_shuffle_f32x4(T0, T2, 0x88); |
| kernel.packet[2] = _mm512_shuffle_f32x4(T0, T2, 0xdd); |
| kernel.packet[1] = _mm512_shuffle_f32x4(T4, T6, 0x88); |
| kernel.packet[3] = _mm512_shuffle_f32x4(T4, T6, 0xdd); |
| kernel.packet[4] = _mm512_shuffle_f32x4(T1, T3, 0x88); |
| kernel.packet[6] = _mm512_shuffle_f32x4(T1, T3, 0xdd); |
| kernel.packet[5] = _mm512_shuffle_f32x4(T5, T7, 0x88); |
| kernel.packet[7] = _mm512_shuffle_f32x4(T5, T7, 0xdd); |
| } |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet16f, 4>& kernel) { |
| __m512 T0 = _mm512_unpacklo_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T1 = _mm512_unpackhi_ps(kernel.packet[0], kernel.packet[1]); |
| __m512 T2 = _mm512_unpacklo_ps(kernel.packet[2], kernel.packet[3]); |
| __m512 T3 = _mm512_unpackhi_ps(kernel.packet[2], kernel.packet[3]); |
| |
| __m512 S0 = _mm512_shuffle_ps(T0, T2, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S1 = _mm512_shuffle_ps(T0, T2, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512 S2 = _mm512_shuffle_ps(T1, T3, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512 S3 = _mm512_shuffle_ps(T1, T3, _MM_SHUFFLE(3, 2, 3, 2)); |
| |
| EIGEN_EXTRACT_8f_FROM_16f(S0, S0); |
| EIGEN_EXTRACT_8f_FROM_16f(S1, S1); |
| EIGEN_EXTRACT_8f_FROM_16f(S2, S2); |
| EIGEN_EXTRACT_8f_FROM_16f(S3, S3); |
| |
| PacketBlock<Packet8f, 8> tmp; |
| |
| tmp.packet[0] = _mm256_permute2f128_ps(S0_0, S1_0, 0x20); |
| tmp.packet[1] = _mm256_permute2f128_ps(S2_0, S3_0, 0x20); |
| tmp.packet[2] = _mm256_permute2f128_ps(S0_0, S1_0, 0x31); |
| tmp.packet[3] = _mm256_permute2f128_ps(S2_0, S3_0, 0x31); |
| |
| tmp.packet[4] = _mm256_permute2f128_ps(S0_1, S1_1, 0x20); |
| tmp.packet[5] = _mm256_permute2f128_ps(S2_1, S3_1, 0x20); |
| tmp.packet[6] = _mm256_permute2f128_ps(S0_1, S1_1, 0x31); |
| tmp.packet[7] = _mm256_permute2f128_ps(S2_1, S3_1, 0x31); |
| |
| PACK_OUTPUT_2(kernel.packet, tmp.packet, 0, 1); |
| PACK_OUTPUT_2(kernel.packet, tmp.packet, 1, 1); |
| PACK_OUTPUT_2(kernel.packet, tmp.packet, 2, 1); |
| PACK_OUTPUT_2(kernel.packet, tmp.packet, 3, 1); |
| } |
| |
| #define PACK_OUTPUT_SQ_D(OUTPUT, INPUT, INDEX, STRIDE) \ |
| OUTPUT[INDEX] = _mm512_insertf64x4(OUTPUT[INDEX], INPUT[INDEX], 0); \ |
| OUTPUT[INDEX] = _mm512_insertf64x4(OUTPUT[INDEX], INPUT[INDEX + STRIDE], 1); |
| |
| #define PACK_OUTPUT_D(OUTPUT, INPUT, INDEX, STRIDE) \ |
| OUTPUT[INDEX] = _mm512_insertf64x4(OUTPUT[INDEX], INPUT[(2 * INDEX)], 0); \ |
| OUTPUT[INDEX] = _mm512_insertf64x4(OUTPUT[INDEX], INPUT[(2 * INDEX) + STRIDE], 1); |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet8d, 4>& kernel) { |
| __m512d T0 = _mm512_shuffle_pd(kernel.packet[0], kernel.packet[1], 0); |
| __m512d T1 = _mm512_shuffle_pd(kernel.packet[0], kernel.packet[1], 0xff); |
| __m512d T2 = _mm512_shuffle_pd(kernel.packet[2], kernel.packet[3], 0); |
| __m512d T3 = _mm512_shuffle_pd(kernel.packet[2], kernel.packet[3], 0xff); |
| |
| PacketBlock<Packet4d, 8> tmp; |
| |
| tmp.packet[0] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T0, 0), _mm512_extractf64x4_pd(T2, 0), 0x20); |
| tmp.packet[1] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T1, 0), _mm512_extractf64x4_pd(T3, 0), 0x20); |
| tmp.packet[2] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T0, 0), _mm512_extractf64x4_pd(T2, 0), 0x31); |
| tmp.packet[3] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T1, 0), _mm512_extractf64x4_pd(T3, 0), 0x31); |
| |
| tmp.packet[4] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T0, 1), _mm512_extractf64x4_pd(T2, 1), 0x20); |
| tmp.packet[5] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T1, 1), _mm512_extractf64x4_pd(T3, 1), 0x20); |
| tmp.packet[6] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T0, 1), _mm512_extractf64x4_pd(T2, 1), 0x31); |
| tmp.packet[7] = _mm256_permute2f128_pd(_mm512_extractf64x4_pd(T1, 1), _mm512_extractf64x4_pd(T3, 1), 0x31); |
| |
| PACK_OUTPUT_D(kernel.packet, tmp.packet, 0, 1); |
| PACK_OUTPUT_D(kernel.packet, tmp.packet, 1, 1); |
| PACK_OUTPUT_D(kernel.packet, tmp.packet, 2, 1); |
| PACK_OUTPUT_D(kernel.packet, tmp.packet, 3, 1); |
| } |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet8d, 8>& kernel) { |
| __m512d T0 = _mm512_unpacklo_pd(kernel.packet[0], kernel.packet[1]); |
| __m512d T1 = _mm512_unpackhi_pd(kernel.packet[0], kernel.packet[1]); |
| __m512d T2 = _mm512_unpacklo_pd(kernel.packet[2], kernel.packet[3]); |
| __m512d T3 = _mm512_unpackhi_pd(kernel.packet[2], kernel.packet[3]); |
| __m512d T4 = _mm512_unpacklo_pd(kernel.packet[4], kernel.packet[5]); |
| __m512d T5 = _mm512_unpackhi_pd(kernel.packet[4], kernel.packet[5]); |
| __m512d T6 = _mm512_unpacklo_pd(kernel.packet[6], kernel.packet[7]); |
| __m512d T7 = _mm512_unpackhi_pd(kernel.packet[6], kernel.packet[7]); |
| |
| kernel.packet[0] = _mm512_permutex_pd(T2, 0x4E); |
| kernel.packet[0] = _mm512_mask_blend_pd(0xCC, T0, kernel.packet[0]); |
| kernel.packet[2] = _mm512_permutex_pd(T0, 0x4E); |
| kernel.packet[2] = _mm512_mask_blend_pd(0xCC, kernel.packet[2], T2); |
| kernel.packet[1] = _mm512_permutex_pd(T3, 0x4E); |
| kernel.packet[1] = _mm512_mask_blend_pd(0xCC, T1, kernel.packet[1]); |
| kernel.packet[3] = _mm512_permutex_pd(T1, 0x4E); |
| kernel.packet[3] = _mm512_mask_blend_pd(0xCC, kernel.packet[3], T3); |
| kernel.packet[4] = _mm512_permutex_pd(T6, 0x4E); |
| kernel.packet[4] = _mm512_mask_blend_pd(0xCC, T4, kernel.packet[4]); |
| kernel.packet[6] = _mm512_permutex_pd(T4, 0x4E); |
| kernel.packet[6] = _mm512_mask_blend_pd(0xCC, kernel.packet[6], T6); |
| kernel.packet[5] = _mm512_permutex_pd(T7, 0x4E); |
| kernel.packet[5] = _mm512_mask_blend_pd(0xCC, T5, kernel.packet[5]); |
| kernel.packet[7] = _mm512_permutex_pd(T5, 0x4E); |
| kernel.packet[7] = _mm512_mask_blend_pd(0xCC, kernel.packet[7], T7); |
| |
| T0 = _mm512_shuffle_f64x2(kernel.packet[4], kernel.packet[4], 0x4E); |
| T0 = _mm512_mask_blend_pd(0xF0, kernel.packet[0], T0); |
| T4 = _mm512_shuffle_f64x2(kernel.packet[0], kernel.packet[0], 0x4E); |
| T4 = _mm512_mask_blend_pd(0xF0, T4, kernel.packet[4]); |
| T1 = _mm512_shuffle_f64x2(kernel.packet[5], kernel.packet[5], 0x4E); |
| T1 = _mm512_mask_blend_pd(0xF0, kernel.packet[1], T1); |
| T5 = _mm512_shuffle_f64x2(kernel.packet[1], kernel.packet[1], 0x4E); |
| T5 = _mm512_mask_blend_pd(0xF0, T5, kernel.packet[5]); |
| T2 = _mm512_shuffle_f64x2(kernel.packet[6], kernel.packet[6], 0x4E); |
| T2 = _mm512_mask_blend_pd(0xF0, kernel.packet[2], T2); |
| T6 = _mm512_shuffle_f64x2(kernel.packet[2], kernel.packet[2], 0x4E); |
| T6 = _mm512_mask_blend_pd(0xF0, T6, kernel.packet[6]); |
| T3 = _mm512_shuffle_f64x2(kernel.packet[7], kernel.packet[7], 0x4E); |
| T3 = _mm512_mask_blend_pd(0xF0, kernel.packet[3], T3); |
| T7 = _mm512_shuffle_f64x2(kernel.packet[3], kernel.packet[3], 0x4E); |
| T7 = _mm512_mask_blend_pd(0xF0, T7, kernel.packet[7]); |
| |
| kernel.packet[0] = T0; |
| kernel.packet[1] = T1; |
| kernel.packet[2] = T2; |
| kernel.packet[3] = T3; |
| kernel.packet[4] = T4; |
| kernel.packet[5] = T5; |
| kernel.packet[6] = T6; |
| kernel.packet[7] = T7; |
| } |
| |
| #define PACK_OUTPUT_I32(OUTPUT, INPUT, INDEX, STRIDE) \ |
| EIGEN_INSERT_8i_INTO_16i(OUTPUT[INDEX], INPUT[INDEX], INPUT[INDEX + STRIDE]); |
| |
| #define PACK_OUTPUT_I32_2(OUTPUT, INPUT, INDEX, STRIDE) \ |
| EIGEN_INSERT_8i_INTO_16i(OUTPUT[INDEX], INPUT[2 * INDEX], INPUT[2 * INDEX + STRIDE]); |
| |
| #define SHUFFLE_EPI32(A, B, M) _mm512_castps_si512(_mm512_shuffle_ps(_mm512_castsi512_ps(A), _mm512_castsi512_ps(B), M)) |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet16i, 16>& kernel) { |
| __m512i T0 = _mm512_unpacklo_epi32(kernel.packet[0], kernel.packet[1]); |
| __m512i T1 = _mm512_unpackhi_epi32(kernel.packet[0], kernel.packet[1]); |
| __m512i T2 = _mm512_unpacklo_epi32(kernel.packet[2], kernel.packet[3]); |
| __m512i T3 = _mm512_unpackhi_epi32(kernel.packet[2], kernel.packet[3]); |
| __m512i T4 = _mm512_unpacklo_epi32(kernel.packet[4], kernel.packet[5]); |
| __m512i T5 = _mm512_unpackhi_epi32(kernel.packet[4], kernel.packet[5]); |
| __m512i T6 = _mm512_unpacklo_epi32(kernel.packet[6], kernel.packet[7]); |
| __m512i T7 = _mm512_unpackhi_epi32(kernel.packet[6], kernel.packet[7]); |
| __m512i T8 = _mm512_unpacklo_epi32(kernel.packet[8], kernel.packet[9]); |
| __m512i T9 = _mm512_unpackhi_epi32(kernel.packet[8], kernel.packet[9]); |
| __m512i T10 = _mm512_unpacklo_epi32(kernel.packet[10], kernel.packet[11]); |
| __m512i T11 = _mm512_unpackhi_epi32(kernel.packet[10], kernel.packet[11]); |
| __m512i T12 = _mm512_unpacklo_epi32(kernel.packet[12], kernel.packet[13]); |
| __m512i T13 = _mm512_unpackhi_epi32(kernel.packet[12], kernel.packet[13]); |
| __m512i T14 = _mm512_unpacklo_epi32(kernel.packet[14], kernel.packet[15]); |
| __m512i T15 = _mm512_unpackhi_epi32(kernel.packet[14], kernel.packet[15]); |
| __m512i S0 = SHUFFLE_EPI32(T0, T2, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S1 = SHUFFLE_EPI32(T0, T2, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S2 = SHUFFLE_EPI32(T1, T3, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S3 = SHUFFLE_EPI32(T1, T3, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S4 = SHUFFLE_EPI32(T4, T6, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S5 = SHUFFLE_EPI32(T4, T6, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S6 = SHUFFLE_EPI32(T5, T7, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S7 = SHUFFLE_EPI32(T5, T7, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S8 = SHUFFLE_EPI32(T8, T10, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S9 = SHUFFLE_EPI32(T8, T10, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S10 = SHUFFLE_EPI32(T9, T11, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S11 = SHUFFLE_EPI32(T9, T11, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S12 = SHUFFLE_EPI32(T12, T14, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S13 = SHUFFLE_EPI32(T12, T14, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S14 = SHUFFLE_EPI32(T13, T15, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S15 = SHUFFLE_EPI32(T13, T15, _MM_SHUFFLE(3, 2, 3, 2)); |
| |
| EIGEN_EXTRACT_8i_FROM_16i(S0, S0); |
| EIGEN_EXTRACT_8i_FROM_16i(S1, S1); |
| EIGEN_EXTRACT_8i_FROM_16i(S2, S2); |
| EIGEN_EXTRACT_8i_FROM_16i(S3, S3); |
| EIGEN_EXTRACT_8i_FROM_16i(S4, S4); |
| EIGEN_EXTRACT_8i_FROM_16i(S5, S5); |
| EIGEN_EXTRACT_8i_FROM_16i(S6, S6); |
| EIGEN_EXTRACT_8i_FROM_16i(S7, S7); |
| EIGEN_EXTRACT_8i_FROM_16i(S8, S8); |
| EIGEN_EXTRACT_8i_FROM_16i(S9, S9); |
| EIGEN_EXTRACT_8i_FROM_16i(S10, S10); |
| EIGEN_EXTRACT_8i_FROM_16i(S11, S11); |
| EIGEN_EXTRACT_8i_FROM_16i(S12, S12); |
| EIGEN_EXTRACT_8i_FROM_16i(S13, S13); |
| EIGEN_EXTRACT_8i_FROM_16i(S14, S14); |
| EIGEN_EXTRACT_8i_FROM_16i(S15, S15); |
| |
| PacketBlock<Packet8i, 32> tmp; |
| |
| tmp.packet[0] = _mm256_permute2f128_si256(S0_0, S4_0, 0x20); |
| tmp.packet[1] = _mm256_permute2f128_si256(S1_0, S5_0, 0x20); |
| tmp.packet[2] = _mm256_permute2f128_si256(S2_0, S6_0, 0x20); |
| tmp.packet[3] = _mm256_permute2f128_si256(S3_0, S7_0, 0x20); |
| tmp.packet[4] = _mm256_permute2f128_si256(S0_0, S4_0, 0x31); |
| tmp.packet[5] = _mm256_permute2f128_si256(S1_0, S5_0, 0x31); |
| tmp.packet[6] = _mm256_permute2f128_si256(S2_0, S6_0, 0x31); |
| tmp.packet[7] = _mm256_permute2f128_si256(S3_0, S7_0, 0x31); |
| |
| tmp.packet[8] = _mm256_permute2f128_si256(S0_1, S4_1, 0x20); |
| tmp.packet[9] = _mm256_permute2f128_si256(S1_1, S5_1, 0x20); |
| tmp.packet[10] = _mm256_permute2f128_si256(S2_1, S6_1, 0x20); |
| tmp.packet[11] = _mm256_permute2f128_si256(S3_1, S7_1, 0x20); |
| tmp.packet[12] = _mm256_permute2f128_si256(S0_1, S4_1, 0x31); |
| tmp.packet[13] = _mm256_permute2f128_si256(S1_1, S5_1, 0x31); |
| tmp.packet[14] = _mm256_permute2f128_si256(S2_1, S6_1, 0x31); |
| tmp.packet[15] = _mm256_permute2f128_si256(S3_1, S7_1, 0x31); |
| |
| // Second set of _m256 outputs |
| tmp.packet[16] = _mm256_permute2f128_si256(S8_0, S12_0, 0x20); |
| tmp.packet[17] = _mm256_permute2f128_si256(S9_0, S13_0, 0x20); |
| tmp.packet[18] = _mm256_permute2f128_si256(S10_0, S14_0, 0x20); |
| tmp.packet[19] = _mm256_permute2f128_si256(S11_0, S15_0, 0x20); |
| tmp.packet[20] = _mm256_permute2f128_si256(S8_0, S12_0, 0x31); |
| tmp.packet[21] = _mm256_permute2f128_si256(S9_0, S13_0, 0x31); |
| tmp.packet[22] = _mm256_permute2f128_si256(S10_0, S14_0, 0x31); |
| tmp.packet[23] = _mm256_permute2f128_si256(S11_0, S15_0, 0x31); |
| |
| tmp.packet[24] = _mm256_permute2f128_si256(S8_1, S12_1, 0x20); |
| tmp.packet[25] = _mm256_permute2f128_si256(S9_1, S13_1, 0x20); |
| tmp.packet[26] = _mm256_permute2f128_si256(S10_1, S14_1, 0x20); |
| tmp.packet[27] = _mm256_permute2f128_si256(S11_1, S15_1, 0x20); |
| tmp.packet[28] = _mm256_permute2f128_si256(S8_1, S12_1, 0x31); |
| tmp.packet[29] = _mm256_permute2f128_si256(S9_1, S13_1, 0x31); |
| tmp.packet[30] = _mm256_permute2f128_si256(S10_1, S14_1, 0x31); |
| tmp.packet[31] = _mm256_permute2f128_si256(S11_1, S15_1, 0x31); |
| |
| // Pack them into the output |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 0, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 1, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 2, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 3, 16); |
| |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 4, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 5, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 6, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 7, 16); |
| |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 8, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 9, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 10, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 11, 16); |
| |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 12, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 13, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 14, 16); |
| PACK_OUTPUT_I32(kernel.packet, tmp.packet, 15, 16); |
| } |
| |
| EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock<Packet16i, 4>& kernel) { |
| __m512i T0 = _mm512_unpacklo_epi32(kernel.packet[0], kernel.packet[1]); |
| __m512i T1 = _mm512_unpackhi_epi32(kernel.packet[0], kernel.packet[1]); |
| __m512i T2 = _mm512_unpacklo_epi32(kernel.packet[2], kernel.packet[3]); |
| __m512i T3 = _mm512_unpackhi_epi32(kernel.packet[2], kernel.packet[3]); |
| |
| __m512i S0 = SHUFFLE_EPI32(T0, T2, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S1 = SHUFFLE_EPI32(T0, T2, _MM_SHUFFLE(3, 2, 3, 2)); |
| __m512i S2 = SHUFFLE_EPI32(T1, T3, _MM_SHUFFLE(1, 0, 1, 0)); |
| __m512i S3 = SHUFFLE_EPI32(T1, T3, _MM_SHUFFLE(3, 2, 3, 2)); |
| |
| EIGEN_EXTRACT_8i_FROM_16i(S0, S0); |
| EIGEN_EXTRACT_8i_FROM_16i(S1, S1); |
| EIGEN_EXTRACT_8i_FROM_16i(S2, S2); |
| EIGEN_EXTRACT_8i_FROM_16i(S3, S3); |
| |
| PacketBlock<Packet8i, 8> tmp; |
| |
| tmp.packet[0] = _mm256_permute2f128_si256(S0_0, S1_0, 0x20); |
| tmp.packet[1] = _mm256_permute2f128_si256(S2_0, S3_0, 0x20); |
| tmp.packet[2] = _mm256_permute2f128_si256(S0_0, S1_0, 0x31); |
| tmp.packet[3] = _mm256_permute2f128_si256(S2_0, S3_0, 0x31); |
| |
| tmp.packet[4] = _mm256_permute2f128_si256(S0_1, S1_1, 0x20); |
| tmp.packet[5] = _mm256_permute2f128_si256(S2_1, S3_1, 0x20); |
| tmp.packet[6] = _mm256_permute2f128_si256(S0_1, S1_1, 0x31); |
| tmp.packet[7] = _mm256_permute2f128_si256(S2_1, S3_1, 0x31); |
| |
| PACK_OUTPUT_I32_2(kernel.packet, tmp.packet, 0, 1); |
| PACK_OUTPUT_I32_2(kernel.packet, tmp.packet, 1, 1); |
| PACK_OUTPUT_I32_2(kernel.packet, tmp.packet, 2, 1); |
| PACK_OUTPUT_I32_2(kernel.packet, tmp.packet, 3, 1); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16f pblend(const Selector<16>& ifPacket, const Packet16f& thenPacket, |
| const Packet16f& elsePacket) { |
| __mmask16 m = (ifPacket.select[0]) | (ifPacket.select[1] << 1) | (ifPacket.select[2] << 2) | |
| (ifPacket.select[3] << 3) | (ifPacket.select[4] << 4) | (ifPacket.select[5] << 5) | |
| (ifPacket.select[6] << 6) | (ifPacket.select[7] << 7) | (ifPacket.select[8] << 8) | |
| (ifPacket.select[9] << 9) | (ifPacket.select[10] << 10) | (ifPacket.select[11] << 11) | |
| (ifPacket.select[12] << 12) | (ifPacket.select[13] << 13) | (ifPacket.select[14] << 14) | |
| (ifPacket.select[15] << 15); |
| return _mm512_mask_blend_ps(m, elsePacket, thenPacket); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet8d pblend(const Selector<8>& ifPacket, const Packet8d& thenPacket, |
| const Packet8d& elsePacket) { |
| __mmask8 m = (ifPacket.select[0]) | (ifPacket.select[1] << 1) | (ifPacket.select[2] << 2) | |
| (ifPacket.select[3] << 3) | (ifPacket.select[4] << 4) | (ifPacket.select[5] << 5) | |
| (ifPacket.select[6] << 6) | (ifPacket.select[7] << 7); |
| return _mm512_mask_blend_pd(m, elsePacket, thenPacket); |
| } |
| |
| // Packet math for Eigen::half |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pset1<Packet16h>(const Eigen::half& from) { |
| return _mm256_set1_epi16(from.x); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Eigen::half pfirst<Packet16h>(const Packet16h& from) { |
| return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm256_extract_epi16(from, 0))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pload<Packet16h>(const Eigen::half* from) { |
| return _mm256_load_si256(reinterpret_cast<const __m256i*>(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h ploadu<Packet16h>(const Eigen::half* from) { |
| return _mm256_loadu_si256(reinterpret_cast<const __m256i*>(from)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pstore<half>(Eigen::half* to, const Packet16h& from) { |
| // (void*) -> workaround clang warning: |
| // cast from 'Eigen::half *' to '__m256i *' increases required alignment from 2 to 32 |
| _mm256_store_si256((__m256i*)(void*)to, from); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pstoreu<half>(Eigen::half* to, const Packet16h& from) { |
| // (void*) -> workaround clang warning: |
| // cast from 'Eigen::half *' to '__m256i *' increases required alignment from 2 to 32 |
| _mm256_storeu_si256((__m256i*)(void*)to, from); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h ploaddup<Packet16h>(const Eigen::half* from) { |
| unsigned short a = from[0].x; |
| unsigned short b = from[1].x; |
| unsigned short c = from[2].x; |
| unsigned short d = from[3].x; |
| unsigned short e = from[4].x; |
| unsigned short f = from[5].x; |
| unsigned short g = from[6].x; |
| unsigned short h = from[7].x; |
| return _mm256_set_epi16(h, h, g, g, f, f, e, e, d, d, c, c, b, b, a, a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h ploadquad(const Eigen::half* from) { |
| unsigned short a = from[0].x; |
| unsigned short b = from[1].x; |
| unsigned short c = from[2].x; |
| unsigned short d = from[3].x; |
| return _mm256_set_epi16(d, d, d, d, c, c, c, c, b, b, b, b, a, a, a, a); |
| } |
| |
| EIGEN_STRONG_INLINE Packet16f half2float(const Packet16h& a) { return _mm512_cvtph_ps(a); } |
| |
| EIGEN_STRONG_INLINE Packet16h float2half(const Packet16f& a) { |
| return _mm512_cvtps_ph(a, _MM_FROUND_TO_NEAREST_INT | _MM_FROUND_NO_EXC); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h ptrue(const Packet16h& a) { |
| return Packet16h(ptrue(Packet8i(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pabs(const Packet16h& a) { |
| const __m256i sign_mask = _mm256_set1_epi16(static_cast<numext::uint16_t>(0x8000)); |
| return _mm256_andnot_si256(sign_mask, a); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pmin<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| return float2half(pmin<Packet16f>(half2float(a), half2float(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pmax<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| return float2half(pmax<Packet16f>(half2float(a), half2float(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h plset<Packet16h>(const half& a) { |
| return float2half(plset<Packet16f>(static_cast<float>(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h por(const Packet16h& a, const Packet16h& b) { |
| // in some cases Packet8i is a wrapper around __m256i, so we need to |
| // cast to Packet8i to call the correct overload. |
| return Packet16h(por(Packet8i(a), Packet8i(b))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pxor(const Packet16h& a, const Packet16h& b) { |
| return Packet16h(pxor(Packet8i(a), Packet8i(b))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pand(const Packet16h& a, const Packet16h& b) { |
| return Packet16h(pand(Packet8i(a), Packet8i(b))); |
| } |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pandnot(const Packet16h& a, const Packet16h& b) { |
| return Packet16h(pandnot(Packet8i(a), Packet8i(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pselect(const Packet16h& mask, const Packet16h& a, const Packet16h& b) { |
| return _mm256_blendv_epi8(b, a, mask); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pround<Packet16h>(const Packet16h& a) { |
| return float2half(pround<Packet16f>(half2float(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h print<Packet16h>(const Packet16h& a) { |
| return float2half(print<Packet16f>(half2float(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pceil<Packet16h>(const Packet16h& a) { |
| return float2half(pceil<Packet16f>(half2float(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pfloor<Packet16h>(const Packet16h& a) { |
| return float2half(pfloor<Packet16f>(half2float(a))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pcmp_eq(const Packet16h& a, const Packet16h& b) { |
| Packet16f af = half2float(a); |
| Packet16f bf = half2float(b); |
| return Pack32To16(pcmp_eq(af, bf)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pcmp_le(const Packet16h& a, const Packet16h& b) { |
| return Pack32To16(pcmp_le(half2float(a), half2float(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pcmp_lt(const Packet16h& a, const Packet16h& b) { |
| return Pack32To16(pcmp_lt(half2float(a), half2float(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pcmp_lt_or_nan(const Packet16h& a, const Packet16h& b) { |
| return Pack32To16(pcmp_lt_or_nan(half2float(a), half2float(b))); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pconj(const Packet16h& a) { |
| return a; |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pnegate(const Packet16h& a) { |
| Packet16h sign_mask = _mm256_set1_epi16(static_cast<unsigned short>(0x8000)); |
| return _mm256_xor_si256(a, sign_mask); |
| } |
| |
| #ifndef EIGEN_VECTORIZE_AVX512FP16 |
| template <> |
| EIGEN_STRONG_INLINE Packet16h padd<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| Packet16f af = half2float(a); |
| Packet16f bf = half2float(b); |
| Packet16f rf = padd(af, bf); |
| return float2half(rf); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h psub<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| Packet16f af = half2float(a); |
| Packet16f bf = half2float(b); |
| Packet16f rf = psub(af, bf); |
| return float2half(rf); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pmul<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| Packet16f af = half2float(a); |
| Packet16f bf = half2float(b); |
| Packet16f rf = pmul(af, bf); |
| return float2half(rf); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pdiv<Packet16h>(const Packet16h& a, const Packet16h& b) { |
| Packet16f af = half2float(a); |
| Packet16f bf = half2float(b); |
| Packet16f rf = pdiv(af, bf); |
| return float2half(rf); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE half predux<Packet16h>(const Packet16h& from) { |
| Packet16f from_float = half2float(from); |
| return half(predux(from_float)); |
| } |
| |
| #endif |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet8h predux_half_dowto4<Packet16h>(const Packet16h& a) { |
| Packet8h lane0 = _mm256_extractf128_si256(a, 0); |
| Packet8h lane1 = _mm256_extractf128_si256(a, 1); |
| return padd<Packet8h>(lane0, lane1); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Eigen::half predux_max<Packet16h>(const Packet16h& a) { |
| Packet16f af = half2float(a); |
| float reduced = predux_max<Packet16f>(af); |
| return Eigen::half(reduced); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Eigen::half predux_min<Packet16h>(const Packet16h& a) { |
| Packet16f af = half2float(a); |
| float reduced = predux_min<Packet16f>(af); |
| return Eigen::half(reduced); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE half predux_mul<Packet16h>(const Packet16h& from) { |
| Packet16f from_float = half2float(from); |
| return half(predux_mul(from_float)); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h preverse(const Packet16h& a) { |
| __m128i m = _mm_setr_epi8(14, 15, 12, 13, 10, 11, 8, 9, 6, 7, 4, 5, 2, 3, 0, 1); |
| return _mm256_insertf128_si256(_mm256_castsi128_si256(_mm_shuffle_epi8(_mm256_extractf128_si256(a, 1), m)), |
| _mm_shuffle_epi8(_mm256_extractf128_si256(a, 0), m), 1); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE Packet16h pgather<Eigen::half, Packet16h>(const Eigen::half* from, Index stride) { |
| return _mm256_set_epi16(from[15 * stride].x, from[14 * stride].x, from[13 * stride].x, from[12 * stride].x, |
| from[11 * stride].x, from[10 * stride].x, from[9 * stride].x, from[8 * stride].x, |
| from[7 * stride].x, from[6 * stride].x, from[5 * stride].x, from[4 * stride].x, |
| from[3 * stride].x, from[2 * stride].x, from[1 * stride].x, from[0 * stride].x); |
| } |
| |
| template <> |
| EIGEN_STRONG_INLINE void pscatter<half, Packet16h>(half* to, const Packet16h& from, Index stride) { |
| EIGEN_ALIGN64 half aux[16]; |
| pstore(aux, from); |
| to[stride * 0] = aux[0]; |
| to[stride * 1] = aux[1]; |
| to[stride * 2] = aux[2]; |
| to[stride * 3] = aux[3]; |
| to[stride * 4] = aux[4]; |
| to[stride * 5] = aux[5]; |
| to[stride * 6] = aux[6]; |
| to[stride * 7] = aux[7]; |
| to[stride * 8] = aux[8]; |
| to[stride * 9] = aux[9]; |
| to[stride * 10] = aux[10]; |
| to[stride * 11] = aux[11]; |
| to[stride * 12] = aux[12]; |
| to[stride * 13] = aux[13]; |
| to[stride * 14] = aux[14]; |
| to[stride * 15] = aux[15]; |
| } |
| |
| EIGEN_STRONG_INLINE void ptranspose(PacketBlock<Packet16h, 16>& kernel) { |
| __m256i a = kernel.packet[0]; |
| __m256i b = kernel.packet[1]; |
| __m256i c = kernel.packet[2]; |
| __m256i d = kernel.packet[3]; |
| __m256i e = kernel.packet[4]; |
| __m256i f = kernel.packet[5]; |
| __m256i g = kernel.packet[6]; |
| __m256i h = kernel.packet[7]; |
| __m256i i = kernel.packet[8]; |
| __m256i j = kernel.packet[9]; |
| __m256i k = kernel.packet[10]; |
| __m256i l = kernel.packet[11]; |
| __m256i m = kernel.packet[12]; |
| __m256i n = kernel.packet[13]; |
| __m256i o = kernel.packet[14]; |
| __m256i p = kernel.packet[15]; |
| |
| __m256i ab_07 = _mm256_unpacklo_epi16(a, b); |
| __m256i cd_07 = _mm256_unpacklo_epi16(c, d); |
| __m256i ef_07 = _mm256_unpacklo_epi16(e, f); |
| __m256i gh_07 = _mm256_unpacklo_epi16(g, h); |
| __m256i ij_07 = _mm256_unpacklo_epi16(i, j); |
| __m256i kl_07 = _mm256_unpacklo_epi16(k, l); |
| __m256i mn_07 = _mm256_unpacklo_epi16(m, n); |
| __m256i op_07 = _mm256_unpacklo_epi16(o, p); |
| |
| __m256i ab_8f = _mm256_unpackhi_epi16(a, b); |
| __m256i cd_8f = _mm256_unpackhi_epi16(c, d); |
| __m256i ef_8f = _mm256_unpackhi_epi16(e, f); |
| __m256i gh_8f = _mm256_unpackhi_epi16(g, h); |
| __m256i ij_8f = _mm256_unpackhi_epi16(i, j); |
| __m256i kl_8f = _mm256_unpackhi_epi16(k, l); |
| __m256i mn_8f = _mm256_unpackhi_epi16(m, n); |
| __m256i op_8f = _mm256_unpackhi_epi16(o, p); |
| |
| __m256i abcd_03 = _mm256_unpacklo_epi32(ab_07, cd_07); |
| __m256i abcd_47 = _mm256_unpackhi_epi32(ab_07, cd_07); |
| __m256i efgh_03 = _mm256_unpacklo_epi32(ef_07, gh_07); |
| __m256i efgh_47 = _mm256_unpackhi_epi32(ef_07, gh_07); |
| __m256i ijkl_03 = _mm256_unpacklo_epi32(ij_07, kl_07); |
| __m256i ijkl_47 = _mm256_unpackhi_epi32(ij_07, kl_07); |
| __m256i mnop_03 = _mm256_unpacklo_epi32(mn_07, op_07); |
| __m256i mnop_47 = _mm256_unpackhi_epi32(mn_07, op_07); |
| |
| __m256i abcd_8b = _mm256_unpacklo_epi32(ab_8f, cd_8f); |
| __m256i abcd_cf = _mm256_unpackhi_epi32(ab_8f, cd_8f); |
| __m256i efgh_8b = _mm256_unpacklo_epi32(ef_8f, gh_8f); |
| __m256i efgh_cf = _mm256_unpackhi_epi32(ef_8f, gh_8f); |
| __m256i ijkl_8b = _mm256_unpacklo_epi32(ij_8f, kl_8f); |
| __m256i ijkl_cf = _mm256_unpackhi_epi32(ij_8f, kl_8f); |
| __m256i mnop_8b = _mm256_unpacklo_epi32(mn_8f, op_8f); |
| __m256i mnop_cf = _mm256_unpackhi_epi32(mn_8f, op_8f); |
| |
| __m256i abcdefgh_01 = _mm256_unpacklo_epi64(abcd_03, efgh_03); |
| __m256i abcdefgh_23 = _mm256_unpackhi_epi64(abcd_03, efgh_03); |
| __m256i ijklmnop_01 = _mm256_unpacklo_epi64(ijkl_03, mnop_03); |
| __m256i ijklmnop_23 = _mm256_unpackhi_epi64(ijkl_03, mnop_03); |
| __m256i abcdefgh_45 = _mm256_unpacklo_epi64(abcd_47, efgh_47); |
| __m256i abcdefgh_67 = _mm256_unpackhi_epi64(abcd_47, efgh_47); |
| __m256i ijklmnop_45 = _mm256_unpacklo_epi64(ijkl_47, mnop_47); |
| __m256i ijklmnop_67 = _mm256_unpackhi_epi64(ijkl_47, mnop_47); |
| __m256i abcdefgh_89 = _mm256_unpacklo_epi64(abcd_8b, efgh_8b); |
| __m256i abcdefgh_ab = _mm256_unpackhi_epi64(abcd_8b, efgh_8b); |
| __m256i ijklmnop_89 = _mm256_unpacklo_epi64(ijkl_8b, mnop_8b); |
| __m256i ijklmnop_ab = _mm256_unpackhi_epi64(ijkl_8b, mnop_8b); |
| __m256i abcdefgh_cd = _mm256_unpacklo_epi64(abcd_cf, efgh_cf); |
| __m256i abcdefgh_ef = _mm256_unpackhi_epi64(abcd_cf, efgh_cf); |
| __m256i ijklmnop_cd = _mm256_unpacklo_epi64(ijkl_cf, mnop_cf); |
| __m256i ijklmnop_ef = _mm256_unpackhi_epi64(ijkl_cf, mnop_cf); |
| |
| // NOTE: no unpacklo/hi instr in this case, so using permute instr. |
| __m256i a_p_0 = _mm256_permute2x128_si256(abcdefgh_01, ijklmnop_01, 0x20); |
| __m256i a_p_1 = _mm256_permute2x128_si256(abcdefgh_23, ijklmnop_23, 0x20); |
| __m256i a_p_2 = _mm256_permute2x128_si256(abcdefgh_45, ijklmnop_45, 0x20); |
| __m256i a_p_3 = _mm256_permute2x128_si256(abcdefgh_67, ijklmnop_67, 0x20); |
| __m256i a_p_4 = _mm256_permute2x128_si256(abcdefgh_89, ijklmnop_89, 0x20); |
| __m256i a_p_5 = _mm256_permute2x128_si256(abcdefgh_ab, ijklmnop_ab, 0x20); |
| __m256i a_p_6 = _mm256_permute2x128_si256(abcdefgh_cd, ijklmnop_cd, 0x20); |
| __m256i a_p_7 = _mm256_permute2x128_si256(abcdefgh_ef, ijklmnop_ef, 0x20); |
| __m256i a_p_8 = _mm256_permute2x128_si256(abcdefgh_01, ijklmnop_01, 0x31); |
| __m256i a_p_9 = _mm256_permute2x128_si256(abcdefgh_23, ijklmnop_23, 0x31); |
| __m256i a_p_a = _mm256_permute2x128_si256(abcdefgh_45, ijklmnop_45, 0x31); |
| __m256i a_p_b = _mm256_permute2x128_si256(abcdefgh_67, ijklmnop_67, 0x31); |
| __m256i a_p_c = _mm256_permute2x128_si256(abcdefgh_89, ijklmnop_89, 0x31); |
| __m256i a_p_d = _mm256_permute2x128_si256(abcdefgh_ab, ijklmnop_ab, 0x31); |
| __m256i a_p_e = _mm256_permute2x128_si256(abcdefgh_cd, ijklmnop_cd, 0x31); |
| __m256i a_p_f = _mm256_permute2x128_si256(abcdefgh_ef, ijklmnop_ef, 0x31); |
| |
| kernel.packet[0] = a_p_0; |
| kernel.packet[1] = a_p_1; |
| kernel.packet[2] = a_p_2; |
| kernel.packet[3] = a_p_3; |
| kernel.packet[4] = a_p_4; |
| kernel.packet[5] = a_p_5; |
| kernel.packet[6] = a_p_6; |
| kernel.packet[7] = a_p_7; |
| kernel.packet[8] = a_p_8; |
| kernel.packet[9] = a_p_9; |
| kernel.packet[10] = a_p_a; |
| kernel.packet[11] = a_p_b; |
| kernel.packet[12] = a_p_c; |
| kernel.packet[13] = a_p_d; |
| kernel.packet[14] = a_p_e; |
| kernel.packet[15] = a_p_f; |
| } |
| |
| EIGEN_STRONG_INLINE void ptranspose(PacketBlock<Packet16h, 8>& kernel) { |
| EIGEN_ALIGN64 half in[8][16]; |
| pstore<half>(in[0], kernel.packet[0]); |
| pstore<half>(in[1], kernel.packet[1]); |
| pstore<half>(in[2], kernel.packet[2]); |
| pstore<half>(in[3], kernel.packet[3]); |
| pstore<half>(in[4], kernel.packet[4]); |
| pstore<half>(in[5], kernel.packet[5]); |
| pstore<half>(in[6], kernel.packet[6]); |
| pstore<half>(in[7], kernel.packet[7]); |
| |
| EIGEN_ALIGN64 half out[8][16]; |
| |
| for (int i = 0; i < 8; ++i) { |
| for (int j = 0; j < 8; ++j) { |
| out[i][j] = in[j][2 * i]; |
| } |
| for (int j = 0; j < 8; ++j) { |
| out[i][j + 8] = in[j][2 * i + 1]; |
| } |
| } |
| |
| kernel.packet[0] = pload<Packet16h>(out[0]); |
| kernel.packet[1] = pload<Packet16h>(out[1]); |
| kernel.packet[2] = pload<Packet16h>(out[2]); |
| kernel.packet[3] = pload<Packet16h>(out[3]); |
| kernel.packet[4] = pload<Packet16h>(out[4]); |
| kernel.packet[5] = pload<Packet16h>(out[5]); |
| kernel.packet[6] = pload<Packet16h>(out[6]); |
| kernel.packet[7] = pload<Packet16h>(out[7]); |
| } |
| |
| EIGEN_STRONG_INLINE void ptranspose(PacketBlock<Packet16h, 4>& kernel) { |
| EIGEN_ALIGN64 half in[4][16]; |
| pstore<half>(in[0], kernel.packet[0]); |
| pstore<half>(in[1], kernel.packet[1]); |
| pstore<half>(in[2], kernel.packet[2]); |
| pstore<half>(in[3], kernel.packet[3]); |
| |
| EIGEN_ALIGN64 half out[4][16]; |
| |
| for (int i = 0; i < 4; ++i) { |
| for (int j = 0; j < 4; ++j) { |
| out[i][j] = in[j][4 * i]; |
| } |
| for (int j = 0; j < 4; ++j) { |
| out[i][j + 4] = in[j][4 * i + 1]; |
| } |
| for (int j = 0; j < 4; ++j) { |
| out[i][j + 8] = in[j][4 * i + 2]; |
| } |
| for (int j = 0; j < 4; ++j) { |
| out[i][j + 12] = in[j][4 * i + 3]; |
| }
|