CMakeLists.txt 4.7 KB
Newer Older
1
if(WITH_TESTING)
T
Tao Luo 已提交
2
  include(tests/test.cmake) # some generic cmake funtion for inference
3
endif()
4

5
set(FLUID_CORE_MODULES proto_desc memory lod_tensor executor data_feed_proto)
6

X
Xin Pan 已提交
7
# TODO(panyx0718): Should this be called paddle_fluid_inference_api_internal?
8
cc_library(paddle_fluid_api
9
    SRCS io.cc
W
Wu Yi 已提交
10
    DEPS ${FLUID_CORE_MODULES} ${GLOB_OP_LIB} ${GLOB_OPERATOR_DEPS})
11

Y
Yiqun Liu 已提交
12 13 14 15 16 17 18
# analysis and tensorrt must be added before creating static library,
# otherwise, there would be undefined reference to them in static library.
add_subdirectory(analysis)
add_subdirectory(utils)
if (TENSORRT_FOUND)
  add_subdirectory(tensorrt)
endif()
19

20
if (ANAKIN_SUBGRAPH)
21 22
  add_subdirectory(anakin)
endif()
Y
Yiqun Liu 已提交
23

石晓伟 已提交
24 25 26 27
if (WITH_LITE)
  add_subdirectory(lite)
endif()

28
get_property(fluid_modules GLOBAL PROPERTY FLUID_MODULES)
P
peizhilin 已提交
29
get_property(cuda_modules GLOBAL PROPERTY CUDA_MODULES)
D
done  
dzhwinter 已提交
30 31 32 33
get_property(fluid_third_partys GLOBAL PROPERTY FLUID_THRID_PARTYS)
if (WIN32)
list(APPEND fluid_third_partys gflags glog protobuf cblas)
endif(WIN32)
L
Luo Tao 已提交
34

M
minqiyang 已提交
35 36 37 38 39 40
# paddle_fluid_origin exclude inference api interface
if(WIN32)
  sep_library(paddle_fluid_origin DEPS ${fluid_modules} paddle_fluid_api)
else(WIN32)
  cc_library(paddle_fluid_origin DEPS ${fluid_modules} paddle_fluid_api)
endif(WIN32)
41

42
add_subdirectory(api)
43
add_subdirectory(capi)
44

45 46 47 48 49 50 51 52 53 54 55 56
if(WITH_TESTING)
  # tests/book depends the models that generated by python/paddle/fluid/tests/book
  add_subdirectory(tests/book)
  if(WITH_INFERENCE_API_TEST)
    add_subdirectory(tests/api)
  endif()
endif()

if(NOT ON_INFER)
  return()
endif()

57 58 59 60 61
if(WITH_MKLDNN)
	set(mkldnn_quantizer_src ${CMAKE_CURRENT_SOURCE_DIR}/api/mkldnn_quantizer.cc)
	set(mkldnn_quantizer_cfg mkldnn_quantizer_config)
endif()

N
nhzlx 已提交
62
set(STATIC_INFERENCE_APIS paddle_fluid_api paddle_inference_api analysis_predictor)
63 64 65
if (ANAKIN_FOUND)
    set(ANAKIN_SHARED_INFERENCE_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/api/api_anakin_engine.cc)
endif()
N
nhzlx 已提交
66
set(SHARED_INFERENCE_SRCS
M
mapingshuo 已提交
67
    io.cc ${CMAKE_CURRENT_SOURCE_DIR}/../framework/data_feed.cc ${CMAKE_CURRENT_SOURCE_DIR}/../framework/data_set.cc ${CMAKE_CURRENT_SOURCE_DIR}/../framework/data_feed_factory.cc ${CMAKE_CURRENT_SOURCE_DIR}/../framework/dataset_factory.cc ${CMAKE_CURRENT_SOURCE_DIR}/api/api.cc ${CMAKE_CURRENT_SOURCE_DIR}/api/api_impl.cc
N
nhzlx 已提交
68
    ${CMAKE_CURRENT_SOURCE_DIR}/api/analysis_predictor.cc
69
    ${mkldnn_quantizer_src}
70 71
    ${CMAKE_CURRENT_SOURCE_DIR}/api/details/zero_copy_tensor.cc
    ${ANAKIN_SHARED_INFERENCE_SRCS})
