hide it.

Author: trivialfis

python-package/xgboost/testing/multi_target.py

@@ -29,6 +29,7 @@ def run_multiclass(device: Device, learning_rate: Optional[float]) -> None:
         128, n_features=12, n_informative=10, n_classes=4, random_state=2025
     )
     clf = XGBClassifier(
+        debug_synchronize=True,
         multi_strategy="multi_output_tree",
         callbacks=[ResetStrategy()],
         n_estimators=10,
@@ -50,6 +51,7 @@ def run_multilabel(device: Device, learning_rate: Optional[float]) -> None:
     # pylint: disable=unbalanced-tuple-unpacking
     X, y = make_multilabel_classification(128, random_state=2025)
     clf = XGBClassifier(
+        debug_synchronize=True,
         multi_strategy="multi_output_tree",
         callbacks=[ResetStrategy()],
         n_estimators=10,
@@ -114,6 +116,7 @@ def run_test(
         evals_result: Dict[str, Dict] = {}
         booster = train(
             {
+                "debug_synchronize": True,
                 "device": device,
                 "multi_strategy": "multi_output_tree",
                 "learning_rate": 1,

src/tree/updater_gpu_hist.cuh

@@ -467,7 +467,7 @@ class MultiTargetHistMaker {
 
   void GrowTree(linalg::Matrix<GradientPair>* split_gpair, DMatrix* p_fmat, ObjInfo const*,
                 RegTree* p_tree) {
-    if (this->param_.learning_rate - 1.0 != 0.0) {
+    if (!this->hist_param_->debug_synchronize) {
       LOG(FATAL) << "GPU" << MTNotImplemented();
     }
     Driver<MultiExpandEntry> driver{param_, kMaxNodeBatchSize};