Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

revert

Browse Source
This commit is contained in:
Hendrik Groove 2024-10-22 01:21:22 +02:00
parent c78b0b752a
commit 9122d959e8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tree/gpu_hist/evaluate_splits.cu
View File

@ -507,7 +507,7 @@ GPUExpandEntry GPUHistEvaluator::EvaluateSingleSplit(Context const *ctx, Evaluat
shared_inputs, dh::ToSpan(out_entries)); shared_inputs, dh::ToSpan(out_entries));
GPUExpandEntry root_entry; GPUExpandEntry root_entry;
dh::safe_cuda(cudaMemcpy(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry), dh::safe_cuda(cudaMemcpyAsync(&root_entry, out_entries.data().get(), sizeof(GPUExpandEntry),
cudaMemcpyDeviceToHost)); cudaMemcpyDeviceToHost));
return root_entry; return root_entry;
} }