Merged eigen/eigen

This commit is contained in:
Srinivas Vasudevan
2019-09-04 23:50:52 -04:00
13 changed files with 97 additions and 63 deletions

View File

@@ -276,12 +276,12 @@ pselect(const Packet& mask, const Packet& a, const Packet& b) {
template<> EIGEN_DEVICE_FUNC inline float pselect<float>(
const float& mask, const float& a, const float&b) {
return mask == 0 ? b : a;
return numext::equal_strict(mask,0.f) ? b : a;
}
template<> EIGEN_DEVICE_FUNC inline double pselect<double>(
const double& mask, const double& a, const double& b) {
return mask == 0 ? b : a;
return numext::equal_strict(mask,0.) ? b : a;
}
/** \internal \returns a <= b as a bit mask */

View File

@@ -95,8 +95,8 @@ template<> struct packet_traits<float> : default_packet_traits
#if EIGEN_GNUC_AT_LEAST(5, 3) || (!EIGEN_COMP_GNUC_STRICT)
#ifdef EIGEN_VECTORIZE_AVX512DQ
HasLog = 1,
HasLog1p = 1,
HasExpm1 = 1,
HasLog1p = 1,
HasExpm1 = 1,
HasNdtri = 1,
#endif
HasExp = 1,

View File

@@ -112,6 +112,7 @@ template<> struct packet_traits<float> : default_packet_traits
HasLog = 1,
HasLog1p = 1,
HasExpm1 = 1,
HasNdtri = 1,
HasExp = 1,
HasNdtri = 1,
HasSqrt = 1,