fix init estimation on sycl device

Dmitry Razdoburdin · Dmitry Razdoburdin · commit 14a33607fd6e · 2025-03-11T04:54:57.000-07:00
diff --git 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());
 }

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ void FitStump(Context const* ctx, MetaInfo const& info, linalg::Matrix<GradientP`
`74`	`74`	`out->Reshape(n_targets);`
`75`	`75`
`76`	`76`	`gpair.SetDevice(ctx->Device());`
`77`		`- auto gpair_t = gpair.View(ctx->Device());`
	`77`	`+ auto gpair_t = gpair.View(ctx->Device().IsSycl() ? DeviceOrd::CPU() : ctx->Device());`
`78`	`78`	`ctx->IsCUDA() ? cuda_impl::FitStump(ctx, info, gpair_t, out->View(ctx->Device()))`
`79`	`79`	`: cpu_impl::FitStump(ctx, info, gpair_t, out->HostView());`
`80`	`80`	`}`