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

Fix GPU driver test. (#8008)

Browse Source 
* Initialize the training parameter.
This commit is contained in:
Jiaming Yuan 2022-06-20 19:37:31 +08:00 committed by GitHub
parent 637e42a0c0
commit 9b0eb66b78
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/cpp/tree/gpu_hist/test_driver.cu
View File

@ -7,8 +7,8 @@ namespace tree {
TEST(GpuHist, DriverDepthWise) { TEST(GpuHist, DriverDepthWise) {
TrainParam p; TrainParam p;
p.InitAllowUnknown(Args{}); p.UpdateAllowUnknown(Args{{"grow_policy", "depthwise"}});
p.grow_policy = TrainParam::kDepthWise;
Driver<GPUExpandEntry> driver(p, 2); Driver<GPUExpandEntry> driver(p, 2);
EXPECT_TRUE(driver.Pop().empty()); EXPECT_TRUE(driver.Pop().empty());
DeviceSplitCandidate split; DeviceSplitCandidate split;
@ -49,7 +49,8 @@ TEST(GpuHist, DriverLossGuided) {
low_gain.loss_chg = 1.0f; low_gain.loss_chg = 1.0f;
TrainParam p; TrainParam p;
p.grow_policy=TrainParam::kLossGuide; p.UpdateAllowUnknown(Args{{"grow_policy", "lossguide"}});
Driver<GPUExpandEntry> driver(p); Driver<GPUExpandEntry> driver(p);
EXPECT_TRUE(driver.Pop().empty()); EXPECT_TRUE(driver.Pop().empty());
GPUExpandEntry root(0, 0, high_gain, 2.0f, 1.0f, 1.0f ); GPUExpandEntry root(0, 0, high_gain, 2.0f, 1.0f, 1.0f );