random.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_RANDOM_H
#define FREE_TENSOR_RANDOM_H
 
#include <random>
#include <vector>
 
namespace freetensor {
 
class OpenMPRandomEngine {
    typedef std::default_random_engine RNG;
 
    std::vector<RNG> rngs_; 
 
  public:
    typedef RNG::result_type
        result_type; 
 
    OpenMPRandomEngine(RNG::result_type seed);
 
    OpenMPRandomEngine(const OpenMPRandomEngine &) = delete;
    OpenMPRandomEngine &operator=(const OpenMPRandomEngine &) = delete;
    OpenMPRandomEngine(OpenMPRandomEngine &&) = default;
    OpenMPRandomEngine &operator=(OpenMPRandomEngine &&) = default;
 
    RNG::result_type operator()();
 
    static constexpr RNG::result_type min() { return RNG::min(); }
    static constexpr RNG::result_type max() { return RNG::max(); }
};
 
static_assert(std::uniform_random_bit_generator<OpenMPRandomEngine>);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_RANDOM_H