a5a58102e5
This PR replaces the original RABIT implementation with a new one, which has already been partially merged into XGBoost. The new one features: - Federated learning for both CPU and GPU. - NCCL. - More data types. - A unified interface for all the underlying implementations. - Improved timeout handling for both tracker and workers. - Exhausted tests with metrics (fixed a couple of bugs along the way). - A reusable tracker for Python and JVM packages.
21 lines
488 B
C++
21 lines
488 B
C++
/**
|
|
* Copyright 2016-2024, XGBoost Contributors
|
|
*/
|
|
#include <gtest/gtest.h>
|
|
#include <xgboost/base.h>
|
|
#include <xgboost/logging.h>
|
|
|
|
#include <string>
|
|
|
|
#include "helpers.h"
|
|
|
|
int main(int argc, char** argv) {
|
|
xgboost::Args args{{"verbosity", "2"}};
|
|
xgboost::ConsoleLogger::Configure(args);
|
|
|
|
testing::InitGoogleTest(&argc, argv);
|
|
testing::FLAGS_gtest_death_test_style = "threadsafe";
|
|
auto rmm_alloc = xgboost::SetUpRMMResourceForCppTests(argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|