You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
182 lines
6.8 KiB
182 lines
6.8 KiB
cmake_minimum_required(VERSION 3.0)
|
|
project(cpp_inference_demo CXX C)
|
|
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)
|
|
option(USE_TENSORRT "Compile demo with TensorRT." OFF)
|
|
|
|
if(NOT WITH_STATIC_LIB)
|
|
add_definitions("-DPADDLE_WITH_SHARED_LIB")
|
|
endif()
|
|
|
|
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()
|
|
|
|
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()
|
|
|
|
include_directories("${PADDLE_LIB}/")
|
|
set(PADDLE_LIB_THIRD_PARTY_PATH "${PADDLE_LIB}/third_party/install/")
|
|
include_directories("${PADDLE_LIB_THIRD_PARTY_PATH}protobuf/include")
|
|
include_directories("${PADDLE_LIB_THIRD_PARTY_PATH}glog/include")
|
|
include_directories("${PADDLE_LIB_THIRD_PARTY_PATH}gflags/include")
|
|
include_directories("${PADDLE_LIB_THIRD_PARTY_PATH}xxhash/include")
|
|
|
|
link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}protobuf/lib")
|
|
link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}glog/lib")
|
|
link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}gflags/lib")
|
|
link_directories("${PADDLE_LIB_THIRD_PARTY_PATH}xxhash/lib")
|
|
link_directories("${PADDLE_LIB}/paddle/lib")
|
|
|
|
if (WIN32)
|
|
add_definitions("/DGOOGLE_GLOG_DLL_DECL=")
|
|
option(MSVC_STATIC_CRT "use static C Runtime library by default" ON)
|
|
if (MSVC_STATIC_CRT)
|
|
set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} /bigobj /MTd")
|
|
set(CMAKE_C_FLAGS_RELEASE "${CMAKE_C_FLAGS_RELEASE} /bigobj /MT")
|
|
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /bigobj /MTd")
|
|
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /bigobj /MT")
|
|
if (WITH_STATIC_LIB)
|
|
safe_set_static_flag()
|
|
add_definitions(-DSTATIC_LIB)
|
|
endif()
|
|
endif()
|
|
else()
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
|
|
endif()
|
|
message("flags" ${CMAKE_CXX_FLAGS})
|
|
|
|
if(WITH_GPU)
|
|
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)
|
|
endif()
|
|
|
|
if (USE_TENSORRT AND WITH_GPU)
|
|
set(TENSORRT_ROOT ${PADDLE_LIB_THIRD_PARTY_PATH}tensorrt)
|
|
set(TENSORRT_INCLUDE_DIR ${TENSORRT_ROOT}/include)
|
|
set(TENSORRT_LIB_DIR ${TENSORRT_ROOT}/lib)
|
|
endif()
|
|
|
|
if (NOT WIN32)
|
|
if (USE_TENSORRT AND WITH_GPU)
|
|
include_directories("${TENSORRT_INCLUDE_DIR}")
|
|
link_directories("${TENSORRT_LIB_DIR}")
|
|
endif()
|
|
endif(NOT WIN32)
|
|
|
|
if(WITH_MKL)
|
|
set(MATH_LIB_PATH "${PADDLE_LIB_THIRD_PARTY_PATH}mklml")
|
|
include_directories("${MATH_LIB_PATH}/include")
|
|
if(WIN32)
|
|
set(MATH_LIB ${MATH_LIB_PATH}/lib/mklml${CMAKE_STATIC_LIBRARY_SUFFIX}
|
|
${MATH_LIB_PATH}/lib/libiomp5md${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
else()
|
|
set(MATH_LIB ${MATH_LIB_PATH}/lib/libmklml_intel${CMAKE_SHARED_LIBRARY_SUFFIX}
|
|
${MATH_LIB_PATH}/lib/libiomp5${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
endif()
|
|
set(MKLDNN_PATH "${PADDLE_LIB_THIRD_PARTY_PATH}mkldnn")
|
|
if(EXISTS ${MKLDNN_PATH})
|
|
include_directories("${MKLDNN_PATH}/include")
|
|
if(WIN32)
|
|
set(MKLDNN_LIB ${MKLDNN_PATH}/lib/mkldnn.lib)
|
|
else(WIN32)
|
|
set(MKLDNN_LIB ${MKLDNN_PATH}/lib/libmkldnn.so.0)
|
|
endif(WIN32)
|
|
endif()
|
|
else()
|
|
set(OPENBLAS_LIB_PATH "${PADDLE_LIB_THIRD_PARTY_PATH}openblas")
|
|
include_directories("${OPENBLAS_LIB_PATH}/include/openblas")
|
|
if(WIN32)
|
|
set(MATH_LIB ${OPENBLAS_LIB_PATH}/lib/openblas${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
else()
|
|
set(MATH_LIB ${OPENBLAS_LIB_PATH}/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
endif()
|
|
endif()
|
|
|
|
if(WITH_STATIC_LIB)
|
|
set(DEPS ${PADDLE_LIB}/paddle/lib/libpaddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
else()
|
|
if(WIN32)
|
|
set(DEPS ${PADDLE_LIB}/paddle/lib/paddle_fluid${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
else()
|
|
set(DEPS ${PADDLE_LIB}/paddle/lib/libpaddle_fluid${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
endif()
|
|
endif()
|
|
|
|
if (NOT WIN32)
|
|
set(EXTERNAL_LIB "-lrt -ldl -lpthread")
|
|
set(DEPS ${DEPS}
|
|
${MATH_LIB} ${MKLDNN_LIB}
|
|
glog gflags protobuf xxhash
|
|
${EXTERNAL_LIB})
|
|
else()
|
|
set(DEPS ${DEPS}
|
|
${MATH_LIB} ${MKLDNN_LIB}
|
|
glog gflags_static libprotobuf xxhash ${EXTERNAL_LIB})
|
|
set(DEPS ${DEPS} libcmt shlwapi.lib)
|
|
endif(NOT WIN32)
|
|
|
|
if(WITH_GPU)
|
|
if(NOT WIN32)
|
|
if (USE_TENSORRT)
|
|
set(DEPS ${DEPS} ${TENSORRT_LIB_DIR}/libnvinfer${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
set(DEPS ${DEPS} ${TENSORRT_LIB_DIR}/libnvinfer_plugin${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
endif()
|
|
set(DEPS ${DEPS} ${CUDA_LIB}/libcudart${CMAKE_SHARED_LIBRARY_SUFFIX})
|
|
else()
|
|
if(USE_TENSORRT)
|
|
set(DEPS ${DEPS} ${TENSORRT_LIB_DIR}/nvinfer${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
set(DEPS ${DEPS} ${TENSORRT_LIB_DIR}/nvinfer_plugin${CMAKE_STATIC_LIBRARY_SUFFIX})
|
|
endif()
|
|
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} )
|
|
endif()
|
|
endif()
|
|
|
|
add_executable(${DEMO_NAME} ${DEMO_NAME}.cc)
|
|
target_link_libraries(${DEMO_NAME} ${DEPS})
|
|
if(WIN32)
|
|
if(USE_TENSORRT)
|
|
add_custom_command(TARGET ${DEMO_NAME} POST_BUILD
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${TENSORRT_LIB_DIR}/nvinfer${CMAKE_SHARED_LIBRARY_SUFFIX}
|
|
${CMAKE_BINARY_DIR}/${CMAKE_BUILD_TYPE}
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${TENSORRT_LIB_DIR}/nvinfer_plugin${CMAKE_SHARED_LIBRARY_SUFFIX}
|
|
${CMAKE_BINARY_DIR}/${CMAKE_BUILD_TYPE}
|
|
)
|
|
endif()
|
|
if(WITH_MKL)
|
|
add_custom_command(TARGET ${DEMO_NAME} POST_BUILD
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${MATH_LIB_PATH}/lib/mklml.dll ${CMAKE_BINARY_DIR}/Release
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${MATH_LIB_PATH}/lib/libiomp5md.dll ${CMAKE_BINARY_DIR}/Release
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${MATH_LIB_PATH}/lib/msvcr120.dll ${CMAKE_BINARY_DIR}/Release
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${MKLDNN_PATH}/lib/mkldnn.dll ${CMAKE_BINARY_DIR}/Release
|
|
)
|
|
else()
|
|
add_custom_command(TARGET ${DEMO_NAME} POST_BUILD
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${OPENBLAS_LIB_PATH}/lib/openblas.dll ${CMAKE_BINARY_DIR}/Release
|
|
)
|
|
endif()
|
|
if(NOT WITH_STATIC_LIB)
|
|
add_custom_command(TARGET ${DEMO_NAME} POST_BUILD
|
|
COMMAND ${CMAKE_COMMAND} -E copy ${PADDLE_LIB}/paddle/lib/paddle_fluid${CMAKE_SHARED_LIBRARY_SUFFIX} ${CMAKE_BINARY_DIR}/Release
|
|
)
|
|
endif()
|
|
endif()
|