Revert "Fix."

This reverts commit 46793a2.
dmlc · Jul 9, 2024 · f21ab04 · f21ab04
1 parent 46793a2
commit f21ab04
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/common/hist_util.cc b/src/common/hist_util.cc
@@ -328,10 +328,8 @@ void BuildHistDispatch(Span<GradientPair const> gpair, const RowSetCollection::E
       RowsWiseBuildHistKernel<false, BuildingManager>(
           gpair, Span{row_indices.begin(), row_indices.Size()}, gmat, hist);
     } else {
-      auto n = static_cast<std::size_t>(
-          std::distance(row_indices.begin(), row_indices.end() - no_prefetch_size));
-      auto span1 = Span{row_indices.begin(), n};
-      auto span2 = Span{row_indices.end() - no_prefetch_size, no_prefetch_size};
+      auto span1 = Span{row_indices.begin(), row_indices.end() - no_prefetch_size};
+      auto span2 = Span{row_indices.end() - no_prefetch_size, row_indices.end()};
 
       RowsWiseBuildHistKernel<true, BuildingManager>(gpair, span1, gmat, hist);
       // no prefetching to avoid loading extra memory