add pslib flag on fleet_wrapper CMakefile

7 years ago · d65cb13ad5
parent 6de9ebc65c
commit d65cb13ad5
1 changed files with 5 additions and 1 deletions
--- a/paddle/fluid/framework/fleet/CMakeLists.txt
+++ b/paddle/fluid/framework/fleet/CMakeLists.txt
@ -1 +1,5 @@
-cc_library(fleet_wrapper SRCS fleet_wrapper.cc DEPS pslib_brpc pslib)
+if(WITH_PSLIB)
+    cc_library(fleet_wrapper SRCS fleet_wrapper.cc DEPS pslib_brpc pslib)
+else()
+    cc_library(fleet_wrapper SRCS fleet_wrapper.cc)
+endif(WITH_PSLIB)