m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

p2p offset set for zero

Browse Source
pull/678/head
gengchao4@huawei.com 4 years ago
parent c54d3ed3eb
commit 77ab669191
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
ge/graph/build/memory/graph_mem_assigner.cc
Unescape View File

@ -99,7 +99,7 @@ Status GraphMemoryAssigner::AssignMemory() {
MemoryOffset memory_offset(RT_MEMORY_HBM, mem_assigner->GetMemOffset()); MemoryOffset memory_offset(RT_MEMORY_HBM, mem_assigner->GetMemOffset());
memory_offset_.emplace(RT_MEMORY_HBM, memory_offset); memory_offset_.emplace(RT_MEMORY_HBM, memory_offset);
if (mem_assigner->GetP2PMemOffset() > 0) { if (mem_assigner->GetP2PMemOffset() >= 0) {
MemoryOffset p2p_memory_offset(RT_MEMORY_P2P_DDR, mem_assigner->GetP2PMemOffset()); MemoryOffset p2p_memory_offset(RT_MEMORY_P2P_DDR, mem_assigner->GetP2PMemOffset());
memory_offset_.emplace(RT_MEMORY_P2P_DDR, p2p_memory_offset); memory_offset_.emplace(RT_MEMORY_P2P_DDR, p2p_memory_offset);
} }

Write Preview
Loading…
Cancel
Save