Browse Source

!296 zero_copy_addr_offset

From: @yangyongqiang5033
Reviewed-by: @zhangxiaokun9,@xchu42,@wqtshg
Signed-off-by: @wqtshg
tags/v1.1.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
88f5c5d1e4
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/load/new_model_manager/zero_copy_task.cc
  2. +1
    -1
      ge/graph/load/new_model_manager/zero_copy_task.h

+ 1
- 1
ge/graph/load/new_model_manager/zero_copy_task.cc View File

@@ -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,


+ 1
- 1
ge/graph/load/new_model_manager/zero_copy_task.h View File

@@ -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_

Loading…
Cancel
Save