CMakeLists.txt 1.1 KB
Newer Older
D
dzhwinter 已提交
1

2
set(PYBIND_DEPS pybind python proto_desc memory executor prune  feed_fetch_method pass_builder)
D
dzhwinter 已提交
3
set(PYBIND_SRCS pybind.cc exception.cc protobuf.cc const_value.cc)
D
dzhwinter 已提交
4
if(NOT WIN32)
P
peizhilin 已提交
5 6
  list(APPEND PYBIND_DEPS parallel_executor profiler)
  list(APPEND PYBIND_SRCS recordio.cc)
P
peizhilin 已提交
7
endif(NOT WIN32)
8
if(WITH_PYTHON)
9 10
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
D
dzhwinter 已提交
11
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
12
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
13
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
14 15
  else()
    cc_library(paddle_pybind SHARED
D
dzhwinter 已提交
16
      SRCS ${PYBIND_SRCS}
D
dzhwinter 已提交
17
      DEPS ${PYBIND_DEPS}
W
Wu Yi 已提交
18
      ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
D
dzhwinter 已提交
19
    if(NOT APPLE AND NOT ANDROID AND NOT WIN32)
20
      target_link_libraries(paddle_pybind rt)
D
dzhwinter 已提交
21
    endif(NOT APPLE AND NOT ANDROID AND NOT WIN32)
22
  endif(WITH_AMD_GPU)
23

P
peizhilin 已提交
24
  if(WIN32)
P
peizhilin 已提交
25 26 27 28
    if(WITH_GPU AND NOT WITH_DSO)
      get_property(cuda_modules GLOBAL PROPERTY CUDA_MODULES)
      target_link_libraries(paddle_pybind ${cuda_modules})
    endif(WITH_GPU AND NOT WITH_DSO)
P
peizhilin 已提交
29 30 31
    target_link_libraries(paddle_pybind shlwapi)
  endif(WIN32)

32
  cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python)
33
endif(WITH_PYTHON)