Follow comments

8 years ago · 94b58a29d6
parent d8560ec2e8
commit 94b58a29d6
1 changed files with 1 additions and 2 deletions
--- a/paddle/memory/memory.cc
+++ b/paddle/memory/memory.cc
@ -30,8 +30,7 @@ std::once_flag cpu_allocator_flag;
 std::once_flag gpu_allocator_flag;

 BuddyAllocator* GetCPUBuddyAllocator() {
-  static std::unique_ptr<BuddyAllocator, void (*)(BuddyAllocator*)> a{
-      nullptr, [](BuddyAllocator* p) { delete p; }};
+  static std::unique_ptr<BuddyAllocator> a{nullptr};

  std::call_once(cpu_allocator_flag, [&]() {
    a.reset(new BuddyAllocator(new detail::CPUAllocator,