CPU compilation.

dmlc · Jul 14, 2021 · 4aa7d48 · 4aa7d48
1 parent d52a2b0
commit 4aa7d48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/data/iterative_device_dmatrix.h b/src/data/iterative_device_dmatrix.h
@@ -83,8 +83,8 @@ inline void IterativeDeviceDMatrix::Initialize(DataIterHandle iter, float missin
 }
 inline BatchSet<EllpackPage> IterativeDeviceDMatrix::GetEllpackBatches(const BatchParam& param) {
   common::AssertGPUSupport();
-  auto begin_iter =
-      BatchIterator<EllpackPage>(new SimpleBatchIteratorImpl<EllpackPage>(page_));
+  auto begin_iter = BatchIterator<EllpackPage>(
+      std::make_shared<SimpleBatchIteratorImpl<EllpackPage>>(page_));
   return BatchSet<EllpackPage>(BatchIterator<EllpackPage>(begin_iter));
 }
 #endif  // !defined(XGBOOST_USE_CUDA)