fix merge

razdoburdin · Nov 25, 2024 · 7c22f22 · 7c22f22
1 parent 107005b
commit 7c22f22
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/src/objective/regression_obj.cu b/src/objective/regression_obj.cu
@@ -121,6 +121,7 @@ class RegLossObj : public FitInterceptGlmLike {
     if (iter == 0) {
       ValidateLabel(info);
     }
+
     size_t const ndata = preds.Size();
     out_gpair->SetDevice(ctx_->Device());
     auto device = ctx_->Device();

diff --git a/tests/cpp/objective/test_regression_obj.cc b/tests/cpp/objective/test_regression_obj.cc
@@ -6,6 +6,8 @@
 #include <xgboost/json.h>
 #include <xgboost/objective.h>
 
+#include <numeric>  // for iota
+
 #include "../../../src/common/linalg_op.h"  // for begin, end
 #include "../../../src/objective/adaptive.h"
 #include "../../../src/tree/param.h"        // for TrainParam

diff --git a/tests/cpp/objective/test_regression_obj_cpu.cc b/tests/cpp/objective/test_regression_obj_cpu.cc
@@ -5,8 +5,6 @@
 #include <xgboost/context.h>
 #include <xgboost/objective.h>
 
-#include <numeric>  // for iota
-
 #include "../../../src/objective/adaptive.h"
 #include "../../../src/tree/param.h"  // for TrainParam
 #include "../helpers.h"