31 lines
1.1 KiB
31 lines
1.1 KiB
set(PYBIND_DEPS pybind python proto_desc memory executor async_executor prune
|
|
feed_fetch_method pass_builder parallel_executor profiler layer scope_pool
|
|
tracer analysis_predictor imperative_profiler)
|
|
|
|
if(WITH_PYTHON)
|
|
list(APPEND PYBIND_DEPS py_func_op)
|
|
endif()
|
|
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc async_executor_py.cc imperative.cc ir.cc inference_api.cc)
|
|
|
|
if(WITH_PYTHON)
|
|
if(WITH_AMD_GPU)
|
|
hip_library(paddle_pybind SHARED
|
|
SRCS ${PYBIND_SRCS}
|
|
DEPS ARCHIVE_START ${PYBIND_DEPS}
|
|
${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} ARCHIVE_END)
|
|
else()
|
|
cc_library(paddle_pybind SHARED
|
|
SRCS ${PYBIND_SRCS}
|
|
DEPS ${PYBIND_DEPS}
|
|
${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
|
|
if(NOT APPLE AND NOT WIN32)
|
|
target_link_libraries(paddle_pybind rt)
|
|
endif(NOT APPLE AND NOT WIN32)
|
|
endif(WITH_AMD_GPU)
|
|
|
|
get_property (os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
|
|
target_link_libraries(paddle_pybind ${os_dependency_modules})
|
|
|
|
cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python pybind)
|
|
endif(WITH_PYTHON)
|