mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-05-18 23:57:39 +08:00
Added more missing EIGEN_DEVICE_FUNC
This commit is contained in:
parent
d27e4f1cba
commit
33cbdc2d15
@ -52,6 +52,7 @@ struct nested<TensorConversionOp<TargetType, XprType>, 1, typename eval<TensorCo
|
|||||||
|
|
||||||
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
|
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
|
||||||
struct PacketConverter {
|
struct PacketConverter {
|
||||||
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
||||||
PacketConverter(const TensorEvaluator& impl)
|
PacketConverter(const TensorEvaluator& impl)
|
||||||
: m_impl(impl) {}
|
: m_impl(impl) {}
|
||||||
|
|
||||||
@ -67,6 +68,7 @@ struct PacketConverter {
|
|||||||
|
|
||||||
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
|
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
|
||||||
struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 2, 1> {
|
struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 2, 1> {
|
||||||
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
||||||
PacketConverter(const TensorEvaluator& impl)
|
PacketConverter(const TensorEvaluator& impl)
|
||||||
: m_impl(impl) {}
|
: m_impl(impl) {}
|
||||||
|
|
||||||
@ -87,6 +89,7 @@ struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 2, 1> {
|
|||||||
|
|
||||||
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
|
template <typename TensorEvaluator, typename SrcPacket, typename TgtPacket>
|
||||||
struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 1, 2> {
|
struct PacketConverter<TensorEvaluator, SrcPacket, TgtPacket, 1, 2> {
|
||||||
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
||||||
PacketConverter(const TensorEvaluator& impl)
|
PacketConverter(const TensorEvaluator& impl)
|
||||||
: m_impl(impl), m_maxIndex(impl.dimensions().TotalSize()) {}
|
: m_impl(impl), m_maxIndex(impl.dimensions().TotalSize()) {}
|
||||||
|
|
||||||
|
@ -23,6 +23,7 @@ namespace Eigen {
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
template <typename Index, std::size_t Rank> struct DimensionList {
|
template <typename Index, std::size_t Rank> struct DimensionList {
|
||||||
|
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
|
||||||
const Index operator[] (const Index i) const { return i; }
|
const Index operator[] (const Index i) const { return i; }
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -96,6 +96,7 @@ template <typename T> struct MeanReducer
|
|||||||
static const bool PacketAccess = true;
|
static const bool PacketAccess = true;
|
||||||
static const bool IsStateful = true;
|
static const bool IsStateful = true;
|
||||||
|
|
||||||
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
|
||||||
MeanReducer() : scalarCount_(0), packetCount_(0) { }
|
MeanReducer() : scalarCount_(0), packetCount_(0) { }
|
||||||
|
|
||||||
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) {
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user