1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
target_sources(
testxgboost PRIVATE
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_coll.cc
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_comm.cc
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_comm_group.cc
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_tracker.cc
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_learner.cc
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_data.cc
)
if(USE_CUDA)
target_sources(
testxgboost PRIVATE
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_coll.cu
${xgboost_SOURCE_DIR}/tests/cpp/plugin/federated/test_federated_comm_group.cu
)
endif()
target_include_directories(testxgboost PRIVATE ${xgboost_SOURCE_DIR}/plugin/federated)
target_link_libraries(testxgboost PRIVATE federated_client)
|
