This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
hummingbird
/
graphengine
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
zero copy optimize the struct of addr offset
tags/v1.1.0
y00500818
4 years ago
parent
ca56699d74
commit
47d52f0d5c
2 changed files
with
2 additions
and
2 deletions
Unified View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
ge/graph/load/new_model_manager/zero_copy_task.cc
+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()) {
if (it == task_addr_offset_.end()) {
task_addr_offset_[addr] = {offset};
task_addr_offset_[addr] = {offset};
} else {
} 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,
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_;
bool is_updated_;
string batch_label_;
string batch_label_;
// <address from Op, {offset in args}>
// <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
} // namespace ge
#endif // GE_GRAPH_LOAD_NEW_MODEL_MANAGER_ZERO_COPY_TASK_H_
#endif // GE_GRAPH_LOAD_NEW_MODEL_MANAGER_ZERO_COPY_TASK_H_
Write
Preview
Loading…
Cancel
Save