CMakeLists.txt 3.5 KB
Newer Older
X
Xin Pan 已提交
1
file(GLOB UTILS_PY_FILES . ./paddle/legacy/utils/*.py)
2
file(GLOB_RECURSE FLUID_PY_FILES ./paddle/fluid/*.py)
3
set(PY_FILES paddle/__init__.py
H
Helin Wang 已提交
4
  ${UTILS_PY_FILES}
5
  ${FLUID_PY_FILES})
H
Helin Wang 已提交
6

7
if(NOT WITH_FLUID_ONLY)
L
Luo Tao 已提交
8 9
  file(GLOB TRAINER_PY_FILES . ./paddle/trainer/*.py)
  file(GLOB HELPERS_PY_FILES . ./paddle/trainer_config_helpers/*.py)
10
  file(GLOB_RECURSE V2_PY_FILES ./paddle/v2/*.py)
L
Luo Tao 已提交
11 12 13 14
  set(PY_FILES ${PY_FILES}
    ${TRAINER_PY_FILES}
    ${HELPERS_PY_FILES}
    ${V2_PY_FILES})
D
dongzhihong 已提交
15

L
Luo Tao 已提交
16 17 18 19 20 21 22 23 24 25 26
  add_custom_target(copy_paddle_master)

  SET(COPY_PADDLE_MASTER "")
  if(WITH_GOLANG)
    SET(COPY_PADDLE_MASTER "copy_paddle_master")
    add_custom_command(TARGET ${COPY_PADDLE_MASTER}
      COMMAND cp ${paddle_master_LIB_PATH} ${PADDLE_SOURCE_DIR}/python/paddle/v2/master/
      )
    add_dependencies(copy_paddle_master paddle_master)
  endif(WITH_GOLANG)
endif()
Z
zhangjinchao01 已提交
27

T
tensor-tang 已提交
28 29 30 31 32 33 34 35
set(MKL_SHARED_LIBS "")
set(MKL_DEPENDS "")
if(WITH_MKLML)
  list(APPEND MKL_SHARED_LIBS ${MKLML_LIB} ${MKLML_IOMP_LIB})
  list(APPEND MKL_DEPENDS mklml)
endif()

if(WITH_MKLDNN)
T
tensor-tang 已提交
36 37
  list(APPEND MKL_SHARED_LIBS "${MKLDNN_SHARED_LIB}")
  list(APPEND MKL_DEPENDS mkldnn mkldnn_shared_lib)
T
tensor-tang 已提交
38 39
endif()

T
typhoonzero 已提交
40
if(WITH_GPU)
T
update  
typhoonzero 已提交
41
  SET(PACKAGE_NAME "paddlepaddle-gpu")
T
typhoonzero 已提交
42 43 44 45
else()
  SET(PACKAGE_NAME "paddlepaddle")
endif()

Z
zhangjinchao01 已提交
46 47 48
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)

49 50 51
set(FLUID_CORE ${PADDLE_BINARY_DIR}/python/paddle/fluid/core.so)
add_custom_command(OUTPUT ${FLUID_CORE}
        COMMAND cmake -E copy $<TARGET_FILE:paddle_pybind> ${FLUID_CORE}
52
        DEPENDS paddle_pybind)
53
add_custom_target(copy_paddle_pybind ALL DEPENDS ${FLUID_CORE})
54 55


Q
qiaolongfei 已提交
56
add_custom_command(OUTPUT ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
T
typhoonzero 已提交
57
    COMMAND touch stub.cc
58
    COMMAND cp -r ${PADDLE_SOURCE_DIR}/python/paddle ${PADDLE_BINARY_DIR}/python
59
    COMMAND cp -r ${PADDLE_SOURCE_DIR}/paddle/py_paddle ${PADDLE_BINARY_DIR}/python/
L
liaogang 已提交
60
    COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
Q
qiaolongfei 已提交
61
    COMMAND ${CMAKE_COMMAND} -E touch ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
Q
qiaolongfei 已提交
62 63
    COMMAND ${CMAKE_COMMAND} -E remove_directory ${PADDLE_PYTHON_BUILD_DIR}/lib-python
    COMMAND ${CMAKE_COMMAND} -E copy_directory ${PADDLE_PYTHON_BUILD_DIR}/lib* ${PADDLE_PYTHON_BUILD_DIR}/lib-python
64
    DEPENDS gen_proto_py copy_paddle_pybind ${FLUID_CORE} framework_py_proto profiler_py_proto ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER})
Z
zhangjinchao01 已提交
65

L
Luo Tao 已提交
66
set(paddle_python_deps ${PADDLE_PYTHON_BUILD_DIR}/.timestamp ${MKL_DEPENDS})
67
if(NOT WITH_FLUID_ONLY)
L
Luo Tao 已提交
68 69 70 71
    set(paddle_python_deps ${paddle_python_deps} paddle_pserver_main paddle_trainer paddle_merge_model)
    if(WITH_SWIG_PY)
        list(APPEND paddle_python_deps python_api_wheel)
    endif()
72 73
endif()
add_custom_target(paddle_python ALL DEPENDS ${paddle_python_deps})
Z
zhangjinchao01 已提交
74

Y
Yu Yang 已提交
75 76
set(PADDLE_PYTHON_PACKAGE_DIR ${CMAKE_CURRENT_BINARY_DIR}/dist/)

77
if (WITH_TESTING)
78 79
  add_subdirectory(paddle/reader/tests)
  add_subdirectory(paddle/dataset/tests)
80
  if(NOT WITH_FLUID_ONLY)
L
Luo Tao 已提交
81 82 83 84 85
    add_subdirectory(paddle/trainer_config_helpers/tests)
    if (WITH_SWIG_PY)
      # enable v2 API unittest only when paddle swig api is compiled
      add_subdirectory(paddle/v2/tests)
      add_subdirectory(paddle/v2/plot/tests)
86
      add_subdirectory(paddle/v2/reader/tests)
L
Luo Tao 已提交
87
    endif()
88
  endif()
L
Luo Tao 已提交
89
  add_subdirectory(paddle/fluid/tests)
90
endif()
Y
Yu Yang 已提交
91
install(DIRECTORY ${PADDLE_PYTHON_PACKAGE_DIR}
Z
zhangjinchao01 已提交
92 93
    DESTINATION opt/paddle/share/wheels
)
94 95 96 97 98 99

find_program(PATCHELF_EXECUTABLE patchelf)
if(NOT PATCHELF_EXECUTABLE)
  message(FATAL_ERROR "patchelf not found, please install it.\n"
          "For Ubuntu, the command is: apt-get install -y patchelf.")
endif()