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

Change runtime to compile-time conditional.

Browse Source
This commit is contained in:
Gael Guennebaud 2016-07-08 11:39:43 +02:00
parent 544935101a
commit 599f8ba617
1 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,25 +29,37 @@ namespace Eigen {
namespace internal { namespace internal {
namespace { namespace {
// Note: result is undefined if val == 0 // Note: result is undefined if val == 0
template <typename T> template <typename T>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE int count_leading_zeros(const T val) EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
typename internal::enable_if<sizeof(T)==4,int>::type count_leading_zeros(const T val)
{ {
#ifdef __CUDA_ARCH__ #ifdef __CUDA_ARCH__
return (sizeof(T) == 8) ? __clzll(val) : __clz(val); return __clz(val);
#elif EIGEN_COMP_MSVC #elif EIGEN_COMP_MSVC
unsigned long index; unsigned long index;
if (sizeof(T) == 8) { _BitScanReverse(&index, val);
_BitScanReverse64(&index, val); return 31 - index;
} else {
_BitScanReverse(&index, val);
}
return (sizeof(T) == 8) ? 63 - index : 31 - index;
#else #else
EIGEN_STATIC_ASSERT(sizeof(unsigned long long) == 8, YOU_MADE_A_PROGRAMMING_MISTAKE); EIGEN_STATIC_ASSERT(sizeof(unsigned long long) == 8, YOU_MADE_A_PROGRAMMING_MISTAKE);
return (sizeof(T) == 8) ? return __builtin_clz(static_cast<uint32_t>(val));
__builtin_clzll(static_cast<uint64_t>(val)) : #endif
__builtin_clz(static_cast<uint32_t>(val)); }
template <typename T>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
typename internal::enable_if<sizeof(T)==8,int>::type count_leading_zeros(const T val)
{
#ifdef __CUDA_ARCH__
return __clzll(val);
#elif EIGEN_COMP_MSVC
unsigned long index;
_BitScanReverse64(&index, val);
return 63 - index;
#else
EIGEN_STATIC_ASSERT(sizeof(unsigned long long) == 8, YOU_MADE_A_PROGRAMMING_MISTAKE);
return __builtin_clzll(static_cast<uint64_t>(val));
#endif #endif
} }