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

Cleanup data generator. (#8094)

Browse Source 
- Avoid duplicated definition of data shape.
- Explicitly define numpy iterator for CPU data.
This commit is contained in:
Jiaming Yuan
2022-07-20 13:48:52 +08:00
committed by GitHub
parent 5156be0f49
commit ef11b024e8
4 changed files with 59 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/cpp/data/test_iterative_dmatrix.cu
View File

@@ -27,8 +27,8 @@ void TestEquivalent(float sparsity) {
offset += num_elements;
}
auto from_iter = page_concatenated->GetDeviceAccessor(0);
ASSERT_EQ(m.Info().num_col_, CudaArrayIterForTest::kCols);
ASSERT_EQ(m.Info().num_row_, CudaArrayIterForTest::kRows);
ASSERT_EQ(m.Info().num_col_, CudaArrayIterForTest::Cols());
ASSERT_EQ(m.Info().num_row_, CudaArrayIterForTest::Rows());
std::string interface_str = iter.AsArray();
auto adapter = CupyAdapter(interface_str);
@@ -98,8 +98,8 @@ TEST(IterativeDeviceDMatrix, RowMajor) {
auto impl = ellpack.Impl();
common::CompressedIterator<uint32_t> iterator(
impl->gidx_buffer.HostVector().data(), impl->NumSymbols());
auto cols = CudaArrayIterForTest::kCols;
auto rows = CudaArrayIterForTest::kRows;
auto cols = CudaArrayIterForTest::Cols();
auto rows = CudaArrayIterForTest::Rows();
auto j_interface =
Json::Load({interface_str.c_str(), interface_str.size()});