Support column-split in row partitioner (#8828)

2023-02-25 12:43:35 -08:00
parent 90c0633a28
commit d9688f93c7
8 changed files with 283 additions and 58 deletions
--- a/src/data/gradient_index.h
+++ b/src/data/gradient_index.h
@@ -228,6 +228,8 @@ class GHistIndexMatrix {

  common::ColumnMatrix const& Transpose() const;

+  bst_bin_t GetGindex(size_t ridx, size_t fidx) const;
+
  float GetFvalue(size_t ridx, size_t fidx, bool is_cat) const;

 private: