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

Upgrade gputreeshap. (#6099)

Browse Source 
* Upgrade gputreeshap.

Co-authored-by: Rory Mitchell <r.a.mitchell.nz@gmail.com>
This commit is contained in:
Jiaming Yuan 2020-09-15 08:57:22 +08:00 committed by GitHub
parent 1453bee3e7
commit c6f2b8c841
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gputreeshap

@ -1 +1 @@
Subproject commit 1de23c95ff07d086db02837fb4a746b6924abbd5 Subproject commit 3fd8bb118ffc8516f86417253aacdae1531977f4

2
src/predictor/gpu_predictor.cu
View File

@ -30,6 +30,7 @@ struct SparsePageView {
common::Span<const bst_row_t> d_row_ptr; common::Span<const bst_row_t> d_row_ptr;
bst_feature_t num_features; bst_feature_t num_features;
SparsePageView() = default;
XGBOOST_DEVICE SparsePageView(common::Span<const Entry> data, XGBOOST_DEVICE SparsePageView(common::Span<const Entry> data,
common::Span<const bst_row_t> row_ptr, common::Span<const bst_row_t> row_ptr,
bst_feature_t num_features) bst_feature_t num_features)
@ -562,6 +563,7 @@ class GPUPredictor : public xgboost::Predictor {
<< " approximate is not implemented in GPU Predictor."; << " approximate is not implemented in GPU Predictor.";
} }
dh::safe_cuda(cudaSetDevice(generic_param_->gpu_id));
uint32_t real_ntree_limit = uint32_t real_ntree_limit =
ntree_limit * model.learner_model_param->num_output_group; ntree_limit * model.learner_model_param->num_output_group;
if (real_ntree_limit == 0 || real_ntree_limit > model.trees.size()) { if (real_ntree_limit == 0 || real_ntree_limit > model.trees.size()) {