fix parallel compilation error of allocator (#19581)

6 years ago · 578cccd48c
parent 635cd62d23
commit 578cccd48c
1 changed files with 3 additions and 3 deletions
--- a/paddle/fluid/memory/detail/CMakeLists.txt
+++ b/paddle/fluid/memory/detail/CMakeLists.txt
@ -1,9 +1,9 @@
-cc_library(memory_block SRCS memory_block.cc memory_block_desc.cc meta_cache.cc)
+cc_library(memory_block SRCS memory_block.cc memory_block_desc.cc meta_cache.cc DEPS place)

 if(${WITH_GPU})
-  nv_library(system_allocator SRCS system_allocator.cc DEPS gflags cpu_info gpu_info)
+  nv_library(system_allocator SRCS system_allocator.cc DEPS gflags cpu_info gpu_info place)
 else(${WITH_GPU})
-  cc_library(system_allocator SRCS system_allocator.cc DEPS gflags cpu_info)
+  cc_library(system_allocator SRCS system_allocator.cc DEPS gflags cpu_info place)
 endif(${WITH_GPU})

 cc_test(system_allocator_test SRCS system_allocator_test.cc DEPS system_allocator)