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

Wide dataset quantile performance improvement (#5306)

Browse Source
This commit is contained in:
Rory Mitchell
2020-02-16 10:24:42 +13:00
committed by GitHub
parent ed2465cce4
commit 7e32af5c21
4 changed files with 23 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/hist_util.cu
View File

@@ -148,7 +148,7 @@ class GPUSketcher {
this->SketchBatch(batch, info);
}
hmat->Init(&sketch_container_->sketches_, max_bin_);
hmat->Init(&sketch_container_->sketches_, max_bin_, info.num_row_);
return row_stride_;
}