From eb102189b2c504ea8270ab7aa564d4032664886c Mon Sep 17 00:00:00 2001 From: wanghuancoder Date: Mon, 7 Nov 2022 14:46:43 +0800 Subject: [PATCH] refine python lib link (#47681) * refine python lib link --- paddle/fluid/eager/CMakeLists.txt | 4 ++++ .../api/generated/eager_generated/backwards/CMakeLists.txt | 1 + .../api/generated/eager_generated/forwards/CMakeLists.txt | 1 + .../eager/auto_code_generator/generate_file_structures.py | 6 ++++-- 4 files changed, 10 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/eager/CMakeLists.txt b/paddle/fluid/eager/CMakeLists.txt index bd005fc41d4..d5d5f1daa1e 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 69bfe4d9415..fc52d6f82a1 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 c32dd2f122a..97d5aed2947 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 5555c17f1b6..bb2a236c28e 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)") -- GitLab