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

Fix BinarySearchBin() argument types (#7026)

Browse Source
This commit is contained in:
TP Boudreau 2021-06-08 04:05:46 -07:00 committed by GitHub
parent 7beb2f7fae
commit bd2ca543c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/common/hist_util.h
View File

@ -295,13 +295,13 @@ struct GHistIndexMatrix {
}; };
template <typename GradientIndex> template <typename GradientIndex>
int32_t XGBOOST_HOST_DEV_INLINE BinarySearchBin(bst_uint begin, bst_uint end, int32_t XGBOOST_HOST_DEV_INLINE BinarySearchBin(size_t begin, size_t end,
GradientIndex const &data, GradientIndex const &data,
uint32_t const fidx_begin, uint32_t const fidx_begin,
uint32_t const fidx_end) { uint32_t const fidx_end) {
uint32_t previous_middle = std::numeric_limits<uint32_t>::max(); size_t previous_middle = std::numeric_limits<size_t>::max();
while (end != begin) { while (end != begin) {
auto middle = begin + (end - begin) / 2; size_t middle = begin + (end - begin) / 2;
if (middle == previous_middle) { if (middle == previous_middle) {
break; break;
} }