[sycl] add partitioning and related tests (#10080)

Co-authored-by: Dmitry Razdoburdin <>
2024-03-02 01:49:27 +08:00
parent 2c12b956da
commit 7a61216690
6 changed files with 371 additions and 21 deletions
@@ -280,7 +280,8 @@ class Predictor : public xgboost::Predictor {
                    uint32_t tree_end = 0) const override {
    ::sycl::queue qu = device_manager.GetQueue(ctx_->Device());
    // TODO(razdoburdin): remove temporary workaround after cache fix
-    sycl::DeviceMatrix device_matrix(qu, dmat);
+    sycl::DeviceMatrix device_matrix;
+    device_matrix.Init(qu, dmat);

    auto* out_preds = &predts->predictions;
    if (tree_end == 0) {