33 lines
768 B
CMake
33 lines
768 B
CMake
find_package(GTest REQUIRED)
|
|
|
|
add_executable(
|
|
unit_tests
|
|
test_io.cc
|
|
test_utils.cc
|
|
allreduce_robust_test.cc
|
|
allreduce_base_test.cc
|
|
allreduce_mock_test.cc
|
|
test_main.cpp)
|
|
|
|
target_link_libraries(
|
|
unit_tests PRIVATE
|
|
GTest::GTest GTest::Main
|
|
rabit_base rabit_mock rabit)
|
|
|
|
target_include_directories(unit_tests PUBLIC
|
|
"$<BUILD_INTERFACE:${rabit_SOURCE_DIR}/include>"
|
|
"$<BUILD_INTERFACE:${DMLC_ROOT}/include>")
|
|
|
|
set_target_properties(unit_tests
|
|
PROPERTIES
|
|
CXX_STANDARD 11
|
|
CXX_STANDARD_REQUIRED ON
|
|
RUNTIME_OUTPUT_DIRECTORY ${rabit_BINARY_DIR}
|
|
RUNTIME_OUTPUT_DIRECTORY_DEBUG ${rabit_BINARY_DIR}
|
|
RUNTIME_OUTPUT_DIRECTORY_RELEASE ${rabit_BINARY_DIR})
|
|
|
|
add_test(
|
|
NAME TestRabitLib
|
|
COMMAND unit_tests
|
|
WORKING_DIRECTORY ${rabit_BINARY_DIR})
|