Add missing x86 pcasts

This commit is contained in:
Charles Schlosser 2023-07-28 23:41:38 +00:00 committed by Rasmus Munk Larsen
parent 24d15e086f
commit 5527e78a64
5 changed files with 177 additions and 238 deletions

View File

@ -206,6 +206,17 @@ struct type_casting_traits {
}; };
}; };
// provides a succint template to define vectorized casting traits with respect to the largest accessible packet types
template <typename Src, typename Tgt>
struct vectorized_type_casting_traits {
enum : int {
DefaultSrcPacketSize = packet_traits<Src>::size,
DefaultTgtPacketSize = packet_traits<Tgt>::size,
VectorizedCast = 1,
SrcCoeffRatio = plain_enum_max(DefaultTgtPacketSize / DefaultSrcPacketSize, 1),
TgtCoeffRatio = plain_enum_max(DefaultSrcPacketSize / DefaultTgtPacketSize, 1)
};
};
/** \internal Wrapper to ensure that multiple packet types can map to the same /** \internal Wrapper to ensure that multiple packet types can map to the same
same underlying vector type. */ same underlying vector type. */

View File

@ -17,76 +17,24 @@ namespace Eigen {
namespace internal { namespace internal {
#ifndef EIGEN_VECTORIZE_AVX512 #ifndef EIGEN_VECTORIZE_AVX512
template <> template<> struct type_casting_traits<float, bool> : vectorized_type_casting_traits<float, bool> {};
struct type_casting_traits<Eigen::half, float> { template<> struct type_casting_traits<bool, float> : vectorized_type_casting_traits<bool, float> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template<> struct type_casting_traits<float, int> : vectorized_type_casting_traits<float, int> {};
template<> struct type_casting_traits<int, float> : vectorized_type_casting_traits<int, float> {};
template <> template<> struct type_casting_traits<float, double> : vectorized_type_casting_traits<float, double> {};
struct type_casting_traits<float, Eigen::half> { template<> struct type_casting_traits<double, float> : vectorized_type_casting_traits<double, float> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <> template<> struct type_casting_traits<double, int> : vectorized_type_casting_traits<double, int> {};
struct type_casting_traits<bfloat16, float> { template<> struct type_casting_traits<int, double> : vectorized_type_casting_traits<int, double> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <> template<> struct type_casting_traits<half, float> : vectorized_type_casting_traits<half, float> {};
struct type_casting_traits<float, bfloat16> { template<> struct type_casting_traits<float, half> : vectorized_type_casting_traits<float, half> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <> template<> struct type_casting_traits<bfloat16, float> : vectorized_type_casting_traits<bfloat16, float> {};
struct type_casting_traits<float, bool> { template<> struct type_casting_traits<float, bfloat16> : vectorized_type_casting_traits<float, bfloat16> {};
enum { #endif
VectorizedCast = 1,
SrcCoeffRatio = 2,
TgtCoeffRatio = 1
};
};
#endif // EIGEN_VECTORIZE_AVX512
template<> EIGEN_STRONG_INLINE Packet8i pcast<Packet8f, Packet8i>(const Packet8f& a) {
return _mm256_cvttps_epi32(a);
}
template<> EIGEN_STRONG_INLINE Packet8f pcast<Packet8i, Packet8f>(const Packet8i& a) {
return _mm256_cvtepi32_ps(a);
}
template<> EIGEN_STRONG_INLINE Packet8f pcast<Packet4d, Packet8f>(const Packet4d& a, const Packet4d& b) {
return _mm256_set_m128(_mm256_cvtpd_ps(b), _mm256_cvtpd_ps(a));
}
template<> EIGEN_STRONG_INLINE Packet8i pcast<Packet4d, Packet8i>(const Packet4d& a, const Packet4d& b) {
return _mm256_set_m128i(_mm256_cvttpd_epi32(b), _mm256_cvttpd_epi32(a));
}
template <> EIGEN_STRONG_INLINE Packet4f pcast<Packet4d, Packet4f>(const Packet4d& a) {
return _mm256_cvtpd_ps(a);
}
template <> EIGEN_STRONG_INLINE Packet4i pcast<Packet4d, Packet4i>(const Packet4d& a) {
return _mm256_cvttpd_epi32(a);
}
template <> template <>
EIGEN_STRONG_INLINE Packet16b pcast<Packet8f, Packet16b>(const Packet8f& a, EIGEN_STRONG_INLINE Packet16b pcast<Packet8f, Packet16b>(const Packet8f& a,
@ -118,6 +66,63 @@ EIGEN_STRONG_INLINE Packet16b pcast<Packet8f, Packet16b>(const Packet8f& a,
#endif #endif
} }
template <>
EIGEN_STRONG_INLINE Packet8f pcast<Packet16b, Packet8f>(const Packet16b& a) {
const __m256 cst_one = _mm256_set1_ps(1.0f);
#ifdef EIGEN_VECTORIZE_AVX2
__m256i a_extended = _mm256_cvtepi8_epi32(a);
__m256i abcd_efgh = _mm256_cmpeq_epi32(a_extended, _mm256_setzero_si256());
#else
__m128i abcd_efhg_ijkl_mnop = _mm_cmpeq_epi8(a, _mm_setzero_si128());
__m128i aabb_ccdd_eeff_gghh = _mm_unpacklo_epi8(abcd_efhg_ijkl_mnop, abcd_efhg_ijkl_mnop);
__m128i aaaa_bbbb_cccc_dddd = _mm_unpacklo_epi8(aabb_ccdd_eeff_gghh, aabb_ccdd_eeff_gghh);
__m128i eeee_ffff_gggg_hhhh = _mm_unpackhi_epi8(aabb_ccdd_eeff_gghh, aabb_ccdd_eeff_gghh);
__m256i abcd_efgh = _mm256_setr_m128i(aaaa_bbbb_cccc_dddd, eeee_ffff_gggg_hhhh);
#endif
__m256 result = _mm256_andnot_ps(_mm256_castsi256_ps(abcd_efgh), cst_one);
return result;
}
template<> EIGEN_STRONG_INLINE Packet8i pcast<Packet8f, Packet8i>(const Packet8f& a) {
return _mm256_cvttps_epi32(a);
}
template<> EIGEN_STRONG_INLINE Packet8i pcast<Packet4d, Packet8i>(const Packet4d& a, const Packet4d& b) {
return _mm256_set_m128i(_mm256_cvttpd_epi32(b), _mm256_cvttpd_epi32(a));
}
template <> EIGEN_STRONG_INLINE Packet4i pcast<Packet4d, Packet4i>(const Packet4d& a) {
return _mm256_cvttpd_epi32(a);
}
template<> EIGEN_STRONG_INLINE Packet8f pcast<Packet8i, Packet8f>(const Packet8i& a) {
return _mm256_cvtepi32_ps(a);
}
template<> EIGEN_STRONG_INLINE Packet8f pcast<Packet4d, Packet8f>(const Packet4d& a, const Packet4d& b) {
return _mm256_set_m128(_mm256_cvtpd_ps(b), _mm256_cvtpd_ps(a));
}
template <> EIGEN_STRONG_INLINE Packet4f pcast<Packet4d, Packet4f>(const Packet4d& a) {
return _mm256_cvtpd_ps(a);
}
template <> EIGEN_STRONG_INLINE Packet4d pcast<Packet8i, Packet4d>(const Packet8i& a) {
return _mm256_cvtepi32_pd(_mm256_castsi256_si128(a));
}
template <> EIGEN_STRONG_INLINE Packet4d pcast<Packet4i, Packet4d>(const Packet4i& a) {
return _mm256_cvtepi32_pd(a);
}
template <> EIGEN_STRONG_INLINE Packet4d pcast<Packet8f, Packet4d>(const Packet8f& a) {
return _mm256_cvtps_pd(_mm256_castps256_ps128(a));
}
template <> EIGEN_STRONG_INLINE Packet4d pcast<Packet4f, Packet4d>(const Packet4f& a) {
return _mm256_cvtps_pd(a);
}
template<> EIGEN_STRONG_INLINE Packet8i preinterpret<Packet8i,Packet8f>(const Packet8f& a) { template<> EIGEN_STRONG_INLINE Packet8i preinterpret<Packet8i,Packet8f>(const Packet8f& a) {
return _mm256_castps_si256(a); return _mm256_castps_si256(a);
} }

View File

@ -16,23 +16,23 @@ namespace Eigen {
namespace internal { namespace internal {
template <> template<> struct type_casting_traits<float, bool> : vectorized_type_casting_traits<float, bool> {};
struct type_casting_traits<float, bool> { template<> struct type_casting_traits<bool, float> : vectorized_type_casting_traits<bool, float> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <> template<> struct type_casting_traits<float, int> : vectorized_type_casting_traits<float, int> {};
struct type_casting_traits<bool, float> { template<> struct type_casting_traits<int, float> : vectorized_type_casting_traits<int, float> {};
enum {
VectorizedCast = 1, template<> struct type_casting_traits<float, double> : vectorized_type_casting_traits<float, double> {};
SrcCoeffRatio = 1, template<> struct type_casting_traits<double, float> : vectorized_type_casting_traits<double, float> {};
TgtCoeffRatio = 1
}; template<> struct type_casting_traits<double, int> : vectorized_type_casting_traits<double, int> {};
}; template<> struct type_casting_traits<int, double> : vectorized_type_casting_traits<int, double> {};
template<> struct type_casting_traits<half, float> : vectorized_type_casting_traits<half, float> {};
template<> struct type_casting_traits<float, half> : vectorized_type_casting_traits<float, half> {};
template<> struct type_casting_traits<bfloat16, float> : vectorized_type_casting_traits<bfloat16, float> {};
template<> struct type_casting_traits<float, bfloat16> : vectorized_type_casting_traits<float, bfloat16> {};
template<> EIGEN_STRONG_INLINE Packet16b pcast<Packet16f, Packet16b>(const Packet16f& a) { template<> EIGEN_STRONG_INLINE Packet16b pcast<Packet16f, Packet16b>(const Packet16f& a) {
__mmask16 mask = _mm512_cmpneq_ps_mask(a, pzero(a)); __mmask16 mask = _mm512_cmpneq_ps_mask(a, pzero(a));
@ -47,10 +47,26 @@ template<> EIGEN_STRONG_INLINE Packet16i pcast<Packet16f, Packet16i>(const Packe
return _mm512_cvttps_epi32(a); return _mm512_cvttps_epi32(a);
} }
template<> EIGEN_STRONG_INLINE Packet8d pcast<Packet16f, Packet8d>(const Packet16f& a) {
return _mm512_cvtps_pd(_mm512_castps512_ps256(a));
}
template<> EIGEN_STRONG_INLINE Packet8d pcast<Packet8f, Packet8d>(const Packet8f& a) {
return _mm512_cvtps_pd(a);
}
template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16i, Packet16f>(const Packet16i& a) { template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16i, Packet16f>(const Packet16i& a) {
return _mm512_cvtepi32_ps(a); return _mm512_cvtepi32_ps(a);
} }
template<> EIGEN_STRONG_INLINE Packet8d pcast<Packet16i, Packet8d>(const Packet16i& a) {
return _mm512_cvtepi32_pd(_mm512_castsi512_si256(a));
}
template<> EIGEN_STRONG_INLINE Packet8d pcast<Packet8i, Packet8d>(const Packet8i& a) {
return _mm512_cvtepi32_pd(a);
}
template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet8d, Packet16f>(const Packet8d& a, const Packet8d& b) { template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet8d, Packet16f>(const Packet8d& a, const Packet8d& b) {
return cat256(_mm512_cvtpd_ps(a), _mm512_cvtpd_ps(b)); return cat256(_mm512_cvtpd_ps(a), _mm512_cvtpd_ps(b));
} }
@ -131,80 +147,26 @@ template<> EIGEN_STRONG_INLINE Packet8bf preinterpret<Packet8bf, Packet16bf>(con
#ifndef EIGEN_VECTORIZE_AVX512FP16 #ifndef EIGEN_VECTORIZE_AVX512FP16
template <>
struct type_casting_traits<half, float> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16h, Packet16f>(const Packet16h& a) { template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16h, Packet16f>(const Packet16h& a) {
return half2float(a); return half2float(a);
} }
template <>
struct type_casting_traits<float, half> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template<> EIGEN_STRONG_INLINE Packet16h pcast<Packet16f, Packet16h>(const Packet16f& a) { template<> EIGEN_STRONG_INLINE Packet16h pcast<Packet16f, Packet16h>(const Packet16f& a) {
return float2half(a); return float2half(a);
} }
#endif #endif
template <>
struct type_casting_traits<bfloat16, float> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16bf, Packet16f>(const Packet16bf& a) { template<> EIGEN_STRONG_INLINE Packet16f pcast<Packet16bf, Packet16f>(const Packet16bf& a) {
return Bf16ToF32(a); return Bf16ToF32(a);
} }
template <>
struct type_casting_traits<float, bfloat16> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template<> EIGEN_STRONG_INLINE Packet16bf pcast<Packet16f, Packet16bf>(const Packet16f& a) { template<> EIGEN_STRONG_INLINE Packet16bf pcast<Packet16f, Packet16bf>(const Packet16f& a) {
return F32ToBf16(a); return F32ToBf16(a);
} }
#ifdef EIGEN_VECTORIZE_AVX512FP16 #ifdef EIGEN_VECTORIZE_AVX512FP16
template <>
struct type_casting_traits<half, float> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 2
};
};
template <>
struct type_casting_traits<float, half> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 2,
TgtCoeffRatio = 1
};
};
template<> EIGEN_STRONG_INLINE Packet16h preinterpret<Packet16h, Packet32h>(const Packet32h& a) { template<> EIGEN_STRONG_INLINE Packet16h preinterpret<Packet16h, Packet32h>(const Packet32h& a) {
return _mm256_castpd_si256(_mm512_extractf64x4_pd(_mm512_castph_pd(a), 0)); return _mm256_castpd_si256(_mm512_extractf64x4_pd(_mm512_castph_pd(a), 0));
} }
@ -257,7 +219,7 @@ EIGEN_STRONG_INLINE Packet8h pcast<Packet4f, Packet8h>(const Packet4f& a, const
__m256 result = _mm256_undefined_ps(); __m256 result = _mm256_undefined_ps();
result = _mm256_insertf128_ps(result, a, 0); result = _mm256_insertf128_ps(result, a, 0);
result = _mm256_insertf128_ps(result, b, 1); result = _mm256_insertf128_ps(result, b, 1);
return _mm256_cvtps_ph(result, _MM_FROUND_TO_NEAREST_INT|_MM_FROUND_NO_EXC); return _mm256_cvtps_ph(result, _MM_FROUND_TO_NEAREST_INT);
} }

View File

@ -17,61 +17,19 @@ namespace Eigen {
namespace internal { namespace internal {
#ifndef EIGEN_VECTORIZE_AVX #ifndef EIGEN_VECTORIZE_AVX
template <> template<> struct type_casting_traits<float, bool> : vectorized_type_casting_traits<float, bool> {};
struct type_casting_traits<float, bool> { template<> struct type_casting_traits<bool, float> : vectorized_type_casting_traits<bool, float> {};
enum {
VectorizedCast = 1,
SrcCoeffRatio = 4,
TgtCoeffRatio = 1
};
};
template <> template<> struct type_casting_traits<float, int> : vectorized_type_casting_traits<float, int> {};
struct type_casting_traits<float, double> { template<> struct type_casting_traits<int, float> : vectorized_type_casting_traits<int, float> {};
enum {
VectorizedCast = 1, template<> struct type_casting_traits<float, double> : vectorized_type_casting_traits<float, double> {};
SrcCoeffRatio = 1, template<> struct type_casting_traits<double, float> : vectorized_type_casting_traits<double, float> {};
TgtCoeffRatio = 2
}; template<> struct type_casting_traits<double, int> : vectorized_type_casting_traits<double, int> {};
}; template<> struct type_casting_traits<int, double> : vectorized_type_casting_traits<int, double> {};
#endif #endif
template <>
struct type_casting_traits<int, float> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <>
struct type_casting_traits<float, int> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 1,
TgtCoeffRatio = 1
};
};
template <>
struct type_casting_traits<double, int> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 2,
TgtCoeffRatio = 1
};
};
template <>
struct type_casting_traits<double, float> {
enum {
VectorizedCast = 1,
SrcCoeffRatio = 2,
TgtCoeffRatio = 1
};
};
template <> template <>
EIGEN_STRONG_INLINE Packet16b pcast<Packet4f, Packet16b>(const Packet4f& a, EIGEN_STRONG_INLINE Packet16b pcast<Packet4f, Packet16b>(const Packet4f& a,
const Packet4f& b, const Packet4f& b,
@ -88,10 +46,31 @@ EIGEN_STRONG_INLINE Packet16b pcast<Packet4f, Packet16b>(const Packet4f& a,
return _mm_and_si128(merged, _mm_set1_epi8(1)); return _mm_and_si128(merged, _mm_set1_epi8(1));
} }
template <>
EIGEN_STRONG_INLINE Packet4f pcast<Packet16b, Packet4f>(const Packet16b& a) {
const __m128 cst_one = _mm_set_ps1(1.0f);
#ifdef EIGEN_VECTORIZE_SSE4_1
__m128i a_extended = _mm_cvtepi8_epi32(a);
__m128i abcd = _mm_cmpeq_epi32(a_extended, _mm_setzero_si128());
#else
__m128i abcd_efhg_ijkl_mnop = _mm_cmpeq_epi8(a, _mm_setzero_si128());
__m128i aabb_ccdd_eeff_gghh = _mm_unpacklo_epi8(abcd_efhg_ijkl_mnop, abcd_efhg_ijkl_mnop);
__m128i abcd = _mm_unpacklo_epi8(aabb_ccdd_eeff_gghh, aabb_ccdd_eeff_gghh);
#endif
__m128 result = _mm_andnot_ps(_mm_castsi128_ps(abcd), cst_one);
return result;
}
template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f, Packet4i>(const Packet4f& a) { template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f, Packet4i>(const Packet4f& a) {
return _mm_cvttps_epi32(a); return _mm_cvttps_epi32(a);
} }
template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet2d, Packet4i>(const Packet2d& a, const Packet2d& b) {
return _mm_castps_si128(_mm_shuffle_ps(_mm_castsi128_ps(_mm_cvttpd_epi32(a)),
_mm_castsi128_ps(_mm_cvttpd_epi32(b)),
(1 << 2) | (1 << 6)));
}
template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i, Packet4f>(const Packet4i& a) { template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i, Packet4f>(const Packet4i& a) {
return _mm_cvtepi32_ps(a); return _mm_cvtepi32_ps(a);
} }
@ -100,10 +79,9 @@ template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet2d, Packet4f>(const Packet2d
return _mm_shuffle_ps(_mm_cvtpd_ps(a), _mm_cvtpd_ps(b), (1 << 2) | (1 << 6)); return _mm_shuffle_ps(_mm_cvtpd_ps(a), _mm_cvtpd_ps(b), (1 << 2) | (1 << 6));
} }
template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet2d, Packet4i>(const Packet2d& a, const Packet2d& b) { template<> EIGEN_STRONG_INLINE Packet2d pcast<Packet4i, Packet2d>(const Packet4i& a) {
return _mm_castps_si128(_mm_shuffle_ps(_mm_castsi128_ps(_mm_cvttpd_epi32(a)), // Simply discard the second half of the input
_mm_castsi128_ps(_mm_cvttpd_epi32(b)), return _mm_cvtepi32_pd(a);
(1 << 2) | (1 << 6)));
} }
template<> EIGEN_STRONG_INLINE Packet2d pcast<Packet4f, Packet2d>(const Packet4f& a) { template<> EIGEN_STRONG_INLINE Packet2d pcast<Packet4f, Packet2d>(const Packet4f& a) {

View File

@ -1211,6 +1211,22 @@ void typed_logicals_test(const ArrayType& m) {
typed_logicals_test_impl<ArrayType>::run(m); typed_logicals_test_impl<ArrayType>::run(m);
} }
// print non-mangled typenames
template<typename T> std::string printTypeInfo(const T&) { return typeid(T).name(); }
template<> std::string printTypeInfo(const int8_t&) { return "int8_t"; }
template<> std::string printTypeInfo(const int16_t&) { return "int16_t"; }
template<> std::string printTypeInfo(const int32_t&) { return "int32_t"; }
template<> std::string printTypeInfo(const int64_t&) { return "int64_t"; }
template<> std::string printTypeInfo(const uint8_t&) { return "uint8_t"; }
template<> std::string printTypeInfo(const uint16_t&) { return "uint16_t"; }
template<> std::string printTypeInfo(const uint32_t&) { return "uint32_t"; }
template<> std::string printTypeInfo(const uint64_t&) { return "uint64_t"; }
template<> std::string printTypeInfo(const float&) { return "float"; }
template<> std::string printTypeInfo(const double&) { return "double"; }
//template<> std::string printTypeInfo(const long double&) { return "long double"; }
template<> std::string printTypeInfo(const half&) { return "half"; }
template<> std::string printTypeInfo(const bfloat16&) { return "bfloat16"; }
template <typename SrcType, typename DstType, int RowsAtCompileTime, int ColsAtCompileTime> template <typename SrcType, typename DstType, int RowsAtCompileTime, int ColsAtCompileTime>
struct cast_test_impl { struct cast_test_impl {
using SrcArray = Array<SrcType, RowsAtCompileTime, ColsAtCompileTime>; using SrcArray = Array<SrcType, RowsAtCompileTime, ColsAtCompileTime>;
@ -1225,63 +1241,30 @@ struct cast_test_impl {
static constexpr int DstPacketSize = internal::packet_traits<DstType>::size; static constexpr int DstPacketSize = internal::packet_traits<DstType>::size;
static constexpr int MaxPacketSize = internal::plain_enum_max(SrcPacketSize, DstPacketSize); static constexpr int MaxPacketSize = internal::plain_enum_max(SrcPacketSize, DstPacketSize);
// print non-mangled typenames
template <typename T>
static std::string printTypeInfo(const T&) {
if (internal::is_same<bool, T>::value)
return "bool";
else if (internal::is_same<int8_t, T>::value)
return "int8_t";
else if (internal::is_same<int16_t, T>::value)
return "int16_t";
else if (internal::is_same<int32_t, T>::value)
return "int32_t";
else if (internal::is_same<int64_t, T>::value)
return "int64_t";
else if (internal::is_same<uint8_t, T>::value)
return "uint8_t";
else if (internal::is_same<uint16_t, T>::value)
return "uint16_t";
else if (internal::is_same<uint32_t, T>::value)
return "uint32_t";
else if (internal::is_same<uint64_t, T>::value)
return "uint64_t";
else if (internal::is_same<float, T>::value)
return "float";
else if (internal::is_same<double, T>::value)
return "double";
//else if (internal::is_same<long double, T>::value)
// return "long double";
else if (internal::is_same<half, T>::value)
return "half";
else if (internal::is_same<bfloat16, T>::value)
return "bfloat16";
else
return typeid(T).name();
}
static void run() { static void run() {
const Index testRows = RowsAtCompileTime == Dynamic ? ((10 * MaxPacketSize) + 1) : RowsAtCompileTime; const Index testRows = RowsAtCompileTime == Dynamic ? ((10 * MaxPacketSize) + 1) : RowsAtCompileTime;
const Index testCols = ColsAtCompileTime == Dynamic ? ((10 * MaxPacketSize) + 1) : ColsAtCompileTime; const Index testCols = ColsAtCompileTime == Dynamic ? ((10 * MaxPacketSize) + 1) : ColsAtCompileTime;
const Index testSize = testRows * testCols; const Index testSize = testRows * testCols;
const Index minTestSize = 100; const Index minTestSize = 100;
const Index repeats = numext::div_ceil(minTestSize, testSize); const Index repeats = numext::div_ceil(minTestSize, testSize);
SrcArray src(testRows, testCols); SrcArray src(testRows, testCols);
DstArray dst(testRows, testCols); DstArray dst(testRows, testCols);
for (Index repeat = 0; repeat < repeats; repeat++) { for (Index repeat = 0; repeat < repeats; repeat++) {
src = src.unaryExpr(RandomOp()); src = src.unaryExpr(RandomOp());
dst = src.template cast<DstType>(); dst = src.template cast<DstType>();
for (Index i = 0; i < testRows; i++)
for (Index j = 0; j < testCols; j++) { for (Index j = 0; j < testCols; j++)
DstType ref = internal::cast_impl<SrcType, DstType>::run(src(i, j)); for (Index i = 0; i < testRows; i++) {
bool all_nan = ((numext::isnan)(src(i, j)) && (numext::isnan)(ref) && (numext::isnan)(dst(i, j))); SrcType srcVal = src(i, j);
bool is_equal = ref == dst(i, j); DstType refVal = internal::cast_impl<SrcType, DstType>::run(srcVal);
bool pass = all_nan || is_equal; DstType dstVal = dst(i, j);
if (!pass) { bool isApprox = verifyIsApprox(dstVal, refVal);
std::cout << printTypeInfo(SrcType()) << ": [" << +src(i, j) << "] to " << printTypeInfo(DstType()) << ": [" if (!isApprox)
<< +dst(i, j) << "] != [" << +ref << "]\n"; std::cout << printTypeInfo(srcVal) << ": [" << +srcVal << "] to " << printTypeInfo(dstVal) << ": ["
} << +dstVal << "] != [" << +refVal << "]\n";
VERIFY(pass); VERIFY(isApprox);
} }
} }
} }