[sycl] Minimize host-device memory synchronization for ColumnSampler. (…

…#10884)
dmlc · Oct 14, 2024 · 78b82e4 · 78b82e4
1 parent 347bb14
commit 78b82e4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/common/random.h b/src/common/random.h
@@ -177,7 +177,10 @@ class ColumnSampler {
     }
     Reset();
 
-    feature_set_tree_->SetDevice(ctx->Device());
+    // We process ColumnSampler on host for SYCL. So don't need to push data to device
+    if (!ctx->Device().IsSycl()) {
+      feature_set_tree_->SetDevice(ctx->Device());
+    }
     feature_set_tree_->Resize(num_col);
     if (ctx->IsCUDA()) {
 #if defined(XGBOOST_USE_CUDA)