Method AddChunkProto

oneflow/core/memory/chunk_manager.cpp:39–49

}

void ChunkMgr::AddChunkProto(const ChunkProto& chunk) {
  std::unique_lock<std::mutex> guard(mutex_);
  const int64_t mem_zone_uid = memory::GetUniqueMemCaseId(chunk.machine_id(), chunk.mem_case());
  CHECK(
      chunk_id2chunk_proto_.emplace(chunk.chunk_id(), std::make_unique<ChunkProto>(chunk)).second);
  auto chunk_ids_it = mzuid2chunk_ids_.find(mem_zone_uid);
  if (chunk_ids_it == mzuid2chunk_ids_.end()) {
    chunk_ids_it = mzuid2chunk_ids_.emplace(mem_zone_uid, HashSet<int64_t>()).first;
  }
  CHECK(chunk_ids_it->second.insert(chunk.chunk_id()).second);
}

char* ChunkMgr::FindOrCreateChunk(const ChunkProto& chunk) {
  std::unique_lock<std::mutex> guard(mutex_);