Merge pull request #3018 from wangkuiyi/fix_memory_deps_on_device_context

Fix memory deps on device context
8 years ago · d1dbf2cf5c
parent a16cd515b4 de6f9c4879
commit d1dbf2cf5c
1 changed files with 5 additions and 3 deletions
--- a/paddle/memory/CMakeLists.txt
+++ b/paddle/memory/CMakeLists.txt
@ -1,14 +1,16 @@
 add_subdirectory(detail)

 cc_library(memory SRCS memory.cc)
-cc_library(memcpy SRCS memcpy.cc)
+cc_library(memcpy SRCS memcpy.cc DEPS device_context)

 cc_library(paddle_memory
    DEPS
    memory
    memcpy
    meta_data
-    meta_cache memory_block
-    buddy_allocator system_allocator)
+    meta_cache
+    memory_block
+    buddy_allocator
+    system_allocator)

 cc_test(memory_test SRCS memory_test.cc DEPS place paddle_memory)