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();
return true;
}
EIGEN_STRONG_INLINE void cleanup() {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void cleanup() {
m_device.deallocate(m_buffer);
m_buffer = NULL;
}
@ -132,7 +132,7 @@ struct TensorEvaluator<const TensorForcedEvalOp<ArgType>, Device>
}
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);
}