Function MakeGenerator

oneflow/core/framework/random_generator.cpp:196–209

}

Maybe<Generator> MakeGenerator(const std::string& device, int device_index) {
  if (device == "auto") {
    return std::make_shared<Generator>(std::make_shared<AutoGenerator>(default_rng_seed_val));
  }
  auto device_type = ep::DeviceManagerRegistry::GetDeviceTypeByDeviceTypeName(device);
  if (device_type == DeviceType::kInvalidDevice) {
    return Error::RuntimeError() << "Expected one of " << PrintGeneratorAvailableDevices()
                                 << " device type at start of device string: " << device;
  }
  auto device_mgr = Singleton<ep::DeviceManagerRegistry>::Get()->GetDeviceManager(device_type);
  if (device_index == -1) { device_index = (device == "cpu" ? 0 : GlobalProcessCtx::LocalRank()); }
  return std::make_shared<Generator>(
      device_mgr->CreateRandomGenerator(default_rng_seed_val, device_index));
}

Maybe<Generator> DefaultGenerator(DeviceType device, int device_index) {
  return DefaultGenerator(*JUST(DeviceTag4DeviceType(device)), device_index);