|
|
|
@ -96,6 +96,20 @@ if(NOT APPLE AND NOT ANDROID)
|
|
|
|
|
set(CMAKE_CXX_LINK_EXECUTABLE "${CMAKE_CXX_LINK_EXECUTABLE} -pthread -ldl -lrt")
|
|
|
|
|
endif(NOT APPLE AND NOT ANDROID)
|
|
|
|
|
|
|
|
|
|
set_property(GLOBAL PROPERTY FLUID_MODULES "")
|
|
|
|
|
# find all fluid modules is used for paddle fluid static library
|
|
|
|
|
# for building inference libs
|
|
|
|
|
function(find_fluid_modules TARGET_NAME)
|
|
|
|
|
get_filename_component(__target_path ${TARGET_NAME} ABSOLUTE)
|
|
|
|
|
string(REGEX REPLACE "^${PADDLE_SOURCE_DIR}/" "" __target_path ${__target_path})
|
|
|
|
|
string(FIND "${__target_path}" "fluid" pos)
|
|
|
|
|
if(pos GREATER 1)
|
|
|
|
|
get_property(fluid_modules GLOBAL PROPERTY FLUID_MODULES)
|
|
|
|
|
set(fluid_modules ${fluid_modules} ${TARGET_NAME})
|
|
|
|
|
set_property(GLOBAL PROPERTY FLUID_MODULES "${fluid_modules}")
|
|
|
|
|
endif()
|
|
|
|
|
endfunction(find_fluid_modules)
|
|
|
|
|
|
|
|
|
|
function(merge_static_libs TARGET_NAME)
|
|
|
|
|
set(libs ${ARGN})
|
|
|
|
|
list(REMOVE_DUPLICATES libs)
|
|
|
|
@ -195,6 +209,15 @@ function(cc_library TARGET_NAME)
|
|
|
|
|
list(REMOVE_ITEM cc_library_DEPS warpctc)
|
|
|
|
|
add_dependencies(${TARGET_NAME} warpctc)
|
|
|
|
|
endif()
|
|
|
|
|
# Only deps libmklml.so, not link
|
|
|
|
|
if("${cc_library_DEPS};" MATCHES "mklml;")
|
|
|
|
|
list(REMOVE_ITEM cc_library_DEPS mklml)
|
|
|
|
|
if(NOT "${TARGET_NAME}" MATCHES "dynload_mklml")
|
|
|
|
|
list(APPEND cc_library_DEPS dynload_mklml)
|
|
|
|
|
endif()
|
|
|
|
|
add_dependencies(${TARGET_NAME} mklml)
|
|
|
|
|
target_link_libraries(${TARGET_NAME} "-L${MKLML_LIB_DIR} -liomp5 -Wl,--as-needed")
|
|
|
|
|
endif()
|
|
|
|
|
target_link_libraries(${TARGET_NAME} ${cc_library_DEPS})
|
|
|
|
|
add_dependencies(${TARGET_NAME} ${cc_library_DEPS})
|
|
|
|
|
endif()
|
|
|
|
@ -234,13 +257,14 @@ function(cc_test TARGET_NAME)
|
|
|
|
|
set(multiValueArgs SRCS DEPS ARGS)
|
|
|
|
|
cmake_parse_arguments(cc_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
|
|
|
|
|
add_executable(${TARGET_NAME} ${cc_test_SRCS})
|
|
|
|
|
target_link_libraries(${TARGET_NAME} ${cc_test_DEPS} paddle_gtest_main memory gtest gflags glog)
|
|
|
|
|
add_dependencies(${TARGET_NAME} ${cc_test_DEPS} paddle_gtest_main memory gtest gflags glog)
|
|
|
|
|
target_link_libraries(${TARGET_NAME} ${cc_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
|
|
|
|
|
add_dependencies(${TARGET_NAME} ${cc_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
|
|
|
|
|
add_test(NAME ${TARGET_NAME}
|
|
|
|
|
COMMAND ${TARGET_NAME} ${cc_test_ARGS}
|
|
|
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
|
if (${cc_test_SERIAL})
|
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY SERIAL 1)
|
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_init_allocated_mem=true)
|
|
|
|
|
endif()
|
|
|
|
|
endif()
|
|
|
|
|
endfunction(cc_test)
|
|
|
|
@ -300,11 +324,12 @@ function(nv_test TARGET_NAME)
|
|
|
|
|
set(multiValueArgs SRCS DEPS)
|
|
|
|
|
cmake_parse_arguments(nv_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
|
|
|
|
|
cuda_add_executable(${TARGET_NAME} ${nv_test_SRCS})
|
|
|
|
|
target_link_libraries(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main memory gtest gflags glog)
|
|
|
|
|
add_dependencies(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main memory gtest gflags glog)
|
|
|
|
|
target_link_libraries(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
|
|
|
|
|
add_dependencies(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
|
|
|
|
|
add_test(${TARGET_NAME} ${TARGET_NAME})
|
|
|
|
|
if (nv_test_SERIAL)
|
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY SERIAL 1)
|
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_init_allocated_mem=true)
|
|
|
|
|
endif()
|
|
|
|
|
endif()
|
|
|
|
|
endfunction(nv_test)
|
|
|
|
@ -552,7 +577,7 @@ function(py_test TARGET_NAME)
|
|
|
|
|
set(multiValueArgs SRCS DEPS ARGS ENVS)
|
|
|
|
|
cmake_parse_arguments(py_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
|
|
|
|
|
add_test(NAME ${TARGET_NAME}
|
|
|
|
|
COMMAND env PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_ENVS}
|
|
|
|
|
COMMAND env FLAGS_init_allocated_mem=true PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_ENVS}
|
|
|
|
|
${PYTHON_EXECUTABLE} -u ${py_test_SRCS} ${py_test_ARGS}
|
|
|
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
|
endif()
|
|
|
|
|