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; }