diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index 779f62711542493dbfe47a86cab9e6ea9b9068a3..8117dbc53e4597c8b0ab3558e6eb1afc352823d5 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -87,6 +87,9 @@
 #   go_library(example SHARED)
 #
 
+# including binary directory for generated headers.
+include_directories(${CMAKE_CURRENT_BINARY_DIR})
+
 if(NOT APPLE)
     find_package(Threads REQUIRED)
     link_libraries(${CMAKE_THREAD_LIBS_INIT})
@@ -339,6 +342,5 @@ function(proto_library TARGET_NAME)
   set(proto_srcs)
   set(proto_hdrs)
   protobuf_generate_cpp(proto_srcs proto_hdrs ${proto_library_SRCS})
-  include_directories(${CMAKE_CURRENT_BINARY_DIR})
-  cc_library(${TARGET_NAME} SRCS ${proto_srcs})
+  cc_library(${TARGET_NAME} SRCS ${proto_srcs} DEPS protobuf)
 endfunction()
diff --git a/proto/CMakeLists.txt b/proto/CMakeLists.txt
index 70dd4d674cdd7967e997e10e98477e565f98b8f5..18584cafe7971bad281b498908c54780250791b7 100644
--- a/proto/CMakeLists.txt
+++ b/proto/CMakeLists.txt
@@ -1,4 +1,5 @@
 file(GLOB proto_filenames . *.proto)
+include_directories(${CMAKE_CURRENT_BINARY_DIR})
 proto_library(paddle_proto SRCS ${proto_filenames})
 
 set(PROTO_GEN)