CMakeLists.txt 4.3 KB
Newer Older
1 2
cmake_minimum_required(VERSION 3.0)
project(cpp_inference_demo CXX C)
D
dzhwinter 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15
option(WITH_MKL        "Compile demo with MKL/OpenBlas support, default use MKL."       ON)
option(WITH_GPU        "Compile demo with GPU/CPU, default use CPU."                    OFF)
option(WITH_STATIC_LIB "Compile demo with static/shared library, default use static."   ON)

macro(safe_set_static_flag)
    foreach(flag_var
        CMAKE_CXX_FLAGS CMAKE_CXX_FLAGS_DEBUG CMAKE_CXX_FLAGS_RELEASE
        CMAKE_CXX_FLAGS_MINSIZEREL CMAKE_CXX_FLAGS_RELWITHDEBINFO)
      if(${flag_var} MATCHES "/MD")
        string(REGEX REPLACE "/MD" "/MT" ${flag_var} "${${flag_var}}")
      endif(${flag_var} MATCHES "/MD")
    endforeach(flag_var)
endmacro()
16

D
dzhwinter 已提交
17
if (WIN32)
D
dzhwinter 已提交
18 19 20 21 22 23
  if (WITH_STATIC_LIB)
    safe_set_static_flag()
    set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS} "/w")
    set(CMAKE_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE} "/w")
  endif()
  set(CMAKE_STATIC_LIBRARY_PREFIX "lib")
D
dzhwinter 已提交
24
else()
D
dzhwinter 已提交
25 26
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_STATIC_LIBRARY_PREFIX "")
D
dzhwinter 已提交
27
endif()
D
dzhwinter 已提交
28
message("flags" ${CMAKE_CXX_FLAGS})
29 30 31 32 33 34 35 36 37 38

if(NOT DEFINED PADDLE_LIB)
  message(FATAL_ERROR "please set PADDLE_LIB with -DPADDLE_LIB=/path/paddle/lib")
endif()
if(NOT DEFINED DEMO_NAME)
  message(FATAL_ERROR "please set DEMO_NAME with -DDEMO_NAME=demo_name")
endif()


if(WITH_GPU)
D
dzhwinter 已提交
39 40 41 42 43 44 45
  if(NOT WIN32)
    set(CUDA_LIB "/usr/local/cuda/lib64/" CACHE STRING "CUDA Library")
  else()
    if(CUDA_LIB STREQUAL "")
    set(CUDA_LIB "C:\\Program\ Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v8.0\\lib\\x64")
    endif()
  endif(NOT WIN32)
46 47
endif()

D
dzhwinter 已提交
48
include_directories("D:/Paddle/")
49 50 51 52
include_directories("${PADDLE_LIB}")
include_directories("${PADDLE_LIB}/third_party/install/protobuf/include")
include_directories("${PADDLE_LIB}/third_party/install/glog/include")
include_directories("${PADDLE_LIB}/third_party/install/gflags/include")
D
dzhwinter 已提交
53
if (NOT WIN32)
54 55 56
include_directories("${PADDLE_LIB}/third_party/install/snappy/include")
include_directories("${PADDLE_LIB}/third_party/install/snappystream/include")
include_directories("${PADDLE_LIB}/third_party/install/zlib/include")
D
dzhwinter 已提交
57
endif(NOT WIN32)
58 59 60 61

include_directories("${PADDLE_LIB}/third_party/boost")
include_directories("${PADDLE_LIB}/third_party/eigen3")

D
dzhwinter 已提交
62
if (NOT WIN32)
63 64
link_directories("${PADDLE_LIB}/third_party/install/snappy/lib")
link_directories("${PADDLE_LIB}/third_party/install/snappystream/lib")
D
dzhwinter 已提交
65 66 67
link_directories("${PADDLE_LIB}/third_party/install/zlib/lib")
endif(NOT WIN32)

68 69 70
link_directories("${PADDLE_LIB}/third_party/install/protobuf/lib")
link_directories("${PADDLE_LIB}/third_party/install/glog/lib")
link_directories("${PADDLE_LIB}/third_party/install/gflags/lib")
D
dzhwinter 已提交
71
link_directories("${PADDLE_LIB}/paddle/fluid/inference")
72 73 74 75 76

add_executable(${DEMO_NAME} ${DEMO_NAME}.cc)

if(WITH_MKL)
  include_directories("${PADDLE_LIB}/third_party/install/mklml/include")
D
dzhwinter 已提交
77 78
  set(MATH_LIB ${PADDLE_LIB}/third_party/install/mklml/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX} 
               ${PADDLE_LIB}/third_party/install/mklml/lib/libiomp5${CMAKE_SHARED_LIBRARY_SUFFIX})
79 80 81 82 83 84
  set(MKLDNN_PATH "${PADDLE_LIB}/third_party/install/mkldnn")
  if(EXISTS ${MKLDNN_PATH})
    include_directories("${MKLDNN_PATH}/include")
    set(MKLDNN_LIB ${MKLDNN_PATH}/lib/libmkldnn.so.0)
  endif()
else()
D
dzhwinter 已提交
85
  set(MATH_LIB ${PADDLE_LIB}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
86 87
endif()

88
# Note: libpaddle_inference_api.so/a must put before libpaddle_fluid.so/a
L
Luo Tao 已提交
89 90
if(WITH_STATIC_LIB)
  set(DEPS
D
dzhwinter 已提交
91
      ${PADDLE_LIB}/paddle/fluid/inference/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
L
Luo Tao 已提交
92 93
else()
  set(DEPS
D
dzhwinter 已提交
94
      ${PADDLE_LIB}/paddle/fluid/inference/libpaddle_fluid${CMAKE_SHARED_LIBRARY_SUFFIX})
L
Luo Tao 已提交
95
endif()
96

D
dzhwinter 已提交
97 98
if (NOT WIN32)
set(EXTERNAL_LIB "-lrt -ldl -lpthread")
L
Luo Tao 已提交
99 100
set(DEPS ${DEPS}
    ${MATH_LIB} ${MKLDNN_LIB}
101 102
    glog gflags protobuf snappystream snappy z
    ${EXTERNAL_LIB})
D
dzhwinter 已提交
103 104 105 106 107
else()
set(DEPS ${DEPS}
    ${MATH_LIB} ${MKLDNN_LIB}
    ${CMAKE_STATIC_LIBRARY_PREFIX}glog  ${CMAKE_STATIC_LIBRARY_PREFIX}gflags  ${CMAKE_STATIC_LIBRARY_PREFIX}protobuf
    ${EXTERNAL_LIB})
D
dzhwinter 已提交
108 109
# NOTE(dzhwinter) shlwapi is deprecated.
set(DEPS ${DEPS} libcmt shlwapi)
D
dzhwinter 已提交
110 111
endif(NOT WIN32)

112
if(WITH_GPU)
D
dzhwinter 已提交
113 114 115 116 117
  if(NOT WIN32)
    set(DEPS ${DEPS} ${CUDA_LIB}/libcudart${CMAKE_SHARED_LIBRARY_SUFFIX})
  else()
    set(DEPS ${DEPS} ${CUDA_LIB}/cudart${CMAKE_STATIC_LIBRARY_SUFFIX})
  endif()
118 119 120
endif()

target_link_libraries(${DEMO_NAME} ${DEPS})