| 39 | namespace xgboost::tree { |
| 40 | namespace { |
| 41 | void InitRowPartitionForTest(common::RowSetCollection *row_set, size_t n_samples, |
| 42 | size_t base_rowid = 0) { |
| 43 | auto &row_indices = *row_set->Data(); |
| 44 | row_indices.resize(n_samples); |
| 45 | std::iota(row_indices.begin(), row_indices.end(), base_rowid); |
| 46 | row_set->Init(); |
| 47 | } |
| 48 | } // anonymous namespace |
| 49 | |
| 50 | void TestAddHistRows(bool is_distributed) { |