Made the index type an explicit template parameter to help some compilers compile the code.

This commit is contained in:
Benoit Steiner 2016-09-02 15:29:34 -07:00
parent 5a6be66cef
commit 05b0518077

View File

@ -243,13 +243,13 @@ struct TensorEvaluator<const TensorCwiseNullaryOp<NullaryOp, ArgType>, Device>
EIGEN_DEVICE_FUNC CoeffReturnType coeff(Index index) const EIGEN_DEVICE_FUNC CoeffReturnType coeff(Index index) const
{ {
return m_wrapper(m_functor,index); return m_wrapper(m_functor, index);
} }
template<int LoadMode> template<int LoadMode>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType packet(Index index) const EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType packet(Index index) const
{ {
return m_wrapper.template packetOp<PacketReturnType>(m_functor,index); return m_wrapper.template packetOp<PacketReturnType, Index>(m_functor, index);
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost