Set device in device dmatrix. (#5596)

2020-04-25 13:42:53 +08:00
parent 3728855ce9
commit 844d7c1d5b
8 changed files with 41 additions and 5 deletions
--- a/src/learner.cc
+++ b/src/learner.cc
@@ -1052,7 +1052,7 @@ class LearnerImpl : public LearnerIO {

  void ValidateDMatrix(DMatrix* p_fmat) const {
    MetaInfo const& info = p_fmat->Info();
-    info.Validate();
+    info.Validate(generic_parameters_.gpu_id);

    auto const row_based_split = [this]() {
      return tparam_.dsplit == DataSplitMode::kRow ||