From d8cc93f3f23716274a05fa31c9f6c2ba5ce82cc0 Mon Sep 17 00:00:00 2001 From: amdsc21 <96135754+amdsc21@users.noreply.github.com> Date: Wed, 8 Mar 2023 06:38:35 +0100 Subject: [PATCH] enable rocm, fix algorithm.cuh --- src/common/algorithm.cuh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/common/algorithm.cuh b/src/common/algorithm.cuh index 53acc65e1..b1c5a4271 100644 --- a/src/common/algorithm.cuh +++ b/src/common/algorithm.cuh @@ -148,8 +148,13 @@ void SegmentedArgSort(Context const *ctx, Span values, Span group_ptr, sorted_idx_out.data().get(), sorted_idx.size(), n_groups, group_ptr.data(), group_ptr.data() + 1, ctx->CUDACtx()->Stream()); +#if defined(XGBOOST_USE_HIP) + dh::safe_cuda(hipMemcpyAsync(sorted_idx.data(), sorted_idx_out.data().get(), + sorted_idx.size_bytes(), hipMemcpyDeviceToDevice)); +#else dh::safe_cuda(cudaMemcpyAsync(sorted_idx.data(), sorted_idx_out.data().get(), sorted_idx.size_bytes(), cudaMemcpyDeviceToDevice)); +#endif } /**