From 14a33607fd6ed72e6766b0e97eaa4a414204104c Mon Sep 17 00:00:00 2001
From: Dmitry Razdoburdin <>
Date: Tue, 11 Mar 2025 04:54:57 -0700
Subject: [PATCH] fix init estimation on sycl device

---
 src/tree/fit_stump.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tree/fit_stump.cc b/src/tree/fit_stump.cc
index 8fdcb3131646..144abcbd8131 100644
--- a/src/tree/fit_stump.cc
+++ b/src/tree/fit_stump.cc
@@ -74,7 +74,7 @@ void FitStump(Context const* ctx, MetaInfo const& info, linalg::Matrix<GradientP
   out->Reshape(n_targets);
 
   gpair.SetDevice(ctx->Device());
-  auto gpair_t = gpair.View(ctx->Device());
+  auto gpair_t = gpair.View(ctx->Device().IsSycl() ? DeviceOrd::CPU() : ctx->Device());
   ctx->IsCUDA() ? cuda_impl::FitStump(ctx, info, gpair_t, out->View(ctx->Device()))
                 : cpu_impl::FitStump(ctx, info, gpair_t, out->HostView());
 }