add zero buffer check to predictor

razdoburdin · Nov 17, 2023 · d0e086c · d0e086c
1 parent 4546a23
commit d0e086c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugin/sycl/predictor/predictor.cc b/plugin/sycl/predictor/predictor.cc
@@ -184,9 +184,9 @@ void DevicePredictInternal(::sycl::queue qu,
                            const gbm::GBTreeModel& model,
                            size_t tree_begin,
                            size_t tree_end) {
-  if (tree_end - tree_begin == 0) {
-    return;
-  }
+  if (tree_end - tree_begin == 0) return;
+  if (out_preds->HostVector().size() == 0) return;
+
   DeviceModel device_model;
   device_model.Init(qu, model, tree_begin, tree_end);