89da9f9741
- Collect all federated test files into the same directory. - Independently list the files.
21 lines
870 B
CMake
21 lines
870 B
CMake
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)
|