diff --git a/src/tree/gpu_hist/evaluate_splits.cu b/src/tree/gpu_hist/evaluate_splits.cu index 98811741a..35d68e68a 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(hipMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), + dh::safe_cuda(hipMemcpy(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), hipMemcpyDeviceToHost)); return root_entry; }