Device dmatrix (#5420)

2020-03-28 14:42:21 +13:00
parent 780de49ddb
commit 13b10a6370
24 changed files with 915 additions and 310 deletions
--- a/src/common/hist_util.h
+++ b/src/common/hist_util.h
@@ -105,10 +105,10 @@ class HistogramCuts {
    auto end = cut_ptrs_.ConstHostVector().at(column_id + 1);
    const auto &values = cut_values_.ConstHostVector();
    auto it = std::upper_bound(values.cbegin() + beg, values.cbegin() + end, value);
-    if (it == values.cend()) {
-      it = values.cend() - 1;
-    }
    BinIdx idx = it - values.cbegin();
+    if (idx == end) {
+      idx -= 1;
+    }
    return idx;
  }