From d65cb13ad591a529f32b63a8023082a5cd1891fe Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Mon, 4 Feb 2019 10:02:17 +0800 Subject: [PATCH] add pslib flag on fleet_wrapper CMakefile --- paddle/fluid/framework/fleet/CMakeLists.txt | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/fleet/CMakeLists.txt b/paddle/fluid/framework/fleet/CMakeLists.txt index 58881b80c..7a3812bd5 100644 --- 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) -- GitLab