diff --git a/src/tree/updater_colmaker.cc b/src/tree/updater_colmaker.cc index 93bd0189e..fd612b04b 100644 --- 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;