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 (#10794)

Browse Source
This commit is contained in:
Maksim Kutakov
2022-03-05 12:31:57 +03:00
committed by GitHub
parent 3c75a4fd16
commit cdd4f56ba1
1 changed files with 7 additions and 1 deletions
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);
}