Function DefaultGenerator

oneflow/core/framework/random_generator.cpp:128–134

}

Maybe<Generator> DefaultGenerator(const std::string& device, int device_index) {
  static auto* default_auto_generator =
      dynamic_cast<AutoGenerator*>(JUST(DefaultAutoGenerator())->internal().get());
  if (device_index == -1) { device_index = (device == "cpu" ? 0 : GlobalProcessCtx::LocalRank()); }
  return std::make_shared<Generator>(
      JUST(default_auto_generator->GetOrCreate(device, device_index)));
}

Maybe<Generator> DefaultAutoGenerator() {
  // Skip destructing to avoid calling symbols in other dynamic libraries when the global object is