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

Fix gpu_hist apply_split test. (#4158)

Browse Source
This commit is contained in:
Jiaming Yuan 2019-02-18 02:48:28 +08:00 committed by GitHub
parent 2e618af743
commit e1240413c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/cpp/predictor/test_gpu_predictor.cu
View File

@ -90,6 +90,7 @@ TEST(gpu_predictor, Test) {
delete dmat; delete dmat;
} }
#if defined(XGBOOST_USE_NCCL)
// Test whether pickling preserves predictor parameters // Test whether pickling preserves predictor parameters
TEST(gpu_predictor, MGPU_PicklingTest) { TEST(gpu_predictor, MGPU_PicklingTest) {
int ngpu; int ngpu;
@ -163,7 +164,9 @@ TEST(gpu_predictor, MGPU_PicklingTest) {
CheckCAPICall(XGBoosterFree(bst2)); CheckCAPICall(XGBoosterFree(bst2));
} }
#endif // defined(XGBOOST_USE_NCCL)
#if defined(XGBOOST_USE_NCCL)
// multi-GPU predictor test // multi-GPU predictor test
TEST(gpu_predictor, MGPU_Test) { TEST(gpu_predictor, MGPU_Test) {
std::unique_ptr<Predictor> gpu_predictor = std::unique_ptr<Predictor> gpu_predictor =
@ -202,6 +205,6 @@ TEST(gpu_predictor, MGPU_Test) {
delete dmat; delete dmat;
} }
} }
#endif // defined(XGBOOST_USE_NCCL)
} // namespace predictor } // namespace predictor
} // namespace xgboost } // namespace xgboost

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

@ -5,6 +5,9 @@
#include <thrust/device_vector.h> #include <thrust/device_vector.h>
#include <xgboost/base.h> #include <xgboost/base.h>
#include <random> #include <random>
#include <string>
#include <vector>
#include "../helpers.h" #include "../helpers.h"
#include "gtest/gtest.h" #include "gtest/gtest.h"
@ -315,6 +318,8 @@ TEST(GpuHist, ApplySplit) {
int constexpr n_cols = 8; int constexpr n_cols = 8;
TrainParam param; TrainParam param;
std::vector<std::pair<std::string, std::string>> args = {};
param.InitAllowUnknown(args);
// Initialize shard // Initialize shard
for (size_t i = 0; i < n_cols; ++i) { for (size_t i = 0; i < n_cols; ++i) {