From c78b0b752a5a6146db511e7b7323e5e231bc78e8 Mon Sep 17 00:00:00 2001 From: Hendrik Groove Date: Tue, 22 Oct 2024 01:18:12 +0200 Subject: [PATCH] remove async --- 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 8e8023ac7..55762ff78 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(cudaMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), + dh::safe_cuda(cudaMemcpy(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), cudaMemcpyDeviceToHost)); return root_entry; }