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

[sycl] optimise hist building (#10311)

Browse Source 
Co-authored-by: Dmitry Razdoburdin <>
This commit is contained in:
Dmitry Razdoburdin
2024-05-27 04:21:22 +02:00
committed by GitHub
parent 9def441e9a
commit 0058301e6f
3 changed files with 54 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugin/sycl/tree/hist_updater.cc
View File

@@ -136,10 +136,7 @@ void HistUpdater<GradientSumT>::InitData(
hist_buffer_.Init(qu_, nbins);
size_t buffer_size = kBufferSize;
if (buffer_size > info.num_row_ / kMinBlockSize + 1) {
buffer_size = info.num_row_ / kMinBlockSize + 1;
}
hist_buffer_.Reset(buffer_size);
hist_buffer_.Reset(kBufferSize);
// initialize histogram builder
hist_builder_ = common::GHistBuilder<GradientSumT>(qu_, nbins);

1
plugin/sycl/tree/hist_updater.h
View File

@@ -130,7 +130,6 @@ class HistUpdater {
DataLayout data_layout_;
constexpr static size_t kBufferSize = 2048;
constexpr static size_t kMinBlockSize = 128;
common::GHistBuilder<GradientSumT> hist_builder_;
common::ParallelGHistBuilder<GradientSumT> hist_buffer_;
/*! \brief culmulative histogram of gradients. */