Function GetNodeCPUMemZoneId

oneflow/core/memory/memory_zone.cpp:35–37

const MemZoneId kInvalidMemZoneId = MemZoneId{0, DeviceType::kInvalidDevice, 0};

MemZoneId GetNodeCPUMemZoneId(MemZoneId::rank_t node_index) {
  return MemZoneId{node_index, DeviceType::kCPU, 0};
}

int64_t EncodeMemZoneIdToInt64(const MemZoneId& mem_zone_id) {
  int64_t id = static_cast<int64_t>(mem_zone_id.device_index());