Enabled final state Eager Dygraph Codegen

jim19930609 · Feb 7, 2022 · 025135d · 025135d
1 parent a821c4a
commit 025135d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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)