diff --git a/paddle/fluid/framework/fleet/CMakeLists.txt b/paddle/fluid/framework/fleet/CMakeLists.txt
index 58881b80c7fdec994d72cbd7a8d3f078fbf52cea..7a3812bd58ae77bc6fe182d0d7c01f1b1dd26931 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)