fix.

dmlc · Jun 19, 2023 · 8be4326 · 8be4326
1 parent 7aeb305
commit 8be4326
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gbm/gbtree.cc b/src/gbm/gbtree.cc
@@ -636,7 +636,7 @@ GBTree::GetPredictor(HostDeviceVector<float> const *out_pred,
 
   // Data comes from SparsePageDMatrix. Since we are loading data in pages, no need to
   // prevent data copy.
-  if (!f_dmat->SingleColBlock()) {
+  if (f_dmat && !f_dmat->SingleColBlock()) {
     if (ctx_->IsCPU()) {
       return cpu_predictor_;
     } else {