CPUGenerator.cpp

#include <ATen/CPUGenerator.h>

#define const_generator_cast(generator) \
  dynamic_cast<const CPUGenerator&>(generator)

namespace at {

CPUGenerator::CPUGenerator(Context * context_)
  : context(context_), generator(THGenerator_new())
{}

CPUGenerator::~CPUGenerator() {
  if (generator)
    THGenerator_free(generator);
}

CPUGenerator& CPUGenerator::copy(const Generator& from) {
  THGenerator_copy(generator, const_generator_cast(from).generator);
  return *this;
}

CPUGenerator& CPUGenerator::free() {
  THGenerator_free(generator);
  return *this;
}

uint64_t CPUGenerator::seed() {
  return THRandom_seed(generator);
}

uint64_t CPUGenerator::initialSeed() {
  return THRandom_initialSeed(generator);
}

CPUGenerator& CPUGenerator::manualSeed(uint64_t seed) {
  THRandom_manualSeed(generator, seed);
  return *this;
}

CPUGenerator& CPUGenerator::manualSeedAll(uint64_t seed) {
  // There's only one CPU generator
  return manualSeed(seed);
}

void * CPUGenerator::unsafeGetTH() {
  return generator;
}

} // namespace at