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::MatrixReshape(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()); }