| 12 | |
| 13 | namespace xgboost::data { |
| 14 | TEST(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 |
nothing calls this directly
no test coverage detected