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

Function TEST

tests/cpp/data/test_proxy_dmatrix.cc:14–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

12
13namespace xgboost::data {
14TEST(ProxyDMatrix, HostData) {
15 DMatrixProxy proxy;
16 std::size_t constexpr kRows = 100, kCols = 10;
17 std::vector<HostDeviceVector<float>> label_storage(1);
18
19 HostDeviceVector<float> storage;
20 auto data =
21 RandomDataGenerator(kRows, kCols, 0.5).Device(FstCU()).GenerateArrayInterface(&storage);
22
23 proxy.SetArray(data.c_str());
24 using cpu_impl::DispatchAny;
25
26 auto n_samples = DispatchAny(&proxy, [](auto const &value) { return value.Size(); });
27 ASSERT_EQ(n_samples, kRows);
28 auto n_features = DispatchAny(&proxy, [](auto const &value) { return value.NumCols(); });
29 ASSERT_EQ(n_features, kCols);
30}
31} // namespace xgboost::data

Callers

nothing calls this directly

Calls 9

RandomDataGeneratorClass · 0.85
FstCUFunction · 0.85
DispatchAnyFunction · 0.85
SetArrayMethod · 0.80
c_strMethod · 0.80
DeviceMethod · 0.45
SizeMethod · 0.45
NumColsMethod · 0.45

Tested by

no test coverage detected