Method New

oneflow/core/framework/device.cpp:61–74

static */

}

/* static */ Maybe<Symbol<Device>> Device::New(const std::string& type, int64_t device_id,
                                               bool rematable) {
  CHECK_GE_OR_RETURN(device_id, 0)
      << Error::InvalidValueError() << "Device ID should be non-negative";
  static thread_local HashMap<std::tuple<std::string, int, bool>, Symbol<Device>> map;
  auto key = std::make_tuple(type, device_id, rematable);
  auto iter = map.find(key);
  if (iter == map.end()) {
    Device device(type, device_id, rematable);
    JUST(device.Init());
    iter = map.emplace(key, SymbolOf(device)).first;
  }
  return iter->second;
}

/* static */ Maybe<Symbol<Device>> Device::New(const std::string& type, int64_t device_id) {
  return New(type, device_id, false);