Set device in device dmatrix. (#5596)

2020-04-25 13:42:53 +08:00
parent ef26bc45bf
commit e726dd9902
8 changed files with 41 additions and 5 deletions
--- a/include/xgboost/data.h
+++ b/include/xgboost/data.h
@@ -102,7 +102,7 @@ class MetaInfo {
  /*!
   * \brief Validate all metainfo.
   */
-  void Validate() const;
+  void Validate(int32_t device) const;

  MetaInfo Slice(common::Span<int32_t const> ridxs) const;
  /*!