diff --git a/paddle/memory/CMakeLists.txt b/paddle/memory/CMakeLists.txt
index 4cd40e2c2ac0c32aaedeec570cb28451ad7adab8..3e53d1ce6532c8154da34fcdfc8af5b867456b60 100644
--- a/paddle/memory/CMakeLists.txt
+++ b/paddle/memory/CMakeLists.txt
@@ -8,7 +8,10 @@ cc_library(paddle_memory
     memory
     memcpy
     meta_data
-    meta_cache memory_block
-    buddy_allocator system_allocator)
+    meta_cache
+    memory_block
+    buddy_allocator
+    system_allocator
+    device_context)
 
 cc_test(memory_test SRCS memory_test.cc DEPS place paddle_memory)