diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt
index ef7f6e1c..bd9edd86 100755
--- a/ge/CMakeLists.txt
+++ b/ge/CMakeLists.txt
@@ -66,6 +66,11 @@ target_compile_definitions(ge_proto_client PRIVATE
     google=ascend_private
 )
 
+target_include_directories(ge_proto_client PRIVATE
+    ${CMAKE_BINARY_DIR}/proto/ge_client
+    ${CMAKE_BINARY_DIR}/proto/ge_client/proto
+)
+
 target_compile_options(ge_proto_client PRIVATE
     -O2
     -fno-common