Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename IterativeDMatrix. (#8045)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-07-04 18:52:31 +08:00
committed by GitHub
parent f24bfc7684
commit 8746f9cddf
5 changed files with 29 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/cpp/helpers.cu
View File

@@ -2,7 +2,7 @@
#include "helpers.h"
#include "../../src/data/device_adapter.cuh"
#include "../../src/data/iterative_device_dmatrix.h"
#include "../../src/data/iterative_dmatrix.h"
namespace xgboost {
@@ -28,12 +28,10 @@ int CudaArrayIterForTest::Next() {
return 1;
}
std::shared_ptr<DMatrix> RandomDataGenerator::GenerateDeviceDMatrix() {
CudaArrayIterForTest iter{this->sparsity_, this->rows_, this->cols_, 1};
auto m = std::make_shared<data::IterativeDeviceDMatrix>(
&iter, iter.Proxy(), Reset, Next, std::numeric_limits<float>::quiet_NaN(),
0, bins_);
auto m = std::make_shared<data::IterativeDMatrix>(
&iter, iter.Proxy(), Reset, Next, std::numeric_limits<float>::quiet_NaN(), 0, bins_);
return m;
}
} // namespace xgboost