From c38295f0a03edcd9f8325fcb08484eb579b7841f Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Mon, 28 Mar 2016 15:53:02 -0700
Subject: Added support for fmod

---
 unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h')

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
-- 
cgit v1.2.3