Skip to content

Commit eb10218

Browse files
refine python lib link (#47681)
* refine python lib link
1 parent 624ffdf commit eb10218

File tree

4 files changed

+10
-2
lines changed

4 files changed

+10
-2
lines changed

paddle/fluid/eager/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,10 @@ cc_library(
7878
autograd_meta
7979
hook_utils)
8080

81+
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
82+
target_link_libraries(utils ${PYTHON_LIBRARIES})
83+
endif()
84+
8185
if(NOT ((NOT WITH_PYTHON) AND ON_INFER))
8286
cc_library(
8387
saved_tensors_hooks

paddle/fluid/eager/api/generated/eager_generated/backwards/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,5 @@ if(NOT (NOT WITH_PYTHON AND ON_INFER))
99
SRCS nodes.cc ${eager_manual_nodes}
1010
DEPS ${eager_deps})
1111
add_dependencies(final_dygraph_node eager_codegen)
12+
target_link_libraries(final_dygraph_node ${PYTHON_LIBRARIES})
1213
endif()

paddle/fluid/eager/api/generated/eager_generated/forwards/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,5 @@ if(NOT (NOT WITH_PYTHON AND ON_INFER))
99
SRCS dygraph_functions.cc ${eager_manual_functions}
1010
DEPS ${eager_deps})
1111
add_dependencies(final_dygraph_function eager_codegen)
12+
target_link_libraries(final_dygraph_function ${PYTHON_LIBRARIES})
1213
endif()

paddle/fluid/eager/auto_code_generator/generate_file_structures.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,8 @@ def GenerateFileStructureForIntermediateDygraph(eager_dir, split_count):
143143
for i in range(split_count):
144144
f.write("nodes" + str(i + 1) + ".cc ")
145145
f.write("${fluid_manual_nodes} DEPS ${eager_deps} ${fluid_deps})\n")
146-
f.write("add_dependencies(dygraph_node copy_dygraph_node)")
146+
f.write("add_dependencies(dygraph_node copy_dygraph_node)\n")
147+
f.write("target_link_libraries(dygraph_node ${PYTHON_LIBRARIES})\n")
147148

148149
with open(forwards_level_cmakelist_path, "w") as f:
149150
f.write("add_custom_target(\n")
@@ -181,8 +182,9 @@ def GenerateFileStructureForIntermediateDygraph(eager_dir, split_count):
181182
"${fluid_manual_functions} DEPS ${eager_deps} ${fluid_deps} ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})\n"
182183
)
183184
f.write(
184-
"add_dependencies(dygraph_function copy_dygraph_forward_functions)"
185+
"add_dependencies(dygraph_function copy_dygraph_forward_functions)\n"
185186
)
187+
f.write("target_link_libraries(dygraph_function ${PYTHON_LIBRARIES})\n")
186188

187189
with open(generated_level_cmakelist_path, "w") as f:
188190
f.write("add_subdirectory(forwards)\nadd_subdirectory(nodes)")

0 commit comments

Comments
 (0)