39390cc2ee
- A `DeviceOrd` struct is implemented to indicate the device. It will eventually replace the `gpu_id` parameter. - The `predictor` parameter is removed. - Fallback to `DMatrix` when `inplace_predict` is not available. - The heuristic for choosing a predictor is only used during training.
32 lines
1.0 KiB
Plaintext
32 lines
1.0 KiB
Plaintext
/**
|
|
* Copyright 2021-2023, XGBoost contributors
|
|
*/
|
|
#include "../common/device_helpers.cuh" // for CurrentDevice
|
|
#include "proxy_dmatrix.cuh" // for Dispatch, DMatrixProxy
|
|
#include "simple_dmatrix.cuh" // for CopyToSparsePage
|
|
#include "sparse_page_source.h"
|
|
#include "xgboost/data.h" // for SparsePage
|
|
|
|
namespace xgboost::data {
|
|
namespace detail {
|
|
std::size_t NSamplesDevice(DMatrixProxy *proxy) {
|
|
return cuda_impl::Dispatch(proxy, [](auto const &value) { return value.NumRows(); });
|
|
}
|
|
|
|
std::size_t NFeaturesDevice(DMatrixProxy *proxy) {
|
|
return cuda_impl::Dispatch(proxy, [](auto const &value) { return value.NumCols(); });
|
|
}
|
|
} // namespace detail
|
|
|
|
void DevicePush(DMatrixProxy *proxy, float missing, SparsePage *page) {
|
|
auto device = proxy->DeviceIdx();
|
|
if (device < 0) {
|
|
device = dh::CurrentDevice();
|
|
}
|
|
CHECK_GE(device, 0);
|
|
|
|
cuda_impl::Dispatch(proxy,
|
|
[&](auto const &value) { CopyToSparsePage(value, device, missing, page); });
|
|
}
|
|
} // namespace xgboost::data
|