[GPU-Plugin] Add GPU accelerated prediction (#2593)

* [GPU-Plugin] Add GPU accelerated prediction

* Improve allocation message

* Update documentation

* Resolve linker error for predictor

* Add unit tests
This commit is contained in:
Rory Mitchell
2017-08-16 12:31:59 +12:00
committed by GitHub
parent 71e5e622b1
commit ef23e424f1
25 changed files with 876 additions and 203 deletions

14
tests/cpp/test_learner.cc Normal file
View File

@@ -0,0 +1,14 @@
// Copyright by Contributors
#include <gtest/gtest.h>
#include "helpers.h"
#include "xgboost/learner.h"
namespace xgboost {
TEST(learner, Test) {
typedef std::pair<std::string, std::string> arg;
auto args = {arg("tree_method", "exact")};
auto mat = {CreateDMatrix(10, 10, 0)};
auto learner = std::unique_ptr<Learner>(Learner::Create(mat));
learner->Configure(args);
}
} // namespace xgboost