diff options
author | Eugene Brevdo <ebrevdo@gmail.com> | 2015-12-07 15:24:49 -0800 |
---|---|---|
committer | Eugene Brevdo <ebrevdo@gmail.com> | 2015-12-07 15:24:49 -0800 |
commit | fa4f933c0fe65eda6a051f978db12210f11f5cdb (patch) | |
tree | cab17ee4bbffd52a778b00ec40770e7fa4b361cf /test | |
parent | 7dfe75f445835baff18bbe82ba7253f7563cbdc6 (diff) |
Add special functions to Eigen: lgamma, erf, erfc.
Includes CUDA support and unit tests.
Diffstat (limited to 'test')
-rw-r--r-- | test/array.cpp | 3 | ||||
-rw-r--r-- | test/packetmath.cpp | 23 |
2 files changed, 26 insertions, 0 deletions
diff --git a/test/array.cpp b/test/array.cpp index 5395721f5..9994c23c3 100644 --- a/test/array.cpp +++ b/test/array.cpp @@ -217,6 +217,9 @@ template<typename ArrayType> void array_real(const ArrayType& m) VERIFY_IS_APPROX(m1.sinh(), sinh(m1)); VERIFY_IS_APPROX(m1.cosh(), cosh(m1)); VERIFY_IS_APPROX(m1.tanh(), tanh(m1)); + VERIFY_IS_APPROX(m1.lgamma(), lgamma(m1)); + VERIFY_IS_APPROX(m1.erf(), erf(m1)); + VERIFY_IS_APPROX(m1.erfc(), erfc(m1)); VERIFY_IS_APPROX(m1.arg(), arg(m1)); VERIFY_IS_APPROX(m1.round(), round(m1)); VERIFY_IS_APPROX(m1.floor(), floor(m1)); diff --git a/test/packetmath.cpp b/test/packetmath.cpp index b6616ac5e..304fab5de 100644 --- a/test/packetmath.cpp +++ b/test/packetmath.cpp @@ -351,6 +351,25 @@ template<typename Scalar> void packetmath_real() VERIFY_IS_EQUAL(std::exp(-std::numeric_limits<Scalar>::denorm_min()), data2[1]); } + { + data1[0] = std::numeric_limits<Scalar>::quiet_NaN(); + packet_helper<internal::packet_traits<Scalar>::HasLGamma,Packet> h; + h.store(data2, internal::plgamma(h.load(data1))); + VERIFY(std::isnan(data2[0])); + } + { + data1[0] = std::numeric_limits<Scalar>::quiet_NaN(); + packet_helper<internal::packet_traits<Scalar>::HasErf,Packet> h; + h.store(data2, internal::perf(h.load(data1))); + VERIFY(std::isnan(data2[0])); + } + { + data1[0] = std::numeric_limits<Scalar>::quiet_NaN(); + packet_helper<internal::packet_traits<Scalar>::HasErfc,Packet> h; + h.store(data2, internal::perfc(h.load(data1))); + VERIFY(std::isnan(data2[0])); + } + for (int i=0; i<size; ++i) { data1[i] = internal::random<Scalar>(0,1) * std::pow(Scalar(10), internal::random<Scalar>(-6,6)); @@ -360,6 +379,10 @@ template<typename Scalar> void packetmath_real() data1[internal::random<int>(0, PacketSize)] = 0; CHECK_CWISE1_IF(PacketTraits::HasSqrt, std::sqrt, internal::psqrt); CHECK_CWISE1_IF(PacketTraits::HasLog, std::log, internal::plog); + CHECK_CWISE1_IF(internal::packet_traits<Scalar>::HasLGamma, std::lgamma, internal::plgamma); + CHECK_CWISE1_IF(internal::packet_traits<Scalar>::HasErf, std::erf, internal::perf); + CHECK_CWISE1_IF(internal::packet_traits<Scalar>::HasErfc, std::erfc, internal::perfc); + if(PacketTraits::HasLog && PacketTraits::size>=2) { data1[0] = std::numeric_limits<Scalar>::quiet_NaN(); |