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

Use Span in GPU exact updater. (#4020)

Browse Source 
* Use Span in GPU exact updater.

* Add a small test.
This commit is contained in:
Jiaming Yuan
2018-12-26 12:44:46 +08:00
committed by GitHub
parent 7735252925
commit 9897b5042f
4 changed files with 180 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/tree/updater_gpu_hist.cu
View File

@@ -62,7 +62,7 @@ DMLC_REGISTER_PARAMETER(GPUHistMakerTrainParam);
*/
template <int BLOCK_THREADS, typename ReduceT, typename TempStorageT, typename GradientSumT>
__device__ GradientSumT ReduceFeature(common::Span<const GradientSumT> feature_histogram,
TempStorageT* temp_storage) {
TempStorageT* temp_storage) {
__shared__ cub::Uninitialized<GradientSumT> uninitialized_sum;
GradientSumT& shared_sum = uninitialized_sum.Alias();