diff --git a/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt b/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt
index e04d282748c0a6061ccb7d9429f0066c7fabf3ca..e3fafb265ad9887a5683542d79ae07f30edee910 100644
--- a/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt
+++ b/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt
@@ -1,3 +1,6 @@
 cc_library(scale_node SRCS scale_node.cc DEPS global_utils pten pten_api grad_node_info)
-#cc_library(final_dygraph_node SRCS nodes.cc DEPS ${eager_deps})
-#add_dependencies(final_dygraph_node eager_final_state_codegen)
+
+if(NOT ON_INFER)
+cc_library(final_dygraph_node SRCS nodes.cc DEPS ${eager_deps})
+add_dependencies(final_dygraph_node eager_final_state_codegen)
+endif()
diff --git a/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt b/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt
index f682c27992db15e81f28afe0bb9c3b30454a9d88..8ede139ddc0446ddab4404ae2f749a3c84748d73 100644
--- a/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt
+++ b/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt
@@ -1,3 +1,6 @@
 cc_library(eager_scale SRCS scale.cc DEPS pten_api pten autograd_meta scale_node)
-#cc_library(final_dygraph_function SRCS dygraph_functions.cc DEPS ${eager_deps})
-#add_dependencies(final_dygraph_function eager_final_state_codegen)
+
+if(NOT ON_INFER)
+cc_library(final_dygraph_function SRCS dygraph_functions.cc DEPS ${eager_deps})
+add_dependencies(final_dygraph_function eager_final_state_codegen)
+endif()
diff --git a/paddle/fluid/eager/auto_code_generator/CMakeLists.txt b/paddle/fluid/eager/auto_code_generator/CMakeLists.txt
index c504a126ddecaebfcb55313573d6bc490007feef..668e60d857b9ca371243891db686421810fda0bb 100644
--- a/paddle/fluid/eager/auto_code_generator/CMakeLists.txt
+++ b/paddle/fluid/eager/auto_code_generator/CMakeLists.txt
@@ -1,4 +1,4 @@
-#add_subdirectory(final_state_generator)
+add_subdirectory(final_state_generator)
 
 set(EAGER_GENERETOR_DEPS ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} pybind proto_desc executor layer tracer engine imperative_profiler imperative_flag)