From 6b618205167d2d814c123fe2bd3fdc6fc4a516c3 Mon Sep 17 00:00:00 2001
From: Dmitry Razdoburdin <d.razdoburdin@gmail.com>
Date: Mon, 17 Mar 2025 11:19:22 +0100
Subject: [PATCH] Revert "fix init estimation on sycl device (#74)"

This reverts commit 148090321bb54f3099b6bbfdebeefddf6da41b10.
---
 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 144abcbd8131..8fdcb3131646 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().IsSycl() ? DeviceOrd::CPU() : ctx->Device());
+  auto gpair_t = gpair.View(ctx->Device());
   ctx->IsCUDA() ? cuda_impl::FitStump(ctx, info, gpair_t, out->View(ctx->Device()))
                 : cpu_impl::FitStump(ctx, info, gpair_t, out->HostView());
 }