CMakeLists.txt 1002 字节
Newer Older
M
minqiyang 已提交
1 2 3
set(PYBIND_DEPS pybind python proto_desc memory executor async_executor prune
  feed_fetch_method pass_builder parallel_executor profiler layer scope_pool
  tracer)
S
sneaxiy 已提交
4 5 6
if(WITH_PYTHON)
  list(APPEND PYBIND_DEPS py_func_op)
endif()
7 8
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc async_executor_py.cc imperative.cc)

9
if(WITH_PYTHON)
10 11
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
12
      SRCS ${PYBIND_SRCS}
S
sabreshao 已提交
13 14
      DEPS ARCHIVE_START ${PYBIND_DEPS}
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS} ARCHIVE_END)
15 16
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
17
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
18
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
19
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
D
dzhwinter 已提交
20
    if(NOT APPLE AND NOT ANDROID AND NOT WIN32)
21
      target_link_libraries(paddle_pybind rt)
D
dzhwinter 已提交
22
    endif(NOT APPLE AND NOT ANDROID AND NOT WIN32)
23
  endif(WITH_AMD_GPU)
24

P
peizhilin 已提交
25 26 27 28
  if(WIN32)
    target_link_libraries(paddle_pybind shlwapi)
  endif(WIN32)

29
  cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python)
30
endif(WITH_PYTHON)