[libc++] Use correct rand.eng.mers all-zeroes seed sequence fallback

Summary:
When a seed sequence would lead to having no non-zero significant bits
in the initial state of a `mersenne_twister_engine`, the fallback is to
flip the most significant bit of the first value that appears in the
textual representation of the initial state.

rand.eng.mers describes this as setting the value to be 2 to the power
of one less than w; the previous value encoded in the implementation,
namely one less than "2 to the power of w", is replaced by the correct
value in this patch.

Reviewers: mclow.lists, EricWF, jasonliu

Reviewed By: mclow.lists

Subscribers: mclow.lists, jasonliu, EricWF, christof, ldionne, cfe-commits

Differential Revision: https://reviews.llvm.org/D50736

llvm-svn: 339969
This commit is contained in:
Hubert Tong 2018-08-16 23:56:54 +00:00
parent 9982bb8739
commit 2c0fad59df
2 changed files with 83 additions and 2 deletions

View File

@ -2337,7 +2337,7 @@ mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
for (size_t __i = 1; __i < __n; ++__i)
if (__x_[__i] != 0)
return;
__x_[0] = _Max;
__x_[0] = result_type(1) << (__w - 1);
}
}
@ -2363,7 +2363,7 @@ mersenne_twister_engine<_UIntType, __w, __n, __m, __r, __a, __u, __d, __s, __b,
for (size_t __i = 1; __i < __n; ++__i)
if (__x_[__i] != 0)
return;
__x_[0] = _Max;
__x_[0] = result_type(1) << (__w - 1);
}
}

View File

@ -0,0 +1,81 @@
//===----------------------------------------------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is dual licensed under the MIT and the University of Illinois Open
// Source Licenses. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
// <random>
// template <class UIntType, size_t w, size_t n, size_t m, size_t r,
// UIntType a, size_t u, UIntType d, size_t s,
// UIntType b, size_t t,
// UIntType c, size_t l, UIntType f>
// class mersenne_twister_engine;
// template <class Sseq> explicit mersenne_twister_engine(Sseq &q);
//
// [ ... ] Finally, if the most significant $w-r$ bits of $X_{-n}$ are zero,
// and if each of the other resulting $X_i$ is $0$, changes $X_{-n}$ to
// $ 2^{w-1} $.
#include <random>
#include <algorithm>
#include <cassert>
#include <cstddef>
#if TEST_STD_VER >= 11
#include <initializer_list>
#endif
struct all_zero_seed_seq {
typedef unsigned int result_type;
all_zero_seed_seq() {}
template <typename InputIterator>
all_zero_seed_seq(InputIterator, InputIterator) {}
#if TEST_STD_VER >= 11
all_zero_seed_seq(std::initializer_list<result_type>) {}
#endif
template <typename RandomAccessIterator>
void generate(RandomAccessIterator rb, RandomAccessIterator re) {
std::fill(rb, re, 0u);
}
std::size_t size() const { return 0u; }
template <typename OutputIterator> void param(OutputIterator) const {}
};
template <typename result_type, std::size_t word_size>
void test(void) {
const std::size_t state_size = 1u;
const std::size_t shift_size = 1u;
const std::size_t tempering_l = word_size;
all_zero_seed_seq q;
std::mersenne_twister_engine<result_type, word_size, state_size,
shift_size,
0u,
0x0,
0u, 0x0, 0u, 0x0, 0u, 0x0,
tempering_l,
0u>
e(q);
const result_type Xneg1 = result_type(1) << (word_size - 1);
const result_type Y = Xneg1;
const result_type X0 = Xneg1 ^ (Y >> 1);
assert(e() == X0);
}
int main() {
// Test for k == 1: word_size <= 32.
test<unsigned short, 3u>();
// Test for k == 2: (32 < word_size <= 64).
test<unsigned long long, 33u>();
}