diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index cdf917a1e99b2ce068d0ff1ca1f39ccb112c87f0..779f62711542493dbfe47a86cab9e6ea9b9068a3 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -332,13 +332,13 @@ function(go_test TARGET_NAME)
   add_test(${TARGET_NAME} ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME})
 endfunction(go_test)
 
-function(pb_cc_library TARGET_NAME)
+function(proto_library TARGET_NAME)
   set(oneValueArgs "")
   set(multiValueArgs SRCS)
-  cmake_parse_arguments(pb_cc_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
+  cmake_parse_arguments(proto_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
   set(proto_srcs)
   set(proto_hdrs)
-  protobuf_generate_cpp(proto_srcs proto_hdrs ${pb_cc_library_SRCS})
+  protobuf_generate_cpp(proto_srcs proto_hdrs ${proto_library_SRCS})
   include_directories(${CMAKE_CURRENT_BINARY_DIR})
   cc_library(${TARGET_NAME} SRCS ${proto_srcs})
 endfunction()
diff --git a/proto/CMakeLists.txt b/proto/CMakeLists.txt
index 1cf39d69443cfd8006385d9d8f90ae029cabdbd4..436bea53e539ffffdc2010d0e0ec9798b0aa03f7 100644
--- a/proto/CMakeLists.txt
+++ b/proto/CMakeLists.txt
@@ -1,5 +1,5 @@
 file(GLOB proto_filenames . *.proto)
-pb_cc_library(paddle_proto SRCS ${proto_filenames})
+proto_library(paddle_proto SRCS ${proto_filenames})
 
 set(PROTO_GEN)
 set(PROTO_GEN_PY)