From 2bbb8b3786f81cbe3dba27aa7cdc8d93be0164d6 Mon Sep 17 00:00:00 2001 From: Hendrik Groove Date: Tue, 22 Oct 2024 00:58:01 +0200 Subject: [PATCH] revert --- src/tree/gpu_hist/evaluate_splits.cu | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/tree/gpu_hist/evaluate_splits.cu b/src/tree/gpu_hist/evaluate_splits.cu index 35d68e68a..8e8023ac7 100644 --- a/src/tree/gpu_hist/evaluate_splits.cu +++ b/src/tree/gpu_hist/evaluate_splits.cu @@ -423,9 +423,9 @@ void GPUHistEvaluator::CopyToHost(const std::vector &nidx) { for (auto idx : nidx) { copy_stream_.View().Wait(event); - dh::safe_cuda(hipMemcpyAsync( + dh::safe_cuda(cudaMemcpyAsync( h_cats.GetNodeCatStorage(idx).data(), d_cats.GetNodeCatStorage(idx).data(), - d_cats.GetNodeCatStorage(idx).size_bytes(), hipMemcpyDeviceToHost, copy_stream_.View())); + d_cats.GetNodeCatStorage(idx).size_bytes(), cudaMemcpyDeviceToHost, copy_stream_.View())); } } @@ -507,8 +507,8 @@ GPUExpandEntry GPUHistEvaluator::EvaluateSingleSplit(Context const *ctx, Evaluat shared_inputs, dh::ToSpan(out_entries)); GPUExpandEntry root_entry; - dh::safe_cuda(hipMemcpy(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), - hipMemcpyDeviceToHost)); + dh::safe_cuda(cudaMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), + cudaMemcpyDeviceToHost)); return root_entry; } } // namespace xgboost::tree \ No newline at end of file