From 581ffb4b9379bffa19d8bb2c00bc0ea17ee4bef8 Mon Sep 17 00:00:00 2001
From: Dmitry Razdoburdin <>
Date: Thu, 30 Nov 2023 05:02:15 -0800
Subject: [PATCH] mark InitOutPrediction virtual

---
 include/xgboost/predictor.h        | 4 ++--
 plugin/sycl/predictor/predictor.cc | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/include/xgboost/predictor.h b/include/xgboost/predictor.h
index 2c69cf648392..813e111104f4 100644
--- a/include/xgboost/predictor.h
+++ b/include/xgboost/predictor.h
@@ -92,8 +92,8 @@ class Predictor {
    * \param out_predt Prediction vector to be initialized.
    * \param model Tree model used for prediction.
    */
-  void InitOutPredictions(const MetaInfo& info, HostDeviceVector<bst_float>* out_predt,
-                          const gbm::GBTreeModel& model) const;
+  virtual void InitOutPredictions(const MetaInfo& info, HostDeviceVector<bst_float>* out_predt,
+                                  const gbm::GBTreeModel& model) const;
 
   /**
    * \brief Generate batch predictions for a given feature matrix. May use
diff --git a/plugin/sycl/predictor/predictor.cc b/plugin/sycl/predictor/predictor.cc
index 16dd46a8bd0e..3ceb99f1e1c8 100755
--- a/plugin/sycl/predictor/predictor.cc
+++ b/plugin/sycl/predictor/predictor.cc
@@ -233,7 +233,7 @@ class Predictor : public xgboost::Predictor {
  protected:
   void InitOutPredictions(const MetaInfo& info,
                           HostDeviceVector<bst_float>* out_preds,
-                          const gbm::GBTreeModel& model) const {
+                          const gbm::GBTreeModel& model) const override {
     CHECK_NE(model.learner_model_param->num_output_group, 0);
     size_t n = model.learner_model_param->num_output_group * info.num_row_;
     const auto& base_margin = info.base_margin_.Data()->HostVector();