MCPcopy Create free account
hub / github.com/dmlc/xgboost / TestColumnSplit

Function TestColumnSplit

tests/cpp/predictor/test_cpu_predictor.cc:121–132  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

119
120namespace {
121void 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
135TEST(CpuPredictor, BasicColumnSplit) {

Callers

nothing calls this directly

Calls 7

RandomDataGeneratorClass · 0.85
GetWorldSizeFunction · 0.85
GetRankFunction · 0.85
GenerateDMatrixMethod · 0.80
TestBasicFunction · 0.70
SliceColMethod · 0.45
getMethod · 0.45

Tested by

no test coverage detected