Implement GPU predict leaf. (#6187)

2020-11-11 17:33:47 +08:00
parent 7f101d1b33
commit 8a17610666
12 changed files with 252 additions and 42 deletions
--- a/tests/cpp/helpers.cc
+++ b/tests/cpp/helpers.cc
@@ -348,6 +348,13 @@ RandomDataGenerator::GenerateDMatrix(bool with_label, bool float_label,
      gen.GenerateDense(&out->Info().labels_);
    }
  }
+  if (device_ >= 0) {
+    out->Info().labels_.SetDevice(device_);
+    for (auto const& page : out->GetBatches<SparsePage>()) {
+      page.data.SetDevice(device_);
+      page.offset.SetDevice(device_);
+    }
+  }
  return out;
 }