From 59cc2832422e6cae6dd309c1c55a77b27cb5ac97 Mon Sep 17 00:00:00 2001 From: Hendrik Groove Date: Tue, 22 Oct 2024 00:46:07 +0200 Subject: [PATCH] use hip functions --- src/tree/gpu_hist/evaluate_splits.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tree/gpu_hist/evaluate_splits.cu b/src/tree/gpu_hist/evaluate_splits.cu index 8e8023ac7..acce2c4f8 100644 --- a/src/tree/gpu_hist/evaluate_splits.cu +++ b/src/tree/gpu_hist/evaluate_splits.cu @@ -507,8 +507,8 @@ 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), - cudaMemcpyDeviceToHost)); + dh::safe_cuda(hipMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), + hipMemcpyDeviceToHost)); return root_entry; } } // namespace xgboost::tree \ No newline at end of file