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

Fix Google test warnings and error (#2957)

Browse Source
This commit is contained in:
Rory Mitchell
2017-12-20 00:13:56 +13:00
committed by GitHub
parent 76f8f51438
commit 7759ab99ee
3 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -22,9 +22,10 @@ TEST(gpu_hist_experimental, TestSparseShard) {
hmat.Init(dmat.get(), max_bins);
gmat.cut = &hmat;
gmat.Init(dmat.get());
ncclComm_t comm;
TrainParam p;
p.max_depth = 6;
DeviceShard shard(0, 0, gmat, 0, rows, hmat.row_ptr.back(),
TrainParam());
p);
ASSERT_LT(shard.row_stride, columns);
@@ -56,9 +57,10 @@ TEST(gpu_hist_experimental, TestDenseShard) {
hmat.Init(dmat.get(), max_bins);
gmat.cut = &hmat;
gmat.Init(dmat.get());
ncclComm_t comm;
TrainParam p;
p.max_depth = 6;
DeviceShard shard(0, 0, gmat, 0, rows, hmat.row_ptr.back(),
TrainParam());
p);
ASSERT_EQ(shard.row_stride, columns);