CMakeLists.txt 3.4 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

T
tensor-tang 已提交
7 8 9 10 11 12 13 14
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 已提交
15 16
  list(APPEND MKL_SHARED_LIBS "${MKLDNN_SHARED_LIB}")
  list(APPEND MKL_DEPENDS mkldnn mkldnn_shared_lib)
T
tensor-tang 已提交
17 18
endif()

T
typhoonzero 已提交
19
if(WITH_GPU)
T
update  
typhoonzero 已提交
20
  SET(PACKAGE_NAME "paddlepaddle-gpu")
T
typhoonzero 已提交
21 22 23 24
else()
  SET(PACKAGE_NAME "paddlepaddle")
endif()

Z
zhangjinchao01 已提交
25 26
configure_file(${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/setup.py)
P
peizhilin 已提交
27 28
IF(WIN32)
    # Python would use the .pyd by default under Windows series platform
P
peizhilin 已提交
29 30
    set(FLUID_DST_DIR ${PADDLE_BINARY_DIR}/python/paddle/fluid/)
    set(FLUID_CORE ${FLUID_DST_DIR}/core.pyd)
P
peizhilin 已提交
31 32 33
ELSE()
    set(FLUID_CORE ${PADDLE_BINARY_DIR}/python/paddle/fluid/core.so)
ENDIF()
P
peizhilin 已提交
34 35 36
add_custom_command(OUTPUT ${FLUID_CORE}
        COMMAND cmake -E copy $<TARGET_FILE:paddle_pybind> ${FLUID_CORE}
        DEPENDS paddle_pybind)
37
add_custom_target(copy_paddle_pybind ALL DEPENDS ${FLUID_CORE})
38

P
peizhilin 已提交
39
IF(WIN32)
P
peizhilin 已提交
40 41 42 43 44
    add_custom_command(OUTPUT ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
            COMMAND ${CMAKE_COMMAND} -E copy_directory ${PADDLE_SOURCE_DIR}/python/paddle ${PADDLE_BINARY_DIR}/python/paddle/
            COMMAND ${CMAKE_COMMAND} -E env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
            COMMAND ${CMAKE_COMMAND} -E touch ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
            COMMAND ${CMAKE_COMMAND} -E remove_directory ${PADDLE_PYTHON_BUILD_DIR}/lib-python
T
Tao Luo 已提交
45
            DEPENDS copy_paddle_pybind ${FLUID_CORE} framework_py_proto profiler_py_proto ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER})
P
peizhilin 已提交
46 47 48 49 50 51 52 53
ELSE(WIN32)
	add_custom_command(OUTPUT ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
		COMMAND touch stub.cc
		COMMAND cp -r ${PADDLE_SOURCE_DIR}/python/paddle ${PADDLE_BINARY_DIR}/python
		COMMAND env ${py_env} ${PYTHON_EXECUTABLE} setup.py bdist_wheel
		COMMAND ${CMAKE_COMMAND} -E touch ${PADDLE_PYTHON_BUILD_DIR}/.timestamp
		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
T
Tao Luo 已提交
54
		DEPENDS copy_paddle_pybind ${FLUID_CORE} framework_py_proto profiler_py_proto ${PY_FILES} ${external_project_dependencies} ${COPY_PADDLE_MASTER})
P
peizhilin 已提交
55
ENDIF()
Z
zhangjinchao01 已提交
56

L
Luo Tao 已提交
57
set(paddle_python_deps ${PADDLE_PYTHON_BUILD_DIR}/.timestamp ${MKL_DEPENDS})
58
add_custom_target(paddle_python ALL DEPENDS ${paddle_python_deps})
Z
zhangjinchao01 已提交
59

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

62
if (WITH_TESTING)
63 64
  add_subdirectory(paddle/reader/tests)
  add_subdirectory(paddle/dataset/tests)
L
Luo Tao 已提交
65
  add_subdirectory(paddle/fluid/tests)
66
  add_subdirectory(paddle/fluid/contrib/tests)
W
WangZhen 已提交
67
  add_subdirectory(paddle/fluid/contrib/slim/tests)
68
endif()
Y
Yu Yang 已提交
69
install(DIRECTORY ${PADDLE_PYTHON_PACKAGE_DIR}
Z
zhangjinchao01 已提交
70 71
    DESTINATION opt/paddle/share/wheels
)
72

73 74 75 76 77
if(APPLE)
  find_program(INSTALL_NAME_TOOL_EXECUTABLE install_name_tool)
  if(NOT INSTALL_NAME_TOOL_EXECUTABLE)
    message(FATAL_ERROR "install_name_tool not found, please check.\n")
  endif()
D
dzhwinter 已提交
78 79
endif()
if(LINUX)
80 81 82 83 84
  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()
D
dzhwinter 已提交
85
endif(LINUX)