mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-01-24 14:45:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix CUDA build broken by changes to min and max reduction.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2016-09-02 13:41:36 -07:00
parent ef54723dbe
commit 13e93ca8b7
1 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
View File

@ -188,6 +188,32 @@ struct reducer_traits<MeanReducer<T>, Device> {
}; };
template <typename T, bool IsMax = true, bool IsInteger = true>
struct MinMaxBottomValue {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE static T bottom_value() {
return Eigen::NumTraits<T>::lowest();
}
};
template <typename T>
struct MinMaxBottomValue<T, true, false> {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE static T bottom_value() {
return -Eigen::NumTraits<T>::infinity();
}
};
template <typename T>
struct MinMaxBottomValue<T, false, true> {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE static T bottom_value() {
return Eigen::NumTraits<T>::highest();
}
};
template <typename T>
struct MinMaxBottomValue<T, false, false> {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE static T bottom_value() {
return Eigen::NumTraits<T>::infinity();
}
};
template <typename T> struct MaxReducer template <typename T> struct MaxReducer
{ {
static const bool PacketAccess = packet_traits<T>::HasMax; static const bool PacketAccess = packet_traits<T>::HasMax;
@ -201,11 +227,7 @@ template <typename T> struct MaxReducer
(*accum) = pmax<Packet>(*accum, p); (*accum) = pmax<Packet>(*accum, p);
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const {
if (Eigen::NumTraits<T>::IsInteger) { return MinMaxBottomValue<T, true, Eigen::NumTraits<T>::IsInteger>::bottom_value();
return Eigen::NumTraits<T>::lowest();
} else {
return -Eigen::NumTraits<T>::infinity();
}
} }
template <typename Packet> template <typename Packet>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet initializePacket() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet initializePacket() const {
@ -246,11 +268,7 @@ template <typename T> struct MinReducer
(*accum) = pmin<Packet>(*accum, p); (*accum) = pmin<Packet>(*accum, p);
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const {
if (Eigen::NumTraits<T>::IsInteger) { return MinMaxBottomValue<T, false, Eigen::NumTraits<T>::IsInteger>::bottom_value();
return Eigen::NumTraits<T>::highest();
} else {
return Eigen::NumTraits<T>::infinity();
}
} }
template <typename Packet> template <typename Packet>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet initializePacket() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet initializePacket() const {