Support cpu quantile sketch with column-wise data split (#8742)

2023-02-04 22:26:24 -08:00
parent c1786849e3
commit 66191e9926
15 changed files with 250 additions and 118 deletions
--- a/src/data/proxy_dmatrix.h
+++ b/src/data/proxy_dmatrix.h
@@ -87,7 +87,7 @@ class DMatrixProxy : public DMatrix {
    LOG(FATAL) << "Slicing DMatrix is not supported for Proxy DMatrix.";
    return nullptr;
  }
-  DMatrix* SliceCol(std::size_t, std::size_t) override {
+  DMatrix* SliceCol(int num_slices, int slice_id) override {
    LOG(FATAL) << "Slicing DMatrix columns is not supported for Proxy DMatrix.";
    return nullptr;
  }