CMakeLists.txt 2.6 KB
Newer Older
W
wangguibao 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
if (NOT EXISTS
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/text_classification_lstm)
        execute_process(COMMAND wget
                --no-check-certificate https://paddle-serving.bj.bcebos.com/data/text_classification/text_classification_lstm.tar.gz
                --output-document
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/text_classification_lstm.tar.gz)

            execute_process(COMMAND ${CMAKE_COMMAND} -E tar xzf
                "${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/text_classification_lstm.tar.gz"
                WORKING_DIRECTORY
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid
                )
endif()

15
include_directories(SYSTEM  ${CMAKE_CURRENT_LIST_DIR}/../kvdb/include)
W
serving  
wangguibao 已提交
16 17 18
find_library(MKLML_LIBS NAMES libmklml_intel.so libiomp5.so)
include(op/CMakeLists.txt)
include(proto/CMakeLists.txt)
W
wangguibao 已提交
19 20
add_executable(serving ${serving_srcs})
add_dependencies(serving pdcodegen fluid_cpu_engine pdserving paddle_fluid
W
wangguibao 已提交
21
        opencv_imgcodecs cube-api)
W
wangguibao 已提交
22 23 24
if (WITH_GPU)
    add_dependencies(serving fluid_gpu_engine)
endif()
W
wangguibao 已提交
25
target_include_directories(serving PUBLIC
W
wangguibao 已提交
26
        ${CMAKE_CURRENT_BINARY_DIR}/../predictor
W
sdk-cpp  
wangguibao 已提交
27
        )
W
wangguibao 已提交
28 29

if(WITH_GPU)
30
    target_link_libraries(serving -Wl,--whole-archive fluid_gpu_engine
W
wangguibao 已提交
31 32
            -Wl,--no-whole-archive)
endif()
33 34 35 36 37 38

target_link_libraries(serving -Wl,--whole-archive fluid_cpu_engine
        -Wl,--no-whole-archive)

target_link_libraries(serving paddle_fluid ${paddle_depend_libs})

W
wangguibao 已提交
39
target_link_libraries(serving opencv_imgcodecs
40 41 42
        ${opencv_depend_libs})

target_link_libraries(serving pdserving)
W
wangguibao 已提交
43
target_link_libraries(serving cube-api)
44 45 46 47 48 49 50

target_link_libraries(serving kvdb rocksdb)

if(WITH_GPU)
    target_link_libraries(serving ${CUDA_LIBRARIES})
endif()
target_link_libraries(serving ${MKLML_LIB} ${MKLML_IOMP_LIB} -lpthread
51
        -lcrypto -lm -lrt -lssl -ldl -lz -lbz2)
52

W
wangguibao 已提交
53 54 55
install(TARGETS serving
        RUNTIME DESTINATION
        ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/bin)
56 57 58 59
install(DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/conf DESTINATION
        ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/)
install(DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/data DESTINATION
        ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/)
W
wangguibao 已提交
60 61 62 63 64

FILE(GLOB inc ${CMAKE_CURRENT_BINARY_DIR}/*.pb.h)
install(FILES ${inc}
        DESTINATION ${PADDLE_SERVING_INSTALL_DIR}/include/serving)

W
wangguibao 已提交
65 66 67 68 69 70
if (${WITH_MKL})
        install(FILES ${THIRD_PARTY_PATH}/install/mklml/lib/libmklml_intel.so
                ${THIRD_PARTY_PATH}/install/mklml/lib/libmklml_gnu.so
                ${THIRD_PARTY_PATH}/install/mklml/lib/libiomp5.so DESTINATION
                ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/bin)
endif()