mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-27 07:29:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Simplified and generalized the DividerTraits code

Browse Source
This commit is contained in:
Benoit Steiner 2015-07-29 10:02:42 -07:00
parent 6db3a557f4
commit 099597406f
1 changed files with 1 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
unsupported/Eigen/CXX11/src/Tensor/TensorIntDiv.h
View File

@ -57,32 +57,13 @@ namespace {
template <typename T> template <typename T>
struct DividerTraits { struct DividerTraits {
#if defined(__SIZEOF_INT128__) && !defined(__CUDACC__) #if defined(__SIZEOF_INT128__) && !defined(__CUDACC__)
typedef T type; typedef typename conditional<sizeof(T) == 8, uint64_t, uint32_t>::type type;
static const int N = sizeof(T) * 8; static const int N = sizeof(T) * 8;
#else #else
typedef uint32_t type; typedef uint32_t type;
static const int N = 32; static const int N = 32;
#endif #endif
}; };
template <>
struct DividerTraits<int32_t> {
typedef uint32_t type;
static const int N = 32;
};
template <>
struct DividerTraits<int64_t> {
#if defined(__SIZEOF_INT128__) && !defined(__CUDACC__)
typedef uint64_t type;
static const int N = 64;
#else
typedef uint32_t type;
static const int N = 32;
#endif
};
// In C++, long is not int32_t nor int64_t, so we need a specialization for it:
template <>
struct DividerTraits<long> : internal::conditional<sizeof(long)==4,DividerTraits<int32_t>,DividerTraits<int64_t> >::type
{};
template <typename T> template <typename T>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE uint32_t muluh(const uint32_t a, const T b) { EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE uint32_t muluh(const uint32_t a, const T b) {