CMakeLists.txt 774 字节
Newer Older
D
dzhwinter 已提交
1 2 3 4 5
set(PYBIND_DEPS pybind python proto_desc memory executor prune profiler feed_fetch_method
          )
if(NOT WIN32)
list(APPEND PYBIND_DEPS parallel_executor)
endif()
6
if(WITH_PYTHON)
7 8
  if(WITH_AMD_GPU)
    hip_library(paddle_pybind SHARED
S
sabreshao 已提交
9
      SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc
D
dzhwinter 已提交
10
      DEPS ${PYBIND_DEPS}
11 12 13
      ${GLOB_OP_LIB})
  else()
    cc_library(paddle_pybind SHARED
S
sabreshao 已提交
14
      SRCS pybind.cc exception.cc protobuf.cc const_value.cc recordio.cc
D
dzhwinter 已提交
15
      DEPS ${PYBIND_DEPS}
16
      ${GLOB_OP_LIB})
D
dzhwinter 已提交
17
    if(NOT APPLE AND NOT ANDROID AND NOT WIN32)
18
      target_link_libraries(paddle_pybind rt)
D
dzhwinter 已提交
19
    endif(NOT APPLE AND NOT ANDROID AND NOT WIN32)
20
  endif(WITH_AMD_GPU)
21 22

  cc_test(tensor_py_test SRCS tensor_py_test.cc DEPS python)
23
endif(WITH_PYTHON)