| 119 | |
| 120 | namespace { |
| 121 | void TestColumnSplit() { |
| 122 | Context ctx; |
| 123 | size_t constexpr kRows = 5; |
| 124 | size_t constexpr kCols = 5; |
| 125 | auto dmat = RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix(); |
| 126 | |
| 127 | auto const world_size = collective::GetWorldSize(); |
| 128 | auto const rank = collective::GetRank(); |
| 129 | dmat = std::unique_ptr<DMatrix>{dmat->SliceCol(world_size, rank)}; |
| 130 | |
| 131 | TestBasic(dmat.get(), &ctx); |
| 132 | } |
| 133 | } // anonymous namespace |
| 134 | |
| 135 | TEST(CpuPredictor, BasicColumnSplit) { |
nothing calls this directly
no test coverage detected