Add tests for gpu_approx (#9553)

2023-09-07 02:21:58 -07:00
parent 6c791b5b47
commit 66a0832778
2 changed files with 60 additions and 5 deletions
--- a/tests/cpp/plugin/test_federated_learner.cc
+++ b/tests/cpp/plugin/test_federated_learner.cc
@@ -120,6 +120,11 @@ TEST_P(VerticalFederatedLearnerTest, Hist) {
 }

 #if defined(XGBOOST_USE_CUDA)
+TEST_P(VerticalFederatedLearnerTest, GPUApprox) {
+  std::string objective = GetParam();
+  this->Run("approx", "cuda:0", objective);
+}
+
 TEST_P(VerticalFederatedLearnerTest, GPUHist) {
  std::string objective = GetParam();
  this->Run("hist", "cuda:0", objective);