diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index ca358da8f1482422f2811365beb7dcade45bef02..fb2222440ca59a17e51f49ab3729e17f014e5753 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -323,10 +323,10 @@ endfunction(go_test)
 
 function(proto_library TARGET_NAME)
   set(oneValueArgs "")
-  set(multiValueArgs SRCS)
+  set(multiValueArgs SRCS DEPS)
   cmake_parse_arguments(proto_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
   set(proto_srcs)
   set(proto_hdrs)
   protobuf_generate_cpp(proto_srcs proto_hdrs ${proto_library_SRCS})
-  cc_library(${TARGET_NAME} SRCS ${proto_srcs} DEPS protobuf)
+  cc_library(${TARGET_NAME} SRCS ${proto_srcs} DEPS ${proto_library_DEPS} protobuf)
 endfunction()
diff --git a/paddle/framework/CMakeLists.txt b/paddle/framework/CMakeLists.txt
index 8c34a77c20787e59376556c9d8f8ccbfa34d074b..1ef5f26addc2c539525b3745927903481dacb3ef 100644
--- a/paddle/framework/CMakeLists.txt
+++ b/paddle/framework/CMakeLists.txt
@@ -6,8 +6,7 @@ cc_test(variable_test SRCS variable_test.cc)
 cc_test(scope_test SRCS scope_test.cc)
 cc_test(enforce_test SRCS enforce_test.cc)
 proto_library(attr_type SRCS attr_type.proto)
-proto_library(op_proto SRCS op_proto.proto)
-
+proto_library(op_proto SRCS op_proto.proto DEPS attr_type)
 cc_test(op_proto_test SRCS op_proto_test.cc DEPS op_proto attr_type protobuf)
 
 proto_library(net_proto SRCS net_proto.proto)