Method GetDevice

oneflow/core/hardware/node_device_descriptor.cpp:276–284

}

std::shared_ptr<const DeviceDescriptor> NodeDeviceDescriptor::GetDevice(
    const std::string& class_name, size_t ordinal) const {
  const auto device_list = GetDeviceDescriptorList(class_name);
  if (device_list) {
    return device_list->GetDevice(ordinal);
  } else {
    return nullptr;
  }
}

size_t NodeDeviceDescriptor::HostMemorySizeBytes() const { return impl_->host_memory_size_bytes; }