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,

View File

@@ -49,6 +49,7 @@ template<typename UnaryOp, typename ArgType>
class unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>::InnerIterator
: public unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>::EvalIterator
{
protected:
typedef typename XprType::Scalar Scalar;
typedef typename unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>::EvalIterator Base;
public:
@@ -99,6 +100,7 @@ template<typename ViewOp, typename ArgType>
class unary_evaluator<CwiseUnaryView<ViewOp,ArgType>, IteratorBased>::InnerIterator
: public unary_evaluator<CwiseUnaryView<ViewOp,ArgType>, IteratorBased>::EvalIterator
{
protected:
typedef typename XprType::Scalar Scalar;
typedef typename unary_evaluator<CwiseUnaryView<ViewOp,ArgType>, IteratorBased>::EvalIterator Base;
public:

View File

@@ -90,6 +90,7 @@ struct unary_evaluator<SparseView<ArgType>, IteratorBased>
class InnerIterator : public EvalIterator
{
protected:
typedef typename XprType::Scalar Scalar;
public:

View File

@@ -608,16 +608,18 @@ erfc() const
return ErfcReturnType(derived());
}
/** \cpp11 \returns an expression of the coefficient-wise Complementary error
/** \returns an expression of the coefficient-wise inverse of the CDF of the Normal distribution function
* function of *this.
*
* \specialfunctions_module
*
* In other words, considering `x = ndtri(y)`, it returns the argument, x, for which the area under the
* Gaussian probability density function (integrated from minus infinity to x) is equal to y.
*
* \note This function supports only float and double scalar types in c++11 mode. To support other scalar types,
* or float/double in non c++11 mode, the user has to provide implementations of ndtri(T) for any scalar
* type T to be supported.
* \note This function supports only float and double scalar types. To support other scalar types,
* the user has to provide implementations of ndtri(T) for any scalar type T to be supported.
*
* \sa <a href="group__CoeffwiseMathFunctions.html#cwisetable_ndtri">Math functions</a>, erf()
* \sa <a href="group__CoeffwiseMathFunctions.html#cwisetable_ndtri">Math functions</a>
*/
EIGEN_DEVICE_FUNC
inline const NdtriReturnType