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

dmlc · Jul 7, 2021 · 689eb8f · 689eb8f
1 parent 615ab2b
commit 689eb8f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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;