Browse Source

p2p offset set for zero

tags/v1.2.0
gengchao4@huawei.com 3 years ago
parent
commit
77ab669191
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/memory/graph_mem_assigner.cc

+ 1
- 1
ge/graph/build/memory/graph_mem_assigner.cc 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);
} }


Loading…
Cancel
Save