CMakeLists.txt 3.5 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
wangguibao 已提交
28 29 30 31 32 33

find_library(MKLML_LIB NAMES libmklml_intel.so PATHS                                
    ${CMAKE_BINARY_DIR}/Paddle/third_party/install/mklml/lib/)                      
find_library(MKLML_IOMP_LIB NAMES libiomp5.so PATHS                                 
    ${CMAKE_BINARY_DIR}/Paddle/third_party/install/mklml/lib)

W
serving  
wangguibao 已提交
34 35
include(op/CMakeLists.txt)
include(proto/CMakeLists.txt)
W
wangguibao 已提交
36 37
add_executable(serving ${serving_srcs})
add_dependencies(serving pdcodegen fluid_cpu_engine pdserving paddle_fluid
W
wangguibao 已提交
38
        opencv_imgcodecs cube-api)
W
wangguibao 已提交
39 40 41
if (WITH_GPU)
    add_dependencies(serving fluid_gpu_engine)
endif()
W
wangguibao 已提交
42
target_include_directories(serving PUBLIC
W
wangguibao 已提交
43
        ${CMAKE_CURRENT_BINARY_DIR}/../predictor
W
sdk-cpp  
wangguibao 已提交
44
        )
W
wangguibao 已提交
45 46

if(WITH_GPU)
47
    target_link_libraries(serving -Wl,--whole-archive fluid_gpu_engine
W
wangguibao 已提交
48 49
            -Wl,--no-whole-archive)
endif()
50 51 52 53 54 55

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 已提交
56
target_link_libraries(serving opencv_imgcodecs
57 58 59
        ${opencv_depend_libs})

target_link_libraries(serving pdserving)
W
wangguibao 已提交
60
target_link_libraries(serving cube-api)
61 62 63 64 65 66 67

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

W
wangguibao 已提交
70 71 72
install(TARGETS serving
        RUNTIME DESTINATION
        ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/bin)
73 74 75 76
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 已提交
77 78 79 80 81

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

W
wangguibao 已提交
82
if (${WITH_MKL})
W
wangguibao 已提交
83 84
    install(FILES ${CMAKE_BINARY_DIR}/Paddle/third_party/install/mklml/lib/libmklml_intel.so
                ${CMAKE_BINARY_DIR}/Paddle/third_party/install/mklml/lib/libiomp5.so DESTINATION
W
wangguibao 已提交
85 86
                ${PADDLE_SERVING_INSTALL_DIR}/demo/serving/bin)
endif()