From 9122d959e862b2077100390ed6b82ff6e61200c8 Mon Sep 17 00:00:00 2001 From: Hendrik Groove Date: Tue, 22 Oct 2024 01:21:22 +0200 Subject: [PATCH] revert --- src/tree/gpu_hist/evaluate_splits.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tree/gpu_hist/evaluate_splits.cu b/src/tree/gpu_hist/evaluate_splits.cu index 55762ff78..8e8023ac7 100644 --- a/src/tree/gpu_hist/evaluate_splits.cu +++ b/src/tree/gpu_hist/evaluate_splits.cu @@ -507,7 +507,7 @@ GPUExpandEntry GPUHistEvaluator::EvaluateSingleSplit(Context const *ctx, Evaluat shared_inputs, dh::ToSpan(out_entries)); GPUExpandEntry root_entry; - dh::safe_cuda(cudaMemcpy(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), + dh::safe_cuda(cudaMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), cudaMemcpyDeviceToHost)); return root_entry; }