diff --git a/paddle/fluid/eager/CMakeLists.txt b/paddle/fluid/eager/CMakeLists.txt index bd005fc41d42ec1baba278be0708758e012869ca..d5d5f1daa1e5f7ddaf1c4dd371ba8bddf5add995 100644 --- a/paddle/fluid/eager/CMakeLists.txt +++ b/paddle/fluid/eager/CMakeLists.txt @@ -78,6 +78,10 @@ cc_library( autograd_meta hook_utils) +if(NOT ((NOT WITH_PYTHON) AND ON_INFER)) + target_link_libraries(utils ${PYTHON_LIBRARIES}) +endif() + if(NOT ((NOT WITH_PYTHON) AND ON_INFER)) cc_library( saved_tensors_hooks 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 69bfe4d9415722c48dae2f31d375836ba70b695f..fc52d6f82a174c7d1efcc1dbefa445db7cfe975f 100644 --- a/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt +++ b/paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt @@ -9,4 +9,5 @@ if(NOT (NOT WITH_PYTHON AND ON_INFER)) SRCS nodes.cc ${eager_manual_nodes} DEPS ${eager_deps}) add_dependencies(final_dygraph_node eager_codegen) + target_link_libraries(final_dygraph_node ${PYTHON_LIBRARIES}) 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 c32dd2f122a6e2cac8af7ea97d15f43c3b3342c4..97d5aed29476092b054e7ebf49713bef1b9f3e26 100644 --- a/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt +++ b/paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt @@ -9,4 +9,5 @@ if(NOT (NOT WITH_PYTHON AND ON_INFER)) SRCS dygraph_functions.cc ${eager_manual_functions} DEPS ${eager_deps}) add_dependencies(final_dygraph_function eager_codegen) + target_link_libraries(final_dygraph_function ${PYTHON_LIBRARIES}) endif() diff --git a/paddle/fluid/eager/auto_code_generator/generate_file_structures.py b/paddle/fluid/eager/auto_code_generator/generate_file_structures.py index 5555c17f1b6c61d0ce122ebdd648d1da9d3524a2..bb2a236c28e592c796b3c6ba95543ad5437bd999 100644 --- a/paddle/fluid/eager/auto_code_generator/generate_file_structures.py +++ b/paddle/fluid/eager/auto_code_generator/generate_file_structures.py @@ -143,7 +143,8 @@ def GenerateFileStructureForIntermediateDygraph(eager_dir, split_count): for i in range(split_count): f.write("nodes" + str(i + 1) + ".cc ") f.write("${fluid_manual_nodes} DEPS ${eager_deps} ${fluid_deps})\n") - f.write("add_dependencies(dygraph_node copy_dygraph_node)") + f.write("add_dependencies(dygraph_node copy_dygraph_node)\n") + f.write("target_link_libraries(dygraph_node ${PYTHON_LIBRARIES})\n") with open(forwards_level_cmakelist_path, "w") as f: f.write("add_custom_target(\n") @@ -181,8 +182,9 @@ def GenerateFileStructureForIntermediateDygraph(eager_dir, split_count): "${fluid_manual_functions} DEPS ${eager_deps} ${fluid_deps} ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})\n" ) f.write( - "add_dependencies(dygraph_function copy_dygraph_forward_functions)" + "add_dependencies(dygraph_function copy_dygraph_forward_functions)\n" ) + f.write("target_link_libraries(dygraph_function ${PYTHON_LIBRARIES})\n") with open(generated_level_cmakelist_path, "w") as f: f.write("add_subdirectory(forwards)\nadd_subdirectory(nodes)")