Check external memory support for exact tree method. (#7088)

2021-07-08 02:12:57 +08:00
parent 615ab2b03e
commit 689eb8f620
1 changed files with 4 additions and 0 deletions
--- a/src/tree/updater_colmaker.cc
+++ b/src/tree/updater_colmaker.cc
@@ -98,6 +98,10 @@ class ColMaker: public TreeUpdater {
      LOG(FATAL) << "Updater `grow_colmaker` or `exact` tree method doesn't "
                    "support distributed training.";
    }
+    if (!dmat->SingleColBlock()) {
+      LOG(FATAL) << "Updater `grow_colmaker` or `exact` tree method doesn't "
+                    "support external memory training.";
+    }
    this->LazyGetColumnDensity(dmat);
    // rescale learning rate according to size of trees
    float lr = param_.learning_rate;