diff --git a/paddle/fluid/platform/CMakeLists.txt b/paddle/fluid/platform/CMakeLists.txt
index 93f29b57c8650e40324a2ed38791fd07f48d860d..da2c51daa86987f3d610f2f6fcb75efe38a06619 100644
--- a/paddle/fluid/platform/CMakeLists.txt
+++ b/paddle/fluid/platform/CMakeLists.txt
@@ -75,7 +75,7 @@ cc_library(device_context SRCS device_context.cc init.cc DEPS simple_threadpool
     temp_allocator ${dgc_deps})
 
 if (WITH_DISTRIBUTE)
-  cc_library(collective_helper SRCS collective_helper.cc DEPS framework_proto)
+  cc_library(collective_helper SRCS collective_helper.cc DEPS framework_proto  device_context enforce)
 endif()
 
 if(WIN32)