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

Better rand to fix MSVC random tests

Browse Source
This commit is contained in:
Antonio Sanchez 2025-03-13 08:45:50 -07:00
parent 414c42bfcf
commit 3b8644da50
2 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
Eigen/src/Core/MathFunctions.h
View File

@ -624,19 +624,6 @@ struct random_retval
template<typename Scalar> inline EIGEN_MATHFUNC_RETVAL(random, Scalar) random(const Scalar& x, const Scalar& y); template<typename Scalar> inline EIGEN_MATHFUNC_RETVAL(random, Scalar) random(const Scalar& x, const Scalar& y);
template<typename Scalar> inline EIGEN_MATHFUNC_RETVAL(random, Scalar) random(); template<typename Scalar> inline EIGEN_MATHFUNC_RETVAL(random, Scalar) random();
template<typename Scalar>
struct random_default_impl<Scalar, false, false>
{
static inline Scalar run(const Scalar& x, const Scalar& y)
{
return x + (y-x) * Scalar(std::rand()) / Scalar(RAND_MAX);
}
static inline Scalar run()
{
return run(Scalar(NumTraits<Scalar>::IsSigned ? -1 : 0), Scalar(1));
}
};
enum { enum {
meta_floor_log2_terminate, meta_floor_log2_terminate,
meta_floor_log2_move_up, meta_floor_log2_move_up,
@ -684,6 +671,38 @@ struct meta_floor_log2<n, lower, upper, meta_floor_log2_bogus>
// no value, error at compile time // no value, error at compile time
}; };
#define EIGEN_RAND_MAX INT_MAX
// Fill a signed positive int with random bits.
// This is to overcome issues in MSVC which limits RAND_MAX to 32767.
inline int random_int() {
#if RAND_MAX == INT_MAX
return std::rand();
#else
enum {
rand_bits = meta_floor_log2<(unsigned int)(RAND_MAX)+1>::value,
int_bits = meta_floor_log2<(unsigned int)(INT_MAX)+1>::value,
};
unsigned int out = std::rand();
for (int bit = rand_bits; bit < int(int_bits); bit += rand_bits) {
out = (out << rand_bits) ^ std::rand();
}
return static_cast<int>(out & INT_MAX);
#endif
}
template<typename Scalar>
struct random_default_impl<Scalar, false, false>
{
static inline Scalar run(const Scalar& x, const Scalar& y)
{
return x + (y-x) * Scalar(random_int()) / Scalar(EIGEN_RAND_MAX);
}
static inline Scalar run()
{
return run(Scalar(NumTraits<Scalar>::IsSigned ? -1 : 0), Scalar(1));
}
};
template<typename Scalar> template<typename Scalar>
struct random_default_impl<Scalar, false, true> struct random_default_impl<Scalar, false, true>
{ {
@ -704,12 +723,12 @@ struct random_default_impl<Scalar, false, true>
ScalarX offset = 0; ScalarX offset = 0;
ScalarX divisor = 1; ScalarX divisor = 1;
ScalarX multiplier = 1; ScalarX multiplier = 1;
const unsigned rand_max = RAND_MAX; const unsigned rand_max = EIGEN_RAND_MAX;
if (range <= rand_max) divisor = (rand_max + 1) / (range + 1); if (range <= rand_max) divisor = (rand_max + 1) / (range + 1);
else multiplier = 1 + range / (rand_max + 1); else multiplier = 1 + range / (rand_max + 1);
// Rejection sampling. // Rejection sampling.
do { do {
offset = (unsigned(std::rand()) * multiplier) / divisor; offset = (unsigned(random_int()) * multiplier) / divisor;
} while (offset > range); } while (offset > range);
return Scalar(ScalarX(x) + offset); return Scalar(ScalarX(x) + offset);
} }
@ -719,12 +738,12 @@ struct random_default_impl<Scalar, false, true>
#ifdef EIGEN_MAKING_DOCS #ifdef EIGEN_MAKING_DOCS
return run(Scalar(NumTraits<Scalar>::IsSigned ? -10 : 0), Scalar(10)); return run(Scalar(NumTraits<Scalar>::IsSigned ? -10 : 0), Scalar(10));
#else #else
enum { rand_bits = meta_floor_log2<(unsigned int)(RAND_MAX)+1>::value, enum { rand_bits = meta_floor_log2<(unsigned int)(EIGEN_RAND_MAX)+1>::value,
scalar_bits = sizeof(Scalar) * CHAR_BIT, scalar_bits = sizeof(Scalar) * CHAR_BIT,
shift = EIGEN_PLAIN_ENUM_MAX(0, int(rand_bits) - int(scalar_bits)), shift = EIGEN_PLAIN_ENUM_MAX(0, int(rand_bits) - int(scalar_bits)),
offset = NumTraits<Scalar>::IsSigned ? (1 << (EIGEN_PLAIN_ENUM_MIN(rand_bits,scalar_bits)-1)) : 0 offset = NumTraits<Scalar>::IsSigned ? (1 << (EIGEN_PLAIN_ENUM_MIN(rand_bits,scalar_bits)-1)) : 0
}; };
return Scalar((std::rand() >> shift) - offset); return Scalar((random_int() >> shift) - offset);
#endif #endif
} }
}; };

2
Eigen/src/Core/arch/CUDA/Half.h
View File

@ -541,7 +541,7 @@ struct random_default_impl<half, false, false>
{ {
static inline half run(const half& x, const half& y) static inline half run(const half& x, const half& y)
{ {
return x + (y-x) * half(float(std::rand()) / float(RAND_MAX)); return x + (y-x) * half(random<float>());
} }
static inline half run() static inline half run()
{ {