Fix empty partition. (dmlc#10559)

trivialfis · Jul 15, 2024 · d588086 · d588086
1 parent 8674e46
commit d588086
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/common/hist_util.cc b/src/common/hist_util.cc
@@ -329,6 +329,9 @@ void BuildHistDispatch(Span<GradientPair const> gpair, Span<bst_idx_t const> row
         return;
       }
       // contiguous memory access, built-in HW prefetching is enough
+      if (row_indices.size() == 0) {
+        return;
+      }
       RowsWiseBuildHistKernel<false, BuildingManager>(gpair, row_indices, gmat, hist);
     } else {
       auto span1 = row_indices.subspan(0, row_indices.size() - no_prefetch_size);