!296 zero_copy_addr_offset

From: @yangyongqiang5033
Reviewed-by: @zhangxiaokun9,@xchu42,@wqtshg
Signed-off-by: @wqtshg
pull/296/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 88f5c5d1e4

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

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