CMakeLists.txt 3.2 KB
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6
if (WITH_DOUBLE)
  set(PADDLE_FLOAT_TYPE double)
else ()
  set(PADDLE_FLOAT_TYPE float)
endif()

Y
Yu Yang 已提交
7 8 9 10
# config.h used for C-API. It will store Paddle building configuration as a
# header. Make user just include PaddleCAPI.h then can get building
# configuration without explicitly set -DPADDLE_WITH_DOUBLE when building their
# libraries.
Y
Yu Yang 已提交
11 12
configure_file(config.h.in config.h @ONLY)

Y
Yu Yang 已提交
13 14
# PaddleCAPI.h is the only header we exposed. It currently only used for model
# inference.
Y
Yu Yang 已提交
15
file(GLOB CAPI_HEADERS *.h)
Y
Yu Yang 已提交
16
set(CAPI_PRIVATE_HEADER capi_private.h)
Y
Yu Yang 已提交
17
list(REMOVE_ITEM CAPI_HEADERS ${CAPI_PRIVATE_HEADER})
Y
Yu Yang 已提交
18 19
file(GLOB CAPI_SOURCES *.cpp)

Y
Yu Yang 已提交
20
# building paddle_capi
Y
Yu Yang 已提交
21 22
add_library(paddle_capi STATIC ${CAPI_HEADERS} ${CAPI_PRIVATE_HEADER}
  ${CAPI_SOURCES})
Y
Yu Yang 已提交
23 24

target_include_directories(paddle_capi PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
Y
Yu Yang 已提交
25

Y
Yu Yang 已提交
26 27
add_style_check_target(paddle_capi ${CAPI_SOURCES} ${CAPI_HEADER}
  ${CAPI_PRIVATE_HEADER})
Y
Yu Yang 已提交
28

Y
Yu Yang 已提交
29 30
add_dependencies(paddle_capi gen_proto_cpp)

Y
Yu Yang 已提交
31 32 33 34

# combine all paddle static libraries together, into libpaddle_capi_whole.a
# user should use PaddleCAPI as -lpaddle_capi_whole
set(capi_whole_library libpaddle_capi_whole.a)
35
add_custom_target(paddle_capi_whole ALL
Y
Yu Yang 已提交
36 37 38 39 40 41 42 43
        COMMAND mkdir -p o_files/capi && cd o_files/capi/ && ar -x $<TARGET_FILE:paddle_capi>
        COMMAND mkdir -p o_files/utils && cd o_files/utils/ && ar -x $<TARGET_FILE:paddle_utils>
        COMMAND mkdir -p o_files/parameter && cd o_files/parameter/ && ar -x $<TARGET_FILE:paddle_parameter>
        COMMAND mkdir -p o_files/math && cd o_files/math/  && ar -x $<TARGET_FILE:paddle_math>
        COMMAND mkdir -p o_files/cuda && cd o_files/cuda/ && ar -x $<TARGET_FILE:paddle_cuda>
        COMMAND mkdir -p o_files/function && cd o_files/function/ && ar -x $<TARGET_FILE:paddle_function>
        COMMAND mkdir -p o_files/gserver && cd o_files/gserver/ && ar -x $<TARGET_FILE:paddle_gserver>
        COMMAND mkdir -p o_files/proto && cd o_files/proto/ && ar -x $<TARGET_FILE:paddle_proto>
Y
Yu Yang 已提交
44 45
        COMMAND mkdir -p o_files/network && cd o_files/network/ && ar -x $<TARGET_FILE:paddle_network>
        COMMAND mkdir -p o_files/pserver && cd o_files/pserver/ && ar -x $<TARGET_FILE:paddle_pserver>
Y
Yu Yang 已提交
46 47 48 49
        COMMAND ar crs ${capi_whole_library} `find ./o_files -name '*.o'`
        COMMAND rm -rf o_files
        WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
        DEPENDS paddle_capi paddle_utils paddle_parameter paddle_math
Y
Yu Yang 已提交
50 51
                paddle_cuda paddle_function paddle_gserver
                paddle_proto paddle_pserver paddle_network
Y
Yu Yang 已提交
52 53 54 55
        )
set_target_properties(paddle_capi_whole
  PROPERTIES IMPORTED_LOCATION ${CMAKE_CURRENT_BINARY_DIR}/${capi_whole_library})

Y
Yu Yang 已提交
56 57 58 59
add_library(paddle_capi_shared SHARED ${CAPI_SOURCES})
target_include_directories(paddle_capi_shared PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
link_paddle_exe(paddle_capi_shared)

Y
Yu Yang 已提交
60 61
# install library & headers.
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/${capi_whole_library} DESTINATION lib)
Y
Yu Yang 已提交
62
install(FILES ${CAPI_HEADERS} DESTINATION include/paddle)
Y
Yu Yang 已提交
63
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/config.h DESTINATION include/paddle)
Y
Yu Yang 已提交
64
install(TARGETS paddle_capi_shared DESTINATION lib)
Y
Yu Yang 已提交
65 66

# this variable used for unittest
Y
Yu Yang 已提交
67 68 69 70 71 72 73
set(PADDLE_CAPI_INC_PATH
  ${CMAKE_CURRENT_BINARY_DIR}
  ${CMAKE_CURRENT_SOURCE_DIR})

if (WITH_TESTING)
  add_subdirectory(tests)
endif()