!678 p2p offset set for zero

From: @ni100die Reviewed-by: @wqtshg,@ji_chen Signed-off-by: @ji_chen
4 years ago · b682ee8def
parent 08cd18bd63 77ab669191
commit b682ee8def
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/build/memory/graph_mem_assigner.cc
+++ b/ge/graph/build/memory/graph_mem_assigner.cc
@ -99,7 +99,7 @@ Status GraphMemoryAssigner::AssignMemory() {
  MemoryOffset memory_offset(RT_MEMORY_HBM, mem_assigner->GetMemOffset());
  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());
    memory_offset_.emplace(RT_MEMORY_P2P_DDR, p2p_memory_offset);
  }