devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't vectorize the MeanReducer unless pdiv is available.

Browse Source
This commit is contained in:
Rasmus Munk Larsen
2018-09-11 14:09:00 -07:00
parent 5da960702f
commit e289f44c56
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -158,7 +158,8 @@ template <typename T, typename Device>
struct reducer_traits<MeanReducer<T>, Device> { struct reducer_traits<MeanReducer<T>, Device> {
enum { enum {
Cost = NumTraits<T>::AddCost, Cost = NumTraits<T>::AddCost,
PacketAccess = PacketType<T, Device>::HasAdd && !NumTraits<T>::IsInteger, PacketAccess = PacketType<T, Device>::HasAdd &&
PacketType<T, Device>::HasDiv && !NumTraits<T>::IsInteger,
IsStateful = true, IsStateful = true,
IsExactlyAssociative = NumTraits<T>::IsInteger IsExactlyAssociative = NumTraits<T>::IsInteger
}; };