CMakeLists.txt 3.3 KB
Newer Older
W
wangguibao 已提交
1 2 3 4 5 6 7
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)

W
wangguibao 已提交
8
        execute_process(COMMAND ${CMAKE_COMMAND} -E tar xzf
W
wangguibao 已提交
9 10 11 12 13 14
                "${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/text_classification_lstm.tar.gz"
                WORKING_DIRECTORY
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid
                )
endif()

W
wangguibao 已提交
15 16 17 18 19 20 21 22 23 24 25 26
if (NOT EXISTS
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/ctr_prediction)
        execute_process(COMMAND wget --no-check-certificate
                https://paddle-serving.bj.bcebos.com/data/ctr_prediction/ctr_prediction.tar.gz
                --output-document
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/ctr_prediction.tar.gz)
        execute_process(COMMAND ${CMAKE_COMMAND} -E tar xzf
                "${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid/ctr_prediction.tar.gz"
                WORKING_DIRECTORY
                ${CMAKE_CURRENT_LIST_DIR}/data/model/paddle/fluid)
endif()

27
include_directories(SYSTEM  ${CMAKE_CURRENT_LIST_DIR}/../kvdb/include)
W
serving  
wangguibao 已提交
28 29 30
find_library(MKLML_LIBS NAMES libmklml_intel.so libiomp5.so)
include(op/CMakeLists.txt)
include(proto/CMakeLists.txt)
W
wangguibao 已提交
31 32
add_executable(serving ${serving_srcs})
add_dependencies(serving pdcodegen fluid_cpu_engine pdserving paddle_fluid
W
wangguibao 已提交
33
        opencv_imgcodecs cube-api)
W
wangguibao 已提交
34 35 36
if (WITH_GPU)
    add_dependencies(serving fluid_gpu_engine)
endif()
W
wangguibao 已提交
37
target_include_directories(serving PUBLIC
W
wangguibao 已提交
38
        ${CMAKE_CURRENT_BINARY_DIR}/../predictor
W
sdk-cpp  
wangguibao 已提交
39
        )
W
wangguibao 已提交
40 41

if(WITH_GPU)
42
    target_link_libraries(serving -Wl,--whole-archive fluid_gpu_engine
W
wangguibao 已提交
43 44
            -Wl,--no-whole-archive)
endif()
45 46 47 48 49 50

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 已提交
51
target_link_libraries(serving opencv_imgcodecs
52 53 54
        ${opencv_depend_libs})

target_link_libraries(serving pdserving)
W
wangguibao 已提交
55
target_link_libraries(serving cube-api)
56 57 58 59 60 61 62

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
63
        -lcrypto -lm -lrt -lssl -ldl -lz -lbz2)
64

W
wangguibao 已提交
65 66 67
install(TARGETS serving
        RUNTIME DESTINATION
        ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/bin)
68 69 70 71
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 已提交
72 73 74 75 76

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

W
wangguibao 已提交
77 78 79 80 81 82
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()