fix compilation errors; linting

razdoburdin · Nov 21, 2023 · 923ce39 · 923ce39
1 parent 7cf7e90
commit 923ce39
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/include/xgboost/context.h b/include/xgboost/context.h
@@ -251,7 +251,7 @@ struct Context : public XGBoostParameter<Context> {
         // Do not use the device name as this is likely an internal error, the name
         // wouldn't be valid.
         if (this->Device().IsSycl()) {
-          LOG(WARNING) << "The requested feature doesn't have SYCL specific implementation yet. " 
+          LOG(WARNING) << "The requested feature doesn't have SYCL specific implementation yet. "
                        << "CPU implementation is used";
           return cpu_fn();
         } else {

diff --git a/src/gbm/gbtree.cc b/src/gbm/gbtree.cc
@@ -834,7 +834,10 @@ class Dart : public GBTree {
             return gpu_predictor_->InplacePredict(p_fmat, model_, missing, &predts, i, i + 1);
           },
           [&] {
+            common::AssertSYCLSupport();
+#if defined(XGBOOST_USE_SYCL)
             return sycl_predictor_->InplacePredict(p_fmat, model_, missing, &predts, i, i + 1);
+#endif  // defined(XGBOOST_USE_SYCL)
           });
       CHECK(success) << msg;
     };
@@ -853,8 +856,11 @@ class Dart : public GBTree {
                                                        model_);
             },
             [&] {
+              common::AssertSYCLSupport();
+#if defined(XGBOOST_USE_SYCL)
               this->sycl_predictor_->InitOutPredictions(p_fmat->Info(), &p_out_preds->predictions,
                                                         model_);
+#endif  // defined(XGBOOST_USE_SYCL)
             });
       }
       // Multiple the tree weight