26 #ifndef TVM_SUPPORT_RANDOM_ENGINE_H_ 27 #define TVM_SUPPORT_RANDOM_ENGINE_H_ 29 #include <tvm/runtime/logging.h> 94 (*rand_state_ptr_) = ((*rand_state_ptr_) * multiplier +
increment) % modulus;
95 return *rand_state_ptr_;
103 if (rand_state == -1) {
105 }
else if (rand_state == 0) {
108 ICHECK(rand_state >= 0) <<
"The random state should be nonnegative";
109 ICHECK(rand_state_ptr_ !=
nullptr);
110 *rand_state_ptr_ = rand_state %
modulus;
120 return ((*
this)() * 32767) % 1999999973;
131 rand_state_ptr_ = rand_state_ptr;
141 #endif // TVM_SUPPORT_RANDOM_ENGINE_H_ static constexpr TRandState modulus
The modulus.
Definition: random_engine.h:63
runtime implementation for LibTorch/TorchScript.
Definition: analyzer.h:36
void Seed(TRandState rand_state)
Change the start random state of RNG with the seed of a new random state value.
Definition: random_engine.h:102
static constexpr TRandState multiplier
The multiplier.
Definition: random_engine.h:57
int64_t TRandState
Definition: random_engine.h:54
LinearCongruentialEngine(TRandState *rand_state_ptr)
Construct a random number generator with a random state pointer.
Definition: random_engine.h:130
This linear congruential engine is a drop-in replacement for std::minstd_rand. It strictly correspond...
Definition: random_engine.h:47
uint64_t result_type
The result type is defined as uint64_t here to avoid overflow.
Definition: random_engine.h:53
TRandState ForkSeed()
Fork a new seed for another RNG from current random state.
Definition: random_engine.h:117
static constexpr TRandState increment
The increment.
Definition: random_engine.h:60
result_type operator()()
Operator to move the random state to the next and return the new random state. According to definitio...
Definition: random_engine.h:93
static constexpr result_type max()
The maximum possible value of random state here.
Definition: random_engine.h:75
static TRandState DeviceRandom()
Get a device random state.
Definition: random_engine.h:81
static constexpr result_type min()
The minimum possible value of random state here.
Definition: random_engine.h:69