fix

PaddlePaddle · May 11, 2023 · 39926ba · 39926ba
1 parent 25521de
commit 39926ba
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/paddle/fluid/memory/allocation/mmap_allocator.cc b/paddle/fluid/memory/allocation/mmap_allocator.cc
@@ -146,7 +146,11 @@ std::shared_ptr<MemoryMapAllocation> AllocateMemoryMapAllocationAndUnlink(
  std::string handle = memory::allocation::GetIPCName();
  AllocateMemoryMap(handle, flags, size, &ptr, &fd);
  } else {
- AllocateMemoryMap("", flags, size, &ptr, &fd);
+ ptr = mmap(nullptr, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
+ PADDLE_ENFORCE_NE(ptr,
+ MAP_FAILED,
+ platform::errors::Unavailable(
+ "Memory map failed when create shared memory."));
  }
  // 构造1个shmem的wapper
  return std::make_shared<MemoryMapAllocation>(ptr, size, "", flags, fd);