GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-10 10:49:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable vectorized version on GPUs. The underlying bug has been fixed.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2018-09-13 16:12:22 -07:00
parent e289f44c56
commit 2b07018140
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h
View File

@ -218,7 +218,6 @@ struct InnerMostDimReducer<Self, Op, false, true> {
} }
}; };
#if !defined(EIGEN_USE_GPU) || !defined(__CUDACC__)
template <typename Self, typename Op> template <typename Self, typename Op>
struct InnerMostDimReducer<Self, Op, true, true> { struct InnerMostDimReducer<Self, Op, true, true> {
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename Self::CoeffReturnType static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename Self::CoeffReturnType
@ -258,7 +257,6 @@ struct InnerMostDimReducer<Self, Op, true, true> {
} }
} }
}; };
#endif
template <int DimIndex, typename Self, typename Op, bool vectorizable = (Self::InputPacketAccess && Self::ReducerTraits::PacketAccess)> template <int DimIndex, typename Self, typename Op, bool vectorizable = (Self::InputPacketAccess && Self::ReducerTraits::PacketAccess)>
struct InnerMostDimPreserver { struct InnerMostDimPreserver {