CMakeLists.txt 3.8 KB
Newer Older
1 2 3 4 5 6 7
# analysis and tensorrt must be added before creating static library,
# otherwise, there would be undefined reference to them in static library.
add_subdirectory(analysis)
if (TENSORRT_FOUND)
  add_subdirectory(tensorrt)
endif()

L
Luo Tao 已提交
8
set(FLUID_CORE_MODULES proto_desc memory lod_tensor executor)
9

X
Xin Pan 已提交
10
# TODO(panyx0718): Should this be called paddle_fluid_inference_api_internal?
11
cc_library(paddle_fluid_api
12
    SRCS io.cc
L
luotao1 已提交
13
    DEPS ${FLUID_CORE_MODULES} ${GLOB_OP_LIB}) 
14

15
get_property(fluid_modules GLOBAL PROPERTY FLUID_MODULES)
P
peizhilin 已提交
16
get_property(cuda_modules GLOBAL PROPERTY CUDA_MODULES)
L
Luo Tao 已提交
17

18
# paddle_fluid_origin exclude inference api interface
P
peizhilin 已提交
19 20
if(WIN32)
  sep_library(paddle_fluid_origin DEPS ${fluid_modules} paddle_fluid_api)
P
peizhilin 已提交
21 22 23
  if(WITH_GPU AND NOT WITH_DSO)
    target_link_libraries(paddle_fluid_origin ${cuda_modules})
  endif(WITH_GPU AND NOT WITH_DSO)
P
peizhilin 已提交
24 25 26
else(WIN32)
  cc_library(paddle_fluid_origin DEPS ${fluid_modules} paddle_fluid_api)
endif(WIN32)
27

28
add_subdirectory(api)
29

N
nhzlx 已提交
30 31 32 33 34 35 36 37 38 39
set(STATIC_INFERENCE_APIS paddle_fluid_api paddle_inference_api analysis_predictor)
set(SHARED_INFERENCE_SRCS
    io.cc ${CMAKE_CURRENT_SOURCE_DIR}/api/api.cc ${CMAKE_CURRENT_SOURCE_DIR}/api/api_impl.cc
    ${CMAKE_CURRENT_SOURCE_DIR}/api/analysis_predictor.cc
    ${CMAKE_CURRENT_SOURCE_DIR}/api/details/zero_copy_tensor.cc)
if (WITH_GPU AND TENSORRT_FOUND)
  set(STATIC_INFERENCE_APIS ${STATIC_INFERENCE_APIS} paddle_inference_tensorrt_subgraph_engine)
  set(SHARED_INFERENCE_SRCS ${SHARED_INFERENCE_SRCS} ${CMAKE_CURRENT_SOURCE_DIR}/api/api_tensorrt_subgraph_engine.cc)
endif()

Y
Yan Chunwei 已提交
40
# Create static library
P
peizhilin 已提交
41
if(WIN32)
P
peizhilin 已提交
42
  sep_library(paddle_fluid DEPS ${fluid_modules} ${STATIC_INFERENCE_APIS} zero_copy_tensor reset_tensor_array)
P
peizhilin 已提交
43 44 45
  if(WITH_GPU AND NOT WITH_DSO)
    target_link_libraries(paddle_fluid ${cuda_modules})
  endif(WITH_GPU AND NOT WITH_DSO)
P
peizhilin 已提交
46 47 48
else(WIN32)
  cc_library(paddle_fluid DEPS ${fluid_modules} ${STATIC_INFERENCE_APIS} zero_copy_tensor reset_tensor_array)
endif(WIN32)
N
nhzlx 已提交
49

50 51 52 53 54 55
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()

L
Luo Tao 已提交
56
# Create shared library
P
peizhilin 已提交
57 58 59
if(WIN32)
  sep_library(paddle_fluid_shared SHARED SRCS ${SHARED_INFERENCE_SRCS}
          DEPS ${fluid_modules} paddle_fluid_api reset_tensor_array)
P
peizhilin 已提交
60 61 62
  if(WITH_GPU AND NOT WITH_DSO)
    target_link_libraries(paddle_fluid_origin ${cuda_modules})
  endif(WITH_GPU AND NOT WITH_DSO)
P
peizhilin 已提交
63 64 65 66
else(WIN32)
  cc_library(paddle_fluid_shared SHARED SRCS ${SHARED_INFERENCE_SRCS}
      DEPS ${fluid_modules} paddle_fluid_api reset_tensor_array)
endif()
Y
Yan Chunwei 已提交
67

L
Liu Yiqun 已提交
68
set_target_properties(paddle_fluid_shared PROPERTIES OUTPUT_NAME paddle_fluid)
P
peizhilin 已提交
69
if(NOT APPLE AND NOT WIN32)
Y
Update  
Yi Wang 已提交
70 71 72
  # 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}")
73 74 75 76 77 78 79 80 81 82 83 84
  # 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 已提交
85
endif()
86

87
if(WITH_TESTING)
88
    # tests/book depends the models that generated by python/paddle/fluid/tests/book
89
  add_subdirectory(tests/book)
90 91 92
  if(WITH_INFERENCE_API_TEST)
    add_subdirectory(tests/api)
  endif()  
L
Luo Tao 已提交
93
endif()