Added a couple of missing EIGEN_DEVICE_FUNC

This commit is contained in:
Benoit Steiner 2015-11-12 20:01:50 -08:00
parent 0aaa5941df
commit b69248fa2a

View File

@ -121,7 +121,7 @@ struct TensorEvaluator<const TensorForcedEvalOp<ArgType>, Device>
m_impl.cleanup(); m_impl.cleanup();
return true; return true;
} }
EIGEN_STRONG_INLINE void cleanup() { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void cleanup() {
m_device.deallocate(m_buffer); m_device.deallocate(m_buffer);
m_buffer = NULL; m_buffer = NULL;
} }
@ -132,7 +132,7 @@ struct TensorEvaluator<const TensorForcedEvalOp<ArgType>, Device>
} }
template<int LoadMode> template<int LoadMode>
EIGEN_STRONG_INLINE PacketReturnType packet(Index index) const EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType packet(Index index) const
{ {
return internal::ploadt<Packet, LoadMode>(m_buffer + index); return internal::ploadt<Packet, LoadMode>(m_buffer + index);
} }