chunk id add lock

Oneflow-Inc · Oct 8, 2022 · 2adbb13 · lixinqi · Oct 8, 2022 · lixinqi
1 parent 8ca22bf
commit 2adbb13
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/oneflow/core/memory/chunk_manager.cpp b/oneflow/core/memory/chunk_manager.cpp
@@ -36,6 +36,7 @@ void ChunkMgr::GetChunkProtosByMemZoneUniqueId(int64_t mem_zone_uid,
 }
 
 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);