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

Fixed performance bug (#3171)

Browse Source 
Minor performance improvements to gpu predictor
This commit is contained in:
Ray Kim 2018-03-15 05:40:24 +09:00 committed by Rory Mitchell
parent 7a81c87dfa
commit cdc036b752
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/predictor/gpu_predictor.cu
View File

@ -277,7 +277,7 @@ class GPUPredictor : public xgboost::Predictor {
CHECK_EQ(model.param.size_leaf_vector, 0); CHECK_EQ(model.param.size_leaf_vector, 0);
// Copy decision trees to device // Copy decision trees to device
thrust::host_vector<size_t> h_tree_segments; thrust::host_vector<size_t> h_tree_segments;
h_tree_segments.reserve((tree_end - tree_end) + 1); h_tree_segments.reserve((tree_end - tree_begin) + 1);
size_t sum = 0; size_t sum = 0;
h_tree_segments.push_back(sum); h_tree_segments.push_back(sum);
for (auto tree_idx = tree_begin; tree_idx < tree_end; tree_idx++) { for (auto tree_idx = tree_begin; tree_idx < tree_end; tree_idx++) {