N
nhzlx 已提交
72

M
minqiyang 已提交
73 74
if(WIN32)
  sep_library(paddle_fluid DEPS ${fluid_modules} ${STATIC_INFERENCE_APIS} zero_copy_tensor reset_tensor_array
75
              analysis_config ${mkldnn_quantizer_cfg} paddle_pass_builder)
M
minqiyang 已提交
76 77
else(WIN32)
  cc_library(paddle_fluid DEPS ${fluid_modules} ${STATIC_INFERENCE_APIS}
78
            zero_copy_tensor reset_tensor_array analysis_config ${mkldnn_quantizer_cfg} paddle_pass_builder)
M
minqiyang 已提交
79
endif(WIN32)
N
nhzlx 已提交
80

81

M
minqiyang 已提交
82 83 84 85 86
if(NOT APPLE)
  # TODO(liuyiqu: Temporarily disable the link flag because it is not support on Mac.
  set(LINK_FLAGS "-Wl,--retain-symbols-file ${CMAKE_CURRENT_SOURCE_DIR}/paddle_fluid.sym")
  set_target_properties(paddle_fluid PROPERTIES LINK_FLAGS "${LINK_FLAGS}")
endif()
87

L
Luo Tao 已提交
88
# Create shared library
P
peizhilin 已提交
89 90
if(WIN32)
  sep_library(paddle_fluid_shared SHARED SRCS ${SHARED_INFERENCE_SRCS}
Y
Yiqun Liu 已提交
91
              DEPS ${fluid_modules} paddle_fluid_api reset_tensor_array
92
                  analysis_config ${mkldnn_quantizer_cfg} paddle_pass_builder)
P
peizhilin 已提交
93 94
else(WIN32)
  cc_library(paddle_fluid_shared SHARED SRCS ${SHARED_INFERENCE_SRCS}
95
            DEPS ${fluid_modules} paddle_fluid_api reset_tensor_array
96
                  analysis_config ${mkldnn_quantizer_cfg} paddle_pass_builder)
P
peizhilin 已提交
97
endif()
98

99 100
get_property(os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
target_link_libraries(paddle_fluid_shared ${os_dependency_modules})
Y
Yan Chunwei 已提交
101

L
Liu Yiqun 已提交
102
set_target_properties(paddle_fluid_shared PROPERTIES OUTPUT_NAME paddle_fluid)
P
peizhilin 已提交
103
if(NOT APPLE AND NOT WIN32)
Y
Update  
Yi Wang 已提交
104 105 106
  # TODO(liuyiqun): Temporarily disable the link flag because it is not support on Mac.
  set(LINK_FLAGS "-Wl,--version-script ${CMAKE_CURRENT_SOURCE_DIR}/paddle_fluid.map")
  set_target_properties(paddle_fluid_shared PROPERTIES LINK_FLAGS "${LINK_FLAGS}")
107 108 109 110 111 112 113 114 115 116 117 118
  # check symbol hidden
  FILE(WRITE ${CMAKE_CURRENT_BINARY_DIR}/check_symbol.cmake
    "execute_process(COMMAND bash -c \"${CMAKE_CURRENT_SOURCE_DIR}/check_symbol.sh"
    " ${CMAKE_CURRENT_BINARY_DIR}/libpaddle_fluid.so\" RESULT_VARIABLE symbol_res)\n"
    "if(NOT \"\${symbol_res}\" STREQUAL \"0\")\n"
    "  message(FATAL_ERROR \"Check symbol failed.\")\n"
    "endif()\n")
  add_custom_command(
    OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/.check_symbol"
    COMMAND ${CMAKE_COMMAND} -P "${CMAKE_CURRENT_BINARY_DIR}/check_symbol.cmake"
    DEPENDS paddle_fluid_shared)
  add_custom_target(check_symbol ALL DEPENDS "${CMAKE_CURRENT_BINARY_DIR}/.check_symbol")
Y
Update  
Yi Wang 已提交
119
endif()