Simplify sparse and dense CPU hist kernels (#7029)

* Simplify sparse and dense kernels * Extract row partitioner. Co-authored-by: Kirill Shvets <kirill.shvets@intel.com>
2021-06-11 13:26:30 +03:00
parent 1faad825f4
commit 2567404ab6
10 changed files with 369 additions and 434 deletions
--- a/src/common/hist_util.h
+++ b/src/common/hist_util.h
@@ -627,11 +627,11 @@ class GHistBuilder {
  GHistBuilder(size_t nthread, uint32_t nbins) : nthread_{nthread}, nbins_{nbins} {}

  // construct a histogram via histogram aggregation
+  template <bool any_missing>
  void BuildHist(const std::vector<GradientPair>& gpair,
                 const RowSetCollection::Elem row_indices,
                 const GHistIndexMatrix& gmat,
-                 GHistRowT hist,
-                 bool isDense);
+                 GHistRowT hist);
  // construct a histogram via subtraction trick
  void SubtractionTrick(GHistRowT self,
                        GHistRowT sibling,