Obtain CSR matrix from DMatrix. (#8269)

2022-09-29 20:41:43 +08:00
parent b14c44ee5e
commit 55cf24cc32
22 changed files with 400 additions and 74 deletions
--- a/src/data/iterative_dmatrix.cu
+++ b/src/data/iterative_dmatrix.cu
@@ -88,6 +88,9 @@ void IterativeDMatrix::InitFromCUDA(DataIterHandle iter_handle, float missing,
  } while (iter.Next());
  iter.Reset();

+  auto n_features = cols;
+  CHECK_GE(n_features, 1) << "Data must has at least 1 column.";
+
  dh::safe_cuda(cudaSetDevice(get_device()));
  if (!ref) {
    HostDeviceVector<FeatureType> ft;