diff --git a/paddle/memory/memory.cc b/paddle/memory/memory.cc
index c99cc541566d233579982dba6a8a985ff46860d9..0266bf4f7d65c7aafd4242af41cbd1c71f44bff8 100644
--- a/paddle/memory/memory.cc
+++ b/paddle/memory/memory.cc
@@ -16,6 +16,7 @@ limitations under the License. */
 
 #include <algorithm>  // for transform
 #include <cstring>    // for memcpy
+#include <memory>     // for unique_ptr
 #include <mutex>      // for call_once
 
 #include "paddle/memory/detail/buddy_allocator.h"