bc995a4865
- Define a new data type, the proto file is copied for now. - Merge client and communicator into `FederatedColl`. - Define CUDA variant. - Migrate tests for CPU, add tests for CUDA.
57 lines
2.1 KiB
CMake
57 lines
2.1 KiB
CMake
# gRPC needs to be installed first. See README.md.
|
|
set(protobuf_MODULE_COMPATIBLE TRUE)
|
|
set(protobuf_BUILD_SHARED_LIBS TRUE)
|
|
find_package(Protobuf CONFIG REQUIRED)
|
|
find_package(gRPC CONFIG REQUIRED)
|
|
message(STATUS "Found gRPC: ${gRPC_CONFIG}")
|
|
|
|
# Generated code from the protobuf definition.
|
|
add_library(federated_proto STATIC federated.proto)
|
|
target_link_libraries(federated_proto PUBLIC protobuf::libprotobuf gRPC::grpc gRPC::grpc++)
|
|
target_include_directories(federated_proto PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
|
|
xgboost_target_properties(federated_proto)
|
|
|
|
protobuf_generate(
|
|
TARGET federated_proto
|
|
LANGUAGE cpp
|
|
PROTOC_OUT_DIR "${PROTO_BINARY_DIR}")
|
|
protobuf_generate(
|
|
TARGET federated_proto
|
|
LANGUAGE grpc
|
|
GENERATE_EXTENSIONS .grpc.pb.h .grpc.pb.cc
|
|
PLUGIN "protoc-gen-grpc=\$<TARGET_FILE:gRPC::grpc_cpp_plugin>"
|
|
PROTOC_OUT_DIR "${PROTO_BINARY_DIR}")
|
|
|
|
add_library(federated_old_proto STATIC federated.old.proto)
|
|
target_link_libraries(federated_old_proto PUBLIC protobuf::libprotobuf gRPC::grpc gRPC::grpc++)
|
|
target_include_directories(federated_old_proto PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
|
|
xgboost_target_properties(federated_old_proto)
|
|
|
|
protobuf_generate(
|
|
TARGET federated_old_proto
|
|
LANGUAGE cpp
|
|
PROTOC_OUT_DIR "${PROTO_BINARY_DIR}")
|
|
protobuf_generate(
|
|
TARGET federated_old_proto
|
|
LANGUAGE grpc
|
|
GENERATE_EXTENSIONS .grpc.pb.h .grpc.pb.cc
|
|
PLUGIN "protoc-gen-grpc=\$<TARGET_FILE:gRPC::grpc_cpp_plugin>"
|
|
PROTOC_OUT_DIR "${PROTO_BINARY_DIR}")
|
|
|
|
# Wrapper for the gRPC client.
|
|
add_library(federated_client INTERFACE)
|
|
target_sources(federated_client INTERFACE federated_client.h)
|
|
target_link_libraries(federated_client INTERFACE federated_proto)
|
|
target_link_libraries(federated_client INTERFACE federated_old_proto)
|
|
|
|
# Rabit engine for Federated Learning.
|
|
target_sources(
|
|
objxgboost PRIVATE federated_tracker.cc federated_server.cc federated_comm.cc federated_coll.cc
|
|
)
|
|
if(USE_CUDA)
|
|
target_sources(objxgboost PRIVATE federated_comm.cu federated_coll.cu)
|
|
endif()
|
|
|
|
target_link_libraries(objxgboost PRIVATE federated_client "-Wl,--exclude-libs,ALL")
|
|
target_compile_definitions(objxgboost PUBLIC -DXGBOOST_USE_FEDERATED=1)
|