32 lines
640 B
CMake
32 lines
640 B
CMake
option(DMLC_ROOT "Specify root of external dmlc core.")
|
|
|
|
add_library(allreduce_base "")
|
|
add_library(allreduce_mock "")
|
|
|
|
target_sources(
|
|
allreduce_base
|
|
PRIVATE
|
|
allreduce_base.cc
|
|
PUBLIC
|
|
${CMAKE_CURRENT_LIST_DIR}/allreduce_base.h
|
|
)
|
|
target_sources(
|
|
allreduce_mock
|
|
PRIVATE
|
|
allreduce_robust.cc
|
|
PUBLIC
|
|
${CMAKE_CURRENT_LIST_DIR}/allreduce_mock.h
|
|
)
|
|
|
|
target_include_directories(
|
|
allreduce_base
|
|
PUBLIC
|
|
${DMLC_ROOT}/include
|
|
${CMAKE_CURRENT_LIST_DIR}/../../include)
|
|
|
|
target_include_directories(
|
|
allreduce_mock
|
|
PUBLIC
|
|
${DMLC_ROOT}/include
|
|
${CMAKE_CURRENT_LIST_DIR}/../../include)
|