Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make HostDeviceVector single gpu only (#4773)

Browse Source 
* Make HostDeviceVector single gpu only
This commit is contained in:
Rong Ou
2019-08-25 14:51:13 -07:00
committed by Rory Mitchell
parent 41227d1933
commit 38ab79f889
54 changed files with 641 additions and 1621 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/cpp/tree/test_gpu_hist.cu
View File

@@ -366,7 +366,7 @@ TEST(GpuHist, EvaluateSplits) {
ASSERT_NEAR(res[1].fvalue, 0.26, xgboost::kRtEps);
}
void TestHistogramIndexImpl(int n_gpus) {
void TestHistogramIndexImpl() {
// Test if the compressed histogram index matches when using a sparse
// dmatrix with and without using external memory
@@ -384,7 +384,7 @@ void TestHistogramIndexImpl(int n_gpus) {
{"max_leaves", "0"}
};
GenericParameter generic_param(CreateEmptyGenericParam(0, n_gpus));
GenericParameter generic_param(CreateEmptyGenericParam(0));
hist_maker.Configure(training_params, &generic_param);
hist_maker.InitDataOnce(hist_maker_dmat.get());
@@ -412,7 +412,7 @@ void TestHistogramIndexImpl(int n_gpus) {
}
TEST(GpuHist, TestHistogramIndex) {
TestHistogramIndexImpl(1);
TestHistogramIndexImpl();
}
} // namespace tree

2
tests/cpp/tree/test_prune.cc
View File

@@ -29,7 +29,7 @@ TEST(Updater, Prune) {
{0.25f, 0.24f}, {0.25f, 0.24f}, {0.25f, 0.24f}, {0.25f, 0.24f} };
auto dmat = CreateDMatrix(32, 16, 0.4, 3);
auto lparam = CreateEmptyGenericParam(0, 0);
auto lparam = CreateEmptyGenericParam(GPUIDX);
// prepare tree
RegTree tree = RegTree();

2
tests/cpp/tree/test_refresh.cc
View File

@@ -25,7 +25,7 @@ TEST(Updater, Refresh) {
{"reg_lambda", "1"}};
RegTree tree = RegTree();
auto lparam = CreateEmptyGenericParam(0, 0);
auto lparam = CreateEmptyGenericParam(GPUIDX);
tree.param.InitAllowUnknown(cfg);
std::vector<RegTree*> trees {&tree};
std::unique_ptr<TreeUpdater> refresher(TreeUpdater::Create("refresh", &lparam));