Skip to content

Commit 4c378d8

Browse files
committed
fix.
1 parent c466d05 commit 4c378d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/gbm/gbtree.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -636,7 +636,7 @@ GBTree::GetPredictor(HostDeviceVector<float> const *out_pred,
636636

637637
// Data comes from SparsePageDMatrix. Since we are loading data in pages, no need to
638638
// prevent data copy.
639-
if (!f_dmat->SingleColBlock()) {
639+
if (f_dmat && !f_dmat->SingleColBlock()) {
640640
if (ctx_->IsCPU()) {
641641
return cpu_predictor_;
642642
} else {

0 commit comments

Comments
 (0)