138 lines
4.2 KiB
CMake
138 lines
4.2 KiB
CMake
cmake_minimum_required(VERSION 3.0)
|
|
|
|
project(rabit VERSION 0.2.0)
|
|
|
|
option(RABIT_BUILD_TESTS "Build rabit tests" OFF)
|
|
option(RABIT_BUILD_MPI "Build MPI" OFF)
|
|
option(RABIT_BUILD_DMLC "Include DMLC_CORE in build" ON)
|
|
|
|
add_library(rabit src/allreduce_base.cc src/allreduce_robust.cc src/engine.cc src/c_api.cc)
|
|
add_library(rabit_base src/allreduce_base.cc src/engine_base.cc src/c_api.cc)
|
|
add_library(rabit_empty src/engine_empty.cc src/c_api.cc)
|
|
|
|
set(rabit_libs rabit rabit_base rabit_empty)
|
|
|
|
set_target_properties(rabit rabit_base rabit_empty PROPERTIES CXX_STANDARD 11 CXX_STANDARD_REQUIRED ON)
|
|
|
|
if(RABIT_BUILD_MPI)
|
|
find_package(MPI REQUIRED)
|
|
add_library(rabit_mpi src/engine_mpi.cc ${MPI_INCLUDE_PATH})
|
|
target_link_libraries(rabit_mpi ${MPI_CXX_LIBRARIES})
|
|
list(APPEND rabit_libs rabit_mpi)
|
|
endif()
|
|
|
|
if(RABIT_BUILD_TESTS)
|
|
# Use static so that rabit_mock won't be installed when building shared libraries.
|
|
add_library(rabit_mock STATIC src/allreduce_base.cc src/allreduce_robust.cc src/engine_mock.cc src/c_api.cc)
|
|
list(APPEND rabit_libs rabit_mock) # add to list to apply build settings, then remove
|
|
endif()
|
|
|
|
if(RABIT_BUILD_DMLC)
|
|
foreach(lib ${rabit_libs})
|
|
#include "./internal/utils.h"
|
|
target_include_directories(${lib} PUBLIC
|
|
"$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/include>"
|
|
"$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/dmlc-core/include>"
|
|
)
|
|
endforeach()
|
|
else()
|
|
foreach(lib ${rabit_libs})
|
|
#include "./internal/utils.h"
|
|
target_include_directories(${lib} PUBLIC
|
|
"$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/include>"
|
|
"$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/../dmlc_core/include>"
|
|
)
|
|
endforeach()
|
|
endif()
|
|
|
|
if(RABIT_BUILD_TESTS)
|
|
list(REMOVE_ITEM rabit_libs "rabit_mock") # remove here to avoid installing it
|
|
set(tests speed_test lazy_recover local_recover model_recover)
|
|
foreach(test ${tests})
|
|
add_executable(${test} test/${test}.cc)
|
|
target_link_libraries(${test} rabit_mock)
|
|
install(TARGETS ${test} DESTINATION bin)
|
|
endforeach()
|
|
if(RABIT_BUILD_MPI)
|
|
add_executable(speed_test_mpi test/speed_test.cc)
|
|
target_link_libraries(speed_test_mpi rabit_mpi)
|
|
install(TARGETS speed_test_mpi DESTINATION bin)
|
|
endif()
|
|
endif()
|
|
|
|
# Installation (https://github.com/forexample/package-example) {
|
|
|
|
# Layout. This works for all platforms:
|
|
# * <prefix>/lib/cmake/<PROJECT-NAME>
|
|
# * <prefix>/lib/
|
|
# * <prefix>/include/
|
|
set(config_install_dir "lib/cmake/${PROJECT_NAME}")
|
|
set(include_install_dir "include")
|
|
|
|
set(generated_dir "${CMAKE_CURRENT_BINARY_DIR}/generated")
|
|
|
|
# Configuration
|
|
set(version_config "${generated_dir}/${PROJECT_NAME}ConfigVersion.cmake")
|
|
set(project_config "${generated_dir}/${PROJECT_NAME}Config.cmake")
|
|
set(TARGETS_EXPORT_NAME "${PROJECT_NAME}Targets")
|
|
set(namespace "${PROJECT_NAME}::")
|
|
|
|
# Include module with fuction 'write_basic_package_version_file'
|
|
include(CMakePackageConfigHelpers)
|
|
|
|
# Configure '<PROJECT-NAME>ConfigVersion.cmake'
|
|
# Use:
|
|
# * PROJECT_VERSION
|
|
write_basic_package_version_file(
|
|
"${version_config}" COMPATIBILITY SameMajorVersion
|
|
)
|
|
|
|
# Configure '<PROJECT-NAME>Config.cmake'
|
|
# Use variables:
|
|
# * TARGETS_EXPORT_NAME
|
|
# * PROJECT_NAME
|
|
configure_package_config_file(
|
|
"cmake/Config.cmake.in"
|
|
"${project_config}"
|
|
INSTALL_DESTINATION "${config_install_dir}"
|
|
)
|
|
|
|
# Targets:
|
|
# * <prefix>/lib/librabit.a
|
|
# * <prefix>/lib/librabit_base
|
|
# * <prefix>/lib/librabit_empty
|
|
# * header location after install: <prefix>/include/rabit/rabit.h
|
|
# * headers can be included by C++ code `#include <rabit/rabit.h>`
|
|
install(
|
|
TARGETS ${rabit_libs}
|
|
EXPORT "${TARGETS_EXPORT_NAME}"
|
|
LIBRARY DESTINATION "lib"
|
|
ARCHIVE DESTINATION "lib"
|
|
RUNTIME DESTINATION "bin"
|
|
INCLUDES DESTINATION "${include_install_dir}"
|
|
)
|
|
|
|
# Headers:
|
|
install(
|
|
DIRECTORY "include/"
|
|
DESTINATION "${include_install_dir}"
|
|
FILES_MATCHING PATTERN "*.h"
|
|
)
|
|
|
|
# Config
|
|
# * <prefix>/lib/cmake/rabit/rabitConfig.cmake
|
|
# * <prefix>/lib/cmake/rabit/rabitConfigVersion.cmake
|
|
install(
|
|
FILES "${project_config}" "${version_config}"
|
|
DESTINATION "${config_install_dir}"
|
|
)
|
|
|
|
# Config
|
|
# * <prefix>/lib/cmake/Foo/FooTargets.cmake
|
|
install(
|
|
EXPORT "${TARGETS_EXPORT_NAME}"
|
|
NAMESPACE "${namespace}"
|
|
DESTINATION "${config_install_dir}"
|
|
)
|
|
# }
|