668b8a0ea4
* Switch from rabit to the collective communicator * fix size_t specialization * really fix size_t * try again * add include * more include * fix lint errors * remove rabit includes * fix pylint error * return dict from communicator context * fix communicator shutdown * fix dask test * reset communicator mocklist * fix distributed tests * do not save device communicator * fix jvm gpu tests * add python test for federated communicator * Update gputreeshap submodule Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
18 lines
510 B
CMake
18 lines
510 B
CMake
cmake_minimum_required(VERSION 3.3)
|
|
|
|
find_package(Threads REQUIRED)
|
|
|
|
set(RABIT_SOURCES
|
|
${CMAKE_CURRENT_LIST_DIR}/src/allreduce_base.cc
|
|
${CMAKE_CURRENT_LIST_DIR}/src/rabit_c_api.cc)
|
|
|
|
if (RABIT_BUILD_MPI)
|
|
list(APPEND RABIT_SOURCES ${CMAKE_CURRENT_LIST_DIR}/src/engine_mpi.cc)
|
|
elseif (RABIT_MOCK)
|
|
list(APPEND RABIT_SOURCES ${CMAKE_CURRENT_LIST_DIR}/src/engine_mock.cc)
|
|
else ()
|
|
list(APPEND RABIT_SOURCES ${CMAKE_CURRENT_LIST_DIR}/src/engine.cc)
|
|
endif ()
|
|
|
|
set(RABIT_SOURCES ${RABIT_SOURCES} PARENT_SCOPE)
|