e6088366df
* Add Python iterator interface. * Add tests. * Add demo. * Add documents. * Handle empty dataset.
35 lines
958 B
Plaintext
35 lines
958 B
Plaintext
/*!
|
|
* Copyright 2020 XGBoost contributors
|
|
*/
|
|
#include "proxy_dmatrix.h"
|
|
#include "device_adapter.cuh"
|
|
|
|
namespace xgboost {
|
|
namespace data {
|
|
|
|
void DMatrixProxy::FromCudaColumnar(std::string interface_str) {
|
|
std::shared_ptr<data::CudfAdapter> adapter {new data::CudfAdapter(interface_str)};
|
|
auto const& value = adapter->Value();
|
|
this->batch_ = adapter;
|
|
device_ = adapter->DeviceIdx();
|
|
this->Info().num_col_ = adapter->NumColumns();
|
|
this->Info().num_row_ = adapter->NumRows();
|
|
if (device_ < 0) {
|
|
CHECK_EQ(this->Info().num_row_, 0);
|
|
}
|
|
}
|
|
|
|
void DMatrixProxy::FromCudaArray(std::string interface_str) {
|
|
std::shared_ptr<CupyAdapter> adapter(new CupyAdapter(interface_str));
|
|
this->batch_ = adapter;
|
|
device_ = adapter->DeviceIdx();
|
|
this->Info().num_col_ = adapter->NumColumns();
|
|
this->Info().num_row_ = adapter->NumRows();
|
|
if (device_ < 0) {
|
|
CHECK_EQ(this->Info().num_row_, 0);
|
|
}
|
|
}
|
|
|
|
} // namespace data
|
|
} // namespace xgboost
|