Method GetOrCreate

oneflow/core/framework/auto_random_generator.cpp:150–166

}

Maybe<ep::RandomGenerator> AutoGenerator::GetOrCreate(const std::string& device, int device_index) {
  if (device_index == -1) { device_index = (device == "cpu" ? 0 : GlobalProcessCtx::LocalRank()); }
  std::lock_guard<std::mutex> lock(mutex_);
  auto device_key = JUST(Device::New(device, device_index));
  auto it = generators_.find(device_key);
  if (it == generators_.end()) {
    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);
    it = generators_.emplace(device_key, device_mgr->CreateRandomGenerator(seed_, device_index))
             .first;
  }
  return it->second;
}

}  // namespace one
}  // namespace oneflow