From 47d52f0d5c476d01c6e017df8b1b92b3dd9ceea7 Mon Sep 17 00:00:00 2001
From: y00500818 <yangyongqiang7@hisilicon.com>
Date: Sat, 14 Nov 2020 09:29:49 +0800
Subject: [PATCH] zero copy optimize the struct of addr offset

---
 ge/graph/load/new_model_manager/zero_copy_task.cc | 2 +-
 ge/graph/load/new_model_manager/zero_copy_task.h  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ge/graph/load/new_model_manager/zero_copy_task.cc b/ge/graph/load/new_model_manager/zero_copy_task.cc
index 35169726..2079034e 100755
--- a/ge/graph/load/new_model_manager/zero_copy_task.cc
+++ b/ge/graph/load/new_model_manager/zero_copy_task.cc
@@ -45,7 +45,7 @@ Status ZeroCopyTask::SetTaskArgsOffset(uintptr_t addr, size_t offset) {
   if (it == task_addr_offset_.end()) {
     task_addr_offset_[addr] = {offset};
   } else {
-    it->second.push_back(offset);
+    it->second.insert(offset);
   }
 
   GELOGI("[ZCPY] %s set task, virtual_addr: 0x%lx, args_addr: %p, size: %zu, offset: %zu", name_.c_str(), addr,
diff --git a/ge/graph/load/new_model_manager/zero_copy_task.h b/ge/graph/load/new_model_manager/zero_copy_task.h
index 57ccdbaf..d0bb2b6d 100644
--- a/ge/graph/load/new_model_manager/zero_copy_task.h
+++ b/ge/graph/load/new_model_manager/zero_copy_task.h
@@ -103,7 +103,7 @@ class ZeroCopyTask {
   bool is_updated_;
   string batch_label_;
   // <address from Op, {offset in args}>
-  map<uintptr_t, vector<size_t>> task_addr_offset_;
+  map<uintptr_t, set<size_t>> task_addr_offset_;
 };
 } // namespace ge
 #endif  // GE_GRAPH_LOAD_NEW_MODEL_MANAGER_ZERO_COPY_TASK_H_