30 lines
1017 B
C++
30 lines
1017 B
C++
/*!
|
|
* Copyright 2021 by Contributors
|
|
* \file proxy_dmatrix.cc
|
|
*/
|
|
|
|
#include "proxy_dmatrix.h"
|
|
|
|
namespace xgboost {
|
|
namespace data {
|
|
void DMatrixProxy::SetArrayData(char const *c_interface) {
|
|
std::shared_ptr<ArrayAdapter> adapter{new ArrayAdapter(StringView{c_interface})};
|
|
this->batch_ = adapter;
|
|
this->Info().num_col_ = adapter->NumColumns();
|
|
this->Info().num_row_ = adapter->NumRows();
|
|
this->ctx_.gpu_id = Context::kCpuId;
|
|
}
|
|
|
|
void DMatrixProxy::SetCSRData(char const *c_indptr, char const *c_indices,
|
|
char const *c_values, bst_feature_t n_features, bool on_host) {
|
|
CHECK(on_host) << "Not implemented on device.";
|
|
std::shared_ptr<CSRArrayAdapter> adapter{new CSRArrayAdapter(
|
|
StringView{c_indptr}, StringView{c_indices}, StringView{c_values}, n_features)};
|
|
this->batch_ = adapter;
|
|
this->Info().num_col_ = adapter->NumColumns();
|
|
this->Info().num_row_ = adapter->NumRows();
|
|
this->ctx_.gpu_id = Context::kCpuId;
|
|
}
|
|
} // namespace data
|
|
} // namespace xgboost
|