IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] Use raw pointer to share peer data for constants (#10744)

Browse Source
This commit is contained in:
Maksim Kutakov 2022-03-05 12:32:11 +03:00 committed by GitHub
parent 0f8c599ce7
commit e87ea5d611
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/plugins/intel_cpu/src/edge.cpp
View File

@ -404,7 +404,13 @@ MKLDNNMemoryPtr &MKLDNNEdge::getMemoryPtr() {
if (status == Status::NotAllocated) {
memoryPtr.reset(new MKLDNNMemory(getParent()->getEngine()));
const auto &desc = getDesc();
memoryPtr->Create(desc, getSharedEdge()->getMemoryPtr()->getDnnlMemoryMngr());
auto sharedEdge = getSharedEdge();
auto sharedEdgeParent = sharedEdge->getParent();
if (sharedEdgeParent->isConstant()) {
memoryPtr->Create(desc, sharedEdge->getMemoryPtr()->GetData());
} else {
memoryPtr->Create(desc, sharedEdge->getMemoryPtr()->getDnnlMemoryMngr());
}
memoryFromEdge.reset();
changeStatus(Status::Allocated);
}