diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-03-28 15:53:02 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-03-28 15:53:02 -0700 |
commit | c38295f0a03edcd9f8325fcb08484eb579b7841f (patch) | |
tree | aed42b619de1e8719cce1f3f3dc83ed1fdbdeafb /unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | |
parent | 6772f653c33bd78c25623619581836bac1d1d20a (diff) |
Added support for fmod
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h index eb0c8d1ce..b7c13f67f 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h @@ -40,6 +40,20 @@ template <typename Scalar> struct functor_traits<scalar_mod2_op<Scalar> > { enum { Cost = NumTraits<Scalar>::template Div<false>::Cost, PacketAccess = false }; }; +template <typename Scalar> +struct scalar_fmod_op { + EIGEN_EMPTY_STRUCT_CTOR(scalar_fmod_op); + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar + operator()(const Scalar& a, const Scalar& b) const { + return numext::fmod(a, b); + } +}; +template <typename Scalar> +struct functor_traits<scalar_fmod_op<Scalar> > { + enum { Cost = 13, // Reciprocal throughput of FPREM on Haswell. + PacketAccess = false }; +}; + /** \internal * \brief Template functor to compute the sigmoid of a scalar |