CMakeLists.txt 4.5 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
  if (WITH_STATIC_LIB)
    safe_set_static_flag()
D
dzhwinter 已提交
20
    add_definitions(-DSTATIC_LIB)
D
dzhwinter 已提交
21 22 23 24
    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 已提交
25
else()
D
dzhwinter 已提交
26 27
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_STATIC_LIBRARY_PREFIX "")
D
dzhwinter 已提交
28
endif()
D
dzhwinter 已提交
29
message("flags" ${CMAKE_CXX_FLAGS})
30 31 32 33 34 35 36 37 38 39

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 已提交
40 41 42 43 44 45 46
  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)
47 48
endif()

D
dzhwinter 已提交
49
include_directories("D:/Paddle/")
50 51 52 53
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 已提交
54
if (NOT WIN32)
55 56 57
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 已提交
58
endif(NOT WIN32)
59 60 61 62

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

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

69 70 71
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 已提交
72
link_directories("${PADDLE_LIB}/paddle/fluid/inference")
73

D
dzhwinter 已提交
74 75
# add_executable(${DEMO_NAME} ${DEMO_NAME}.cc)
add_library(${DEMO_NAME} ${DEMO_NAME}.cc)
76 77
if(WITH_MKL)
  include_directories("${PADDLE_LIB}/third_party/install/mklml/include")
D
dzhwinter 已提交
78 79
  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})
80 81 82 83 84 85
  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 已提交
86
  set(MATH_LIB ${PADDLE_LIB}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
87 88
endif()

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

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

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

target_link_libraries(${DEMO_NAME} ${DEPS